github.com/xeptore/docker-cli@v20.10.14+incompatible/cli/command/checkpoint/client_test.go (about)

     1  package checkpoint
     2  
     3  import (
     4  	"context"
     5  
     6  	"github.com/docker/docker/api/types"
     7  	"github.com/docker/docker/client"
     8  )
     9  
    10  type fakeClient struct {
    11  	client.Client
    12  	checkpointCreateFunc func(container string, options types.CheckpointCreateOptions) error
    13  	checkpointDeleteFunc func(container string, options types.CheckpointDeleteOptions) error
    14  	checkpointListFunc   func(container string, options types.CheckpointListOptions) ([]types.Checkpoint, error)
    15  }
    16  
    17  func (cli *fakeClient) CheckpointCreate(ctx context.Context, container string, options types.CheckpointCreateOptions) error {
    18  	if cli.checkpointCreateFunc != nil {
    19  		return cli.checkpointCreateFunc(container, options)
    20  	}
    21  	return nil
    22  }
    23  
    24  func (cli *fakeClient) CheckpointDelete(ctx context.Context, container string, options types.CheckpointDeleteOptions) error {
    25  	if cli.checkpointDeleteFunc != nil {
    26  		return cli.checkpointDeleteFunc(container, options)
    27  	}
    28  	return nil
    29  }
    30  
    31  func (cli *fakeClient) CheckpointList(ctx context.Context, container string, options types.CheckpointListOptions) ([]types.Checkpoint, error) {
    32  	if cli.checkpointListFunc != nil {
    33  		return cli.checkpointListFunc(container, options)
    34  	}
    35  	return []types.Checkpoint{}, nil
    36  }