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

     1  package swarm
     2  
     3  import (
     4  	"context"
     5  
     6  	"github.com/docker/docker/api/types"
     7  	"github.com/docker/docker/api/types/swarm"
     8  	"github.com/docker/docker/api/types/system"
     9  	"github.com/docker/docker/client"
    10  )
    11  
    12  type fakeClient struct {
    13  	client.Client
    14  	infoFunc              func() (system.Info, error)
    15  	swarmInitFunc         func() (string, error)
    16  	swarmInspectFunc      func() (swarm.Swarm, error)
    17  	nodeInspectFunc       func() (swarm.Node, []byte, error)
    18  	swarmGetUnlockKeyFunc func() (types.SwarmUnlockKeyResponse, error)
    19  	swarmJoinFunc         func() error
    20  	swarmLeaveFunc        func() error
    21  	swarmUpdateFunc       func(swarm swarm.Spec, flags swarm.UpdateFlags) error
    22  	swarmUnlockFunc       func(req swarm.UnlockRequest) error
    23  }
    24  
    25  func (cli *fakeClient) Info(context.Context) (system.Info, error) {
    26  	if cli.infoFunc != nil {
    27  		return cli.infoFunc()
    28  	}
    29  	return system.Info{}, nil
    30  }
    31  
    32  func (cli *fakeClient) NodeInspectWithRaw(context.Context, string) (swarm.Node, []byte, error) {
    33  	if cli.nodeInspectFunc != nil {
    34  		return cli.nodeInspectFunc()
    35  	}
    36  	return swarm.Node{}, []byte{}, nil
    37  }
    38  
    39  func (cli *fakeClient) SwarmInit(context.Context, swarm.InitRequest) (string, error) {
    40  	if cli.swarmInitFunc != nil {
    41  		return cli.swarmInitFunc()
    42  	}
    43  	return "", nil
    44  }
    45  
    46  func (cli *fakeClient) SwarmInspect(context.Context) (swarm.Swarm, error) {
    47  	if cli.swarmInspectFunc != nil {
    48  		return cli.swarmInspectFunc()
    49  	}
    50  	return swarm.Swarm{}, nil
    51  }
    52  
    53  func (cli *fakeClient) SwarmGetUnlockKey(context.Context) (types.SwarmUnlockKeyResponse, error) {
    54  	if cli.swarmGetUnlockKeyFunc != nil {
    55  		return cli.swarmGetUnlockKeyFunc()
    56  	}
    57  	return types.SwarmUnlockKeyResponse{}, nil
    58  }
    59  
    60  func (cli *fakeClient) SwarmJoin(context.Context, swarm.JoinRequest) error {
    61  	if cli.swarmJoinFunc != nil {
    62  		return cli.swarmJoinFunc()
    63  	}
    64  	return nil
    65  }
    66  
    67  func (cli *fakeClient) SwarmLeave(context.Context, bool) error {
    68  	if cli.swarmLeaveFunc != nil {
    69  		return cli.swarmLeaveFunc()
    70  	}
    71  	return nil
    72  }
    73  
    74  func (cli *fakeClient) SwarmUpdate(_ context.Context, _ swarm.Version, swarmSpec swarm.Spec, flags swarm.UpdateFlags) error {
    75  	if cli.swarmUpdateFunc != nil {
    76  		return cli.swarmUpdateFunc(swarmSpec, flags)
    77  	}
    78  	return nil
    79  }
    80  
    81  func (cli *fakeClient) SwarmUnlock(_ context.Context, req swarm.UnlockRequest) error {
    82  	if cli.swarmUnlockFunc != nil {
    83  		return cli.swarmUnlockFunc(req)
    84  	}
    85  	return nil
    86  }