github.com/thajeztah/cli@v0.0.0-20240223162942-dc6bfac81a8b/cli/command/network/client_test.go (about)

     1  package network
     2  
     3  import (
     4  	"context"
     5  
     6  	"github.com/docker/docker/api/types"
     7  	"github.com/docker/docker/api/types/network"
     8  	"github.com/docker/docker/client"
     9  )
    10  
    11  type fakeClient struct {
    12  	client.Client
    13  	networkCreateFunc     func(ctx context.Context, name string, options types.NetworkCreate) (types.NetworkCreateResponse, error)
    14  	networkConnectFunc    func(ctx context.Context, networkID, container string, config *network.EndpointSettings) error
    15  	networkDisconnectFunc func(ctx context.Context, networkID, container string, force bool) error
    16  	networkRemoveFunc     func(ctx context.Context, networkID string) error
    17  	networkListFunc       func(ctx context.Context, options types.NetworkListOptions) ([]types.NetworkResource, error)
    18  }
    19  
    20  func (c *fakeClient) NetworkCreate(ctx context.Context, name string, options types.NetworkCreate) (types.NetworkCreateResponse, error) {
    21  	if c.networkCreateFunc != nil {
    22  		return c.networkCreateFunc(ctx, name, options)
    23  	}
    24  	return types.NetworkCreateResponse{}, nil
    25  }
    26  
    27  func (c *fakeClient) NetworkConnect(ctx context.Context, networkID, container string, config *network.EndpointSettings) error {
    28  	if c.networkConnectFunc != nil {
    29  		return c.networkConnectFunc(ctx, networkID, container, config)
    30  	}
    31  	return nil
    32  }
    33  
    34  func (c *fakeClient) NetworkDisconnect(ctx context.Context, networkID, container string, force bool) error {
    35  	if c.networkDisconnectFunc != nil {
    36  		return c.networkDisconnectFunc(ctx, networkID, container, force)
    37  	}
    38  	return nil
    39  }
    40  
    41  func (c *fakeClient) NetworkList(ctx context.Context, options types.NetworkListOptions) ([]types.NetworkResource, error) {
    42  	if c.networkListFunc != nil {
    43  		return c.networkListFunc(ctx, options)
    44  	}
    45  	return []types.NetworkResource{}, nil
    46  }
    47  
    48  func (c *fakeClient) NetworkRemove(ctx context.Context, networkID string) error {
    49  	if c.networkRemoveFunc != nil {
    50  		return c.networkRemoveFunc(ctx, networkID)
    51  	}
    52  	return nil
    53  }
    54  
    55  func (c *fakeClient) NetworkInspectWithRaw(context.Context, string, types.NetworkInspectOptions) (types.NetworkResource, []byte, error) {
    56  	return types.NetworkResource{}, nil, nil
    57  }