0ea587b0d7
The code is similar to that used by the volume rm subcommand, however, one difference I noticed was VolumeRemove takes the force flag/option was a parameter. This isn't the case for NetworkRemove. To get NetworkRemove to take a similar parameter, this would require modifying the Docker daemon. For now this isn't a route I wish to take when the code can be arrange to mimic the same behavior. Co-authored-by: Sebastiaan van Stijn <github@gone.nl> Signed-off-by: Conner Crosby <conner@cavcrosby.tech>
58 lines
2.0 KiB
Go
58 lines
2.0 KiB
Go
package network
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/network"
|
|
"github.com/docker/docker/client"
|
|
)
|
|
|
|
type fakeClient struct {
|
|
client.Client
|
|
networkCreateFunc func(ctx context.Context, name string, options types.NetworkCreate) (types.NetworkCreateResponse, error)
|
|
networkConnectFunc func(ctx context.Context, networkID, container string, config *network.EndpointSettings) error
|
|
networkDisconnectFunc func(ctx context.Context, networkID, container string, force bool) error
|
|
networkRemoveFunc func(ctx context.Context, networkID string) error
|
|
networkListFunc func(ctx context.Context, options types.NetworkListOptions) ([]types.NetworkResource, error)
|
|
}
|
|
|
|
func (c *fakeClient) NetworkCreate(ctx context.Context, name string, options types.NetworkCreate) (types.NetworkCreateResponse, error) {
|
|
if c.networkCreateFunc != nil {
|
|
return c.networkCreateFunc(ctx, name, options)
|
|
}
|
|
return types.NetworkCreateResponse{}, nil
|
|
}
|
|
|
|
func (c *fakeClient) NetworkConnect(ctx context.Context, networkID, container string, config *network.EndpointSettings) error {
|
|
if c.networkConnectFunc != nil {
|
|
return c.networkConnectFunc(ctx, networkID, container, config)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *fakeClient) NetworkDisconnect(ctx context.Context, networkID, container string, force bool) error {
|
|
if c.networkDisconnectFunc != nil {
|
|
return c.networkDisconnectFunc(ctx, networkID, container, force)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *fakeClient) NetworkList(ctx context.Context, options types.NetworkListOptions) ([]types.NetworkResource, error) {
|
|
if c.networkListFunc != nil {
|
|
return c.networkListFunc(ctx, options)
|
|
}
|
|
return []types.NetworkResource{}, nil
|
|
}
|
|
|
|
func (c *fakeClient) NetworkRemove(ctx context.Context, networkID string) error {
|
|
if c.networkRemoveFunc != nil {
|
|
return c.networkRemoveFunc(ctx, networkID)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *fakeClient) NetworkInspectWithRaw(ctx context.Context, network string, options types.NetworkInspectOptions) (types.NetworkResource, []byte, error) {
|
|
return types.NetworkResource{}, nil, nil
|
|
}
|