github.1git.de/docker/cli@v26.1.3+incompatible/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/filters"
     8  	"github.com/docker/docker/api/types/network"
     9  	"github.com/docker/docker/client"
    10  )
    11  
    12  type fakeClient struct {
    13  	client.Client
    14  	networkCreateFunc     func(ctx context.Context, name string, options types.NetworkCreate) (types.NetworkCreateResponse, error)
    15  	networkConnectFunc    func(ctx context.Context, networkID, container string, config *network.EndpointSettings) error
    16  	networkDisconnectFunc func(ctx context.Context, networkID, container string, force bool) error
    17  	networkRemoveFunc     func(ctx context.Context, networkID string) error
    18  	networkListFunc       func(ctx context.Context, options types.NetworkListOptions) ([]types.NetworkResource, error)
    19  	networkPruneFunc      func(ctx context.Context, pruneFilters filters.Args) (types.NetworksPruneReport, error)
    20  	networkInspectFunc    func(ctx context.Context, networkID string, options types.NetworkInspectOptions) (types.NetworkResource, []byte, error)
    21  }
    22  
    23  func (c *fakeClient) NetworkCreate(ctx context.Context, name string, options types.NetworkCreate) (types.NetworkCreateResponse, error) {
    24  	if c.networkCreateFunc != nil {
    25  		return c.networkCreateFunc(ctx, name, options)
    26  	}
    27  	return types.NetworkCreateResponse{}, nil
    28  }
    29  
    30  func (c *fakeClient) NetworkConnect(ctx context.Context, networkID, container string, config *network.EndpointSettings) error {
    31  	if c.networkConnectFunc != nil {
    32  		return c.networkConnectFunc(ctx, networkID, container, config)
    33  	}
    34  	return nil
    35  }
    36  
    37  func (c *fakeClient) NetworkDisconnect(ctx context.Context, networkID, container string, force bool) error {
    38  	if c.networkDisconnectFunc != nil {
    39  		return c.networkDisconnectFunc(ctx, networkID, container, force)
    40  	}
    41  	return nil
    42  }
    43  
    44  func (c *fakeClient) NetworkList(ctx context.Context, options types.NetworkListOptions) ([]types.NetworkResource, error) {
    45  	if c.networkListFunc != nil {
    46  		return c.networkListFunc(ctx, options)
    47  	}
    48  	return []types.NetworkResource{}, nil
    49  }
    50  
    51  func (c *fakeClient) NetworkRemove(ctx context.Context, networkID string) error {
    52  	if c.networkRemoveFunc != nil {
    53  		return c.networkRemoveFunc(ctx, networkID)
    54  	}
    55  	return nil
    56  }
    57  
    58  func (c *fakeClient) NetworkInspectWithRaw(ctx context.Context, networkID string, opts types.NetworkInspectOptions) (types.NetworkResource, []byte, error) {
    59  	if c.networkInspectFunc != nil {
    60  		return c.networkInspectFunc(ctx, networkID, opts)
    61  	}
    62  	return types.NetworkResource{}, nil, nil
    63  }
    64  
    65  func (c *fakeClient) NetworksPrune(ctx context.Context, pruneFilter filters.Args) (types.NetworksPruneReport, error) {
    66  	if c.networkPruneFunc != nil {
    67  		return c.networkPruneFunc(ctx, pruneFilter)
    68  	}
    69  	return types.NetworksPruneReport{}, nil
    70  }