internal/testutil/daemon/plugin.go

package daemon

import (
	"context"
	"testing"

	cerrdefs "github.com/containerd/errdefs"
	"github.com/moby/moby/api/types/plugin"
	"github.com/moby/moby/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) poll.Check {
	return withClient(t, d, withPluginInspect(name, func(plugin *plugin.Plugin) 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) poll.Check {
	return withClient(t, d, withPluginInspect(name, func(p *plugin.Plugin) poll.Result {
		if !p.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) poll.Check {
	return withClient(t, d, func(c client.APIClient) poll.Result {
		_, err := c.PluginInspect(t.Context(), name, client.PluginInspectOptions{})
		if cerrdefs.IsNotFound(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) poll.Check {
	return withClient(t, d, withPluginInspect(name, func(p *plugin.Plugin) poll.Result {
		if p.PluginReference == expectedRef {
			return poll.Success()
		}
		return poll.Continue("plugin %q reference is not %q", name, expectedRef)
	}))
}

func withPluginInspect(name string, f func(*plugin.Plugin) poll.Result) func(client.APIClient) poll.Result {
	return func(c client.APIClient) poll.Result {
		res, err := c.PluginInspect(context.Background(), name, client.PluginInspectOptions{})
		if cerrdefs.IsNotFound(err) {
			return poll.Continue("plugin %q not found", name)
		}
		if err != nil {
			return poll.Error(err)
		}
		return f(&res.Plugin)
	}
}

func withClient(t testing.TB, d *Daemon, f func(client.APIClient) poll.Result) poll.Check {
	return func(poll.LogT) poll.Result {
		c := d.NewClientT(t)
		return f(c)
	}
}