9f0b3f5609
full diff: https://github.com/gotestyourself/gotest.tools/compare/v2.3.0...v3.0.1 Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
75 lines
2.5 KiB
Go
75 lines
2.5 KiB
Go
package daemon
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/client"
|
|
"gotest.tools/v3/poll"
|
|
)
|
|
|
|
// PluginIsRunning provides a poller to check if the specified plugin is running
|
|
func (d *Daemon) PluginIsRunning(t testing.TB, name string) func(poll.LogT) poll.Result {
|
|
return withClient(t, d, withPluginInspect(name, func(plugin *types.Plugin, t poll.LogT) poll.Result {
|
|
if plugin.Enabled {
|
|
return poll.Success()
|
|
}
|
|
return poll.Continue("plugin %q is not enabled", name)
|
|
}))
|
|
}
|
|
|
|
// PluginIsNotRunning provides a poller to check if the specified plugin is not running
|
|
func (d *Daemon) PluginIsNotRunning(t testing.TB, name string) func(poll.LogT) poll.Result {
|
|
return withClient(t, d, withPluginInspect(name, func(plugin *types.Plugin, t poll.LogT) poll.Result {
|
|
if !plugin.Enabled {
|
|
return poll.Success()
|
|
}
|
|
return poll.Continue("plugin %q is enabled", name)
|
|
}))
|
|
}
|
|
|
|
// PluginIsNotPresent provides a poller to check if the specified plugin is not present
|
|
func (d *Daemon) PluginIsNotPresent(t testing.TB, name string) func(poll.LogT) poll.Result {
|
|
return withClient(t, d, func(c client.APIClient, t poll.LogT) poll.Result {
|
|
_, _, err := c.PluginInspectWithRaw(context.Background(), name)
|
|
if client.IsErrNotFound(err) {
|
|
return poll.Success()
|
|
}
|
|
if err != nil {
|
|
return poll.Error(err)
|
|
}
|
|
return poll.Continue("plugin %q exists", name)
|
|
})
|
|
}
|
|
|
|
// PluginReferenceIs provides a poller to check if the specified plugin has the specified reference
|
|
func (d *Daemon) PluginReferenceIs(t testing.TB, name, expectedRef string) func(poll.LogT) poll.Result {
|
|
return withClient(t, d, withPluginInspect(name, func(plugin *types.Plugin, t poll.LogT) poll.Result {
|
|
if plugin.PluginReference == expectedRef {
|
|
return poll.Success()
|
|
}
|
|
return poll.Continue("plugin %q reference is not %q", name, expectedRef)
|
|
}))
|
|
}
|
|
|
|
func withPluginInspect(name string, f func(*types.Plugin, poll.LogT) poll.Result) func(client.APIClient, poll.LogT) poll.Result {
|
|
return func(c client.APIClient, t poll.LogT) poll.Result {
|
|
plugin, _, err := c.PluginInspectWithRaw(context.Background(), name)
|
|
if client.IsErrNotFound(err) {
|
|
return poll.Continue("plugin %q not found", name)
|
|
}
|
|
if err != nil {
|
|
return poll.Error(err)
|
|
}
|
|
return f(plugin, t)
|
|
}
|
|
|
|
}
|
|
|
|
func withClient(t testing.TB, d *Daemon, f func(client.APIClient, poll.LogT) poll.Result) func(poll.LogT) poll.Result {
|
|
return func(pt poll.LogT) poll.Result {
|
|
c := d.NewClientT(t)
|
|
return f(c, pt)
|
|
}
|
|
}
|