mirror of
https://github.com/moby/moby.git
synced 2022-11-09 12:21:53 -05:00
b37c214e3c
This was done with something along the lines of:
```
mv internal/test testutil
pushd testutil/; grep -IRl "package test" | xargs -I '{}' sed -i -e 's|package test|package testutil|g' {}; popd
mv internal/testutil/*.go testutil/ && rm -rf internal/
grep -IRl "github.com\/docker\/docker\/internal\/test" | xargs -I '{}' sed -i -e 's|github.com/docker/docker/internal/test|github.com/docker/docker/test|g' {}
goimports .
```
I also modified the basic plugin path in testutil/fixtures/plugin.
Signed-off-by: Sam Whited <sam@samwhited.com>
75 lines
2.5 KiB
Go
75 lines
2.5 KiB
Go
package daemon
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/client"
|
|
"gotest.tools/assert"
|
|
"gotest.tools/poll"
|
|
)
|
|
|
|
// PluginIsRunning provides a poller to check if the specified plugin is running
|
|
func (d *Daemon) PluginIsRunning(t assert.TestingT, 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 assert.TestingT, 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 assert.TestingT, 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 assert.TestingT, 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 assert.TestingT, 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)
|
|
}
|
|
}
|