github.1git.de/docker/cli@v26.1.3+incompatible/cli/command/plugin/client_test.go (about)

     1  package plugin
     2  
     3  import (
     4  	"context"
     5  	"io"
     6  
     7  	"github.com/docker/docker/api/types"
     8  	"github.com/docker/docker/api/types/filters"
     9  	"github.com/docker/docker/api/types/system"
    10  	"github.com/docker/docker/client"
    11  )
    12  
    13  type fakeClient struct {
    14  	client.Client
    15  	pluginCreateFunc  func(createContext io.Reader, createOptions types.PluginCreateOptions) error
    16  	pluginDisableFunc func(name string, disableOptions types.PluginDisableOptions) error
    17  	pluginEnableFunc  func(name string, options types.PluginEnableOptions) error
    18  	pluginRemoveFunc  func(name string, options types.PluginRemoveOptions) error
    19  	pluginInstallFunc func(name string, options types.PluginInstallOptions) (io.ReadCloser, error)
    20  	pluginListFunc    func(filter filters.Args) (types.PluginsListResponse, error)
    21  	pluginInspectFunc func(name string) (*types.Plugin, []byte, error)
    22  	pluginUpgradeFunc func(name string, options types.PluginInstallOptions) (io.ReadCloser, error)
    23  }
    24  
    25  func (c *fakeClient) PluginCreate(_ context.Context, createContext io.Reader, createOptions types.PluginCreateOptions) error {
    26  	if c.pluginCreateFunc != nil {
    27  		return c.pluginCreateFunc(createContext, createOptions)
    28  	}
    29  	return nil
    30  }
    31  
    32  func (c *fakeClient) PluginEnable(_ context.Context, name string, enableOptions types.PluginEnableOptions) error {
    33  	if c.pluginEnableFunc != nil {
    34  		return c.pluginEnableFunc(name, enableOptions)
    35  	}
    36  	return nil
    37  }
    38  
    39  func (c *fakeClient) PluginDisable(_ context.Context, name string, disableOptions types.PluginDisableOptions) error {
    40  	if c.pluginDisableFunc != nil {
    41  		return c.pluginDisableFunc(name, disableOptions)
    42  	}
    43  	return nil
    44  }
    45  
    46  func (c *fakeClient) PluginRemove(_ context.Context, name string, removeOptions types.PluginRemoveOptions) error {
    47  	if c.pluginRemoveFunc != nil {
    48  		return c.pluginRemoveFunc(name, removeOptions)
    49  	}
    50  	return nil
    51  }
    52  
    53  func (c *fakeClient) PluginInstall(_ context.Context, name string, installOptions types.PluginInstallOptions) (io.ReadCloser, error) {
    54  	if c.pluginInstallFunc != nil {
    55  		return c.pluginInstallFunc(name, installOptions)
    56  	}
    57  	return nil, nil
    58  }
    59  
    60  func (c *fakeClient) PluginList(_ context.Context, filter filters.Args) (types.PluginsListResponse, error) {
    61  	if c.pluginListFunc != nil {
    62  		return c.pluginListFunc(filter)
    63  	}
    64  
    65  	return types.PluginsListResponse{}, nil
    66  }
    67  
    68  func (c *fakeClient) PluginInspectWithRaw(_ context.Context, name string) (*types.Plugin, []byte, error) {
    69  	if c.pluginInspectFunc != nil {
    70  		return c.pluginInspectFunc(name)
    71  	}
    72  
    73  	return nil, nil, nil
    74  }
    75  
    76  func (c *fakeClient) Info(context.Context) (system.Info, error) {
    77  	return system.Info{}, nil
    78  }
    79  
    80  func (c *fakeClient) PluginUpgrade(ctx context.Context, name string, options types.PluginInstallOptions) (io.ReadCloser, error) {
    81  	if c.pluginUpgradeFunc != nil {
    82  		return c.pluginUpgradeFunc(name, options)
    83  	}
    84  	return nil, nil
    85  }