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  }