85 lines
2.7 KiB
Go
85 lines
2.7 KiB
Go
package plugin
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
|
|
"github.com/moby/moby/api/types/plugin"
|
|
"github.com/moby/moby/api/types/system"
|
|
"github.com/moby/moby/client"
|
|
)
|
|
|
|
type fakeClient struct {
|
|
client.Client
|
|
pluginCreateFunc func(createContext io.Reader, createOptions client.PluginCreateOptions) error
|
|
pluginDisableFunc func(name string, disableOptions client.PluginDisableOptions) error
|
|
pluginEnableFunc func(name string, options client.PluginEnableOptions) error
|
|
pluginRemoveFunc func(name string, options client.PluginRemoveOptions) error
|
|
pluginInstallFunc func(name string, options client.PluginInstallOptions) (io.ReadCloser, error)
|
|
pluginListFunc func(filter client.Filters) (plugin.ListResponse, error)
|
|
pluginInspectFunc func(name string) (*plugin.Plugin, []byte, error)
|
|
pluginUpgradeFunc func(name string, options client.PluginInstallOptions) (io.ReadCloser, error)
|
|
}
|
|
|
|
func (c *fakeClient) PluginCreate(_ context.Context, createContext io.Reader, createOptions client.PluginCreateOptions) error {
|
|
if c.pluginCreateFunc != nil {
|
|
return c.pluginCreateFunc(createContext, createOptions)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *fakeClient) PluginEnable(_ context.Context, name string, enableOptions client.PluginEnableOptions) error {
|
|
if c.pluginEnableFunc != nil {
|
|
return c.pluginEnableFunc(name, enableOptions)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *fakeClient) PluginDisable(_ context.Context, name string, disableOptions client.PluginDisableOptions) error {
|
|
if c.pluginDisableFunc != nil {
|
|
return c.pluginDisableFunc(name, disableOptions)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *fakeClient) PluginRemove(_ context.Context, name string, removeOptions client.PluginRemoveOptions) error {
|
|
if c.pluginRemoveFunc != nil {
|
|
return c.pluginRemoveFunc(name, removeOptions)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *fakeClient) PluginInstall(_ context.Context, name string, installOptions client.PluginInstallOptions) (io.ReadCloser, error) {
|
|
if c.pluginInstallFunc != nil {
|
|
return c.pluginInstallFunc(name, installOptions)
|
|
}
|
|
return nil, nil
|
|
}
|
|
|
|
func (c *fakeClient) PluginList(_ context.Context, filter client.Filters) (plugin.ListResponse, error) {
|
|
if c.pluginListFunc != nil {
|
|
return c.pluginListFunc(filter)
|
|
}
|
|
|
|
return plugin.ListResponse{}, nil
|
|
}
|
|
|
|
func (c *fakeClient) PluginInspectWithRaw(_ context.Context, name string) (*plugin.Plugin, []byte, error) {
|
|
if c.pluginInspectFunc != nil {
|
|
return c.pluginInspectFunc(name)
|
|
}
|
|
|
|
return nil, nil, nil
|
|
}
|
|
|
|
func (*fakeClient) Info(context.Context) (system.Info, error) {
|
|
return system.Info{}, nil
|
|
}
|
|
|
|
func (c *fakeClient) PluginUpgrade(_ context.Context, name string, options client.PluginInstallOptions) (io.ReadCloser, error) {
|
|
if c.pluginUpgradeFunc != nil {
|
|
return c.pluginUpgradeFunc(name, options)
|
|
}
|
|
return nil, nil
|
|
}
|