github.com/khulnasoft/cli@v0.0.0-20240402070845-01bcad7beefa/cli/command/system/client_test.go (about) 1 package system 2 3 import ( 4 "context" 5 6 "github.com/docker/docker/api/types" 7 "github.com/docker/docker/api/types/events" 8 "github.com/docker/docker/api/types/filters" 9 "github.com/docker/docker/client" 10 ) 11 12 type fakeClient struct { 13 client.Client 14 15 version string 16 serverVersion func(ctx context.Context) (types.Version, error) 17 eventsFn func(context.Context, types.EventsOptions) (<-chan events.Message, <-chan error) 18 containerPruneFunc func(ctx context.Context, pruneFilters filters.Args) (types.ContainersPruneReport, error) 19 networkPruneFunc func(ctx context.Context, pruneFilter filters.Args) (types.NetworksPruneReport, error) 20 } 21 22 func (cli *fakeClient) ServerVersion(ctx context.Context) (types.Version, error) { 23 return cli.serverVersion(ctx) 24 } 25 26 func (cli *fakeClient) ClientVersion() string { 27 return cli.version 28 } 29 30 func (cli *fakeClient) Events(ctx context.Context, opts types.EventsOptions) (<-chan events.Message, <-chan error) { 31 return cli.eventsFn(ctx, opts) 32 } 33 34 func (cli *fakeClient) ContainersPrune(ctx context.Context, pruneFilters filters.Args) (types.ContainersPruneReport, error) { 35 if cli.containerPruneFunc != nil { 36 return cli.containerPruneFunc(ctx, pruneFilters) 37 } 38 return types.ContainersPruneReport{}, nil 39 } 40 41 func (cli *fakeClient) NetworksPrune(ctx context.Context, pruneFilter filters.Args) (types.NetworksPruneReport, error) { 42 if cli.networkPruneFunc != nil { 43 return cli.networkPruneFunc(ctx, pruneFilter) 44 } 45 return types.NetworksPruneReport{}, nil 46 }