49 lines
2.1 KiB
Go
49 lines
2.1 KiB
Go
// Copyright 2024 Matthew Rich <matthewrich.conf@gmail.com>. All rights reserved.
|
|
package mocks
|
|
|
|
import (
|
|
"context"
|
|
"github.com/docker/docker/api/types/container"
|
|
"github.com/docker/docker/api/types/network"
|
|
ocispec "github.com/opencontainers/image-spec/specs-go/v1"
|
|
"github.com/docker/docker/api/types"
|
|
)
|
|
|
|
type MockContainerClient struct {
|
|
InjectContainerCreate func(ctx context.Context, config *container.Config, hostConfig *container.HostConfig, networkingConfig *network.NetworkingConfig, platform *ocispec.Platform, containerName string) (container.CreateResponse, error)
|
|
InjectContainerList func(context.Context, types.ContainerListOptions) ([]types.Container, error)
|
|
InjectContainerInspect func(context.Context, string) (types.ContainerJSON, error)
|
|
InjectContainerRemove func(context.Context, string, container.RemoveOptions) error
|
|
InjectClose func() error
|
|
}
|
|
|
|
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) (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 types.ContainerListOptions) ([]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) Close() error {
|
|
if m.InjectClose == nil {
|
|
return nil
|
|
}
|
|
return m.InjectClose()
|
|
}
|