83 lines
4.1 KiB
Go
83 lines
4.1 KiB
Go
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
|
|
|
package mocks
|
|
|
|
import (
|
|
"context"
|
|
"github.com/docker/docker/api/types"
|
|
"github.com/docker/docker/api/types/container"
|
|
"github.com/docker/docker/api/types/network"
|
|
"github.com/docker/docker/api/types/image"
|
|
ocispec "github.com/opencontainers/image-spec/specs-go/v1"
|
|
"io"
|
|
)
|
|
|
|
type MockContainerClient struct {
|
|
InjectContainerStart func(ctx context.Context, containerID string, options container.StartOptions) error
|
|
InjectContainerCreate func(ctx context.Context, config *container.Config, hostConfig *container.HostConfig, networkingConfig *network.NetworkingConfig, platform *ocispec.Platform, containerName string) (container.CreateResponse, error)
|
|
InjectNetworkCreate func(ctx context.Context, name string, options types.NetworkCreate) (types.NetworkCreateResponse, error)
|
|
InjectContainerList func(context.Context, container.ListOptions) ([]types.Container, error)
|
|
InjectContainerInspect func(context.Context, string) (types.ContainerJSON, error)
|
|
InjectContainerRemove func(context.Context, string, container.RemoveOptions) error
|
|
InjectContainerStop func(context.Context, string, container.StopOptions) error
|
|
InjectContainerWait func(ctx context.Context, containerID string, condition container.WaitCondition) (<-chan container.WaitResponse, <-chan error)
|
|
InjectImagePull func(ctx context.Context, refStr string, options types.ImagePullOptions) (io.ReadCloser, error)
|
|
InjectImageInspectWithRaw func(ctx context.Context, imageID string) (types.ImageInspect, []byte, error)
|
|
InjectImageRemove func(ctx context.Context, imageID string, options types.ImageRemoveOptions) ([]image.DeleteResponse, error)
|
|
InjectClose func() error
|
|
}
|
|
|
|
func (m *MockContainerClient) ContainerWait(ctx context.Context, containerID string, condition container.WaitCondition) (<-chan container.WaitResponse, <-chan error) {
|
|
return m.InjectContainerWait(ctx, containerID, condition)
|
|
}
|
|
|
|
func (m *MockContainerClient) ImageRemove(ctx context.Context, imageID string, options types.ImageRemoveOptions) ([]image.DeleteResponse, error) {
|
|
return m.InjectImageRemove(ctx, imageID, options)
|
|
}
|
|
|
|
func (m *MockContainerClient) ImagePull(ctx context.Context, refStr string, options types.ImagePullOptions) (io.ReadCloser, error) {
|
|
return m.InjectImagePull(ctx, refStr, options)
|
|
}
|
|
|
|
func (m *MockContainerClient) ImageInspectWithRaw(ctx context.Context, imageID string) (types.ImageInspect, []byte, error) {
|
|
return m.InjectImageInspectWithRaw(ctx, imageID)
|
|
}
|
|
|
|
func (m *MockContainerClient) ContainerCreate(ctx context.Context, config *container.Config, hostConfig *container.HostConfig, networkingConfig *network.NetworkingConfig, platform *ocispec.Platform, containerName string) (container.CreateResponse, error) {
|
|
return m.InjectContainerCreate(ctx, config, hostConfig, networkingConfig, platform, containerName)
|
|
}
|
|
|
|
func (m *MockContainerClient) ContainerStart(ctx context.Context, containerID string, options container.StartOptions) error {
|
|
if m.InjectContainerStart == nil {
|
|
return nil
|
|
}
|
|
return m.InjectContainerStart(ctx, containerID, options)
|
|
}
|
|
|
|
func (m *MockContainerClient) ContainerList(ctx context.Context, options container.ListOptions) ([]types.Container, error) {
|
|
return m.InjectContainerList(ctx, options)
|
|
}
|
|
|
|
func (m *MockContainerClient) ContainerInspect(ctx context.Context, containerID string) (types.ContainerJSON, error) {
|
|
return m.InjectContainerInspect(ctx, containerID)
|
|
}
|
|
|
|
func (m *MockContainerClient) ContainerRemove(ctx context.Context, containerID string, options container.RemoveOptions) error {
|
|
return m.InjectContainerRemove(ctx, containerID, options)
|
|
}
|
|
|
|
func (m *MockContainerClient) ContainerStop(ctx context.Context, containerID string, options container.StopOptions) error {
|
|
return m.InjectContainerStop(ctx, containerID, options)
|
|
}
|
|
|
|
func (m *MockContainerClient) Close() error {
|
|
if m.InjectClose == nil {
|
|
return nil
|
|
}
|
|
return m.InjectClose()
|
|
}
|
|
|
|
func (m *MockContainerClient) NetworkCreate(ctx context.Context, name string, options types.NetworkCreate) (types.NetworkCreateResponse, error) {
|
|
return m.InjectNetworkCreate(ctx, name, options)
|
|
}
|