Pulse/internal/dockeragent/test_helpers_test.go

package dockeragent

import (
	"bytes"
	"context"
	"encoding/json"
	"errors"
	"io"
	"testing"

	containertypes "github.com/moby/moby/api/types/container"
	"github.com/moby/moby/api/types/image"
	"github.com/moby/moby/api/types/network"
	swarmtypes "github.com/moby/moby/api/types/swarm"
	systemtypes "github.com/moby/moby/api/types/system"
	"github.com/opencontainers/image-spec/specs-go/v1"
)

type fakeDockerClient struct {
	daemonHost                string
	infoFunc                  func(ctx context.Context) (systemtypes.Info, error)
	containerListFunc         func(ctx context.Context, opts containerListOptions) ([]containertypes.Summary, error)
	containerInspectWithRawFn func(ctx context.Context, id string, size bool) (containertypes.InspectResponse, []byte, error)
	containerStatsOneShotFn   func(ctx context.Context, id string) (containerStatsResult, error)
	containerInspectFn        func(ctx context.Context, id string) (containertypes.InspectResponse, error)
	imagePullFn               func(ctx context.Context, ref string, opts imagePullOptions) (io.ReadCloser, error)
	containerStopFn           func(ctx context.Context, id string, opts containerStopOptions) error
	containerRenameFn         func(ctx context.Context, id, newName string) error
	containerCreateFn         func(ctx context.Context, config *containertypes.Config, hostConfig *containertypes.HostConfig, networkingConfig *network.NetworkingConfig, platform *v1.Platform, containerName string) (containertypes.CreateResponse, error)
	networkConnectFn          func(ctx context.Context, netName, containerID string, endpoint *network.EndpointSettings) error
	containerStartFn          func(ctx context.Context, id string, opts containerStartOptions) error
	containerRemoveFn         func(ctx context.Context, id string, opts containerRemoveOptions) error
	serviceListFn             func(ctx context.Context, opts serviceListOptions) ([]swarmtypes.Service, error)
	taskListFn                func(ctx context.Context, opts taskListOptions) ([]swarmtypes.Task, error)
	imageInspectWithRawFn     func(ctx context.Context, imageID string) (image.InspectResponse, []byte, error)
	closeFn                   func() error
}

func (f *fakeDockerClient) Info(ctx context.Context) (systemtypes.Info, error) {
	if f.infoFunc == nil {
		return systemtypes.Info{}, errors.New("unexpected Info call")
	}
	return f.infoFunc(ctx)
}

func (f *fakeDockerClient) DaemonHost() string {
	return f.daemonHost
}

func (f *fakeDockerClient) ContainerList(ctx context.Context, opts containerListOptions) ([]containertypes.Summary, error) {
	if f.containerListFunc == nil {
		return nil, errors.New("unexpected ContainerList call")
	}
	return f.containerListFunc(ctx, opts)
}

func (f *fakeDockerClient) ContainerInspectWithRaw(ctx context.Context, id string, size bool) (containertypes.InspectResponse, []byte, error) {
	if f.containerInspectWithRawFn == nil {
		return containertypes.InspectResponse{}, nil, errors.New("unexpected ContainerInspectWithRaw call")
	}
	return f.containerInspectWithRawFn(ctx, id, size)
}

func (f *fakeDockerClient) ContainerStatsOneShot(ctx context.Context, id string) (containerStatsResult, error) {
	if f.containerStatsOneShotFn == nil {
		return containerStatsResult{}, errors.New("unexpected ContainerStatsOneShot call")
	}
	return f.containerStatsOneShotFn(ctx, id)
}

func (f *fakeDockerClient) ContainerInspect(ctx context.Context, id string) (containertypes.InspectResponse, error) {
	if f.containerInspectFn == nil {
		return containertypes.InspectResponse{}, errors.New("unexpected ContainerInspect call")
	}
	return f.containerInspectFn(ctx, id)
}

func (f *fakeDockerClient) ImagePull(ctx context.Context, ref string, opts imagePullOptions) (io.ReadCloser, error) {
	if f.imagePullFn == nil {
		return nil, errors.New("unexpected ImagePull call")
	}
	return f.imagePullFn(ctx, ref, opts)
}

func (f *fakeDockerClient) ContainerStop(ctx context.Context, id string, opts containerStopOptions) error {
	if f.containerStopFn == nil {
		return errors.New("unexpected ContainerStop call")
	}
	return f.containerStopFn(ctx, id, opts)
}

func (f *fakeDockerClient) ContainerRename(ctx context.Context, id, newName string) error {
	if f.containerRenameFn == nil {
		return errors.New("unexpected ContainerRename call")
	}
	return f.containerRenameFn(ctx, id, newName)
}

func (f *fakeDockerClient) ContainerCreate(ctx context.Context, config *containertypes.Config, hostConfig *containertypes.HostConfig, networkingConfig *network.NetworkingConfig, platform *v1.Platform, containerName string) (containertypes.CreateResponse, error) {
	if f.containerCreateFn == nil {
		return containertypes.CreateResponse{}, errors.New("unexpected ContainerCreate call")
	}
	return f.containerCreateFn(ctx, config, hostConfig, networkingConfig, platform, containerName)
}

func (f *fakeDockerClient) NetworkConnect(ctx context.Context, netName, containerID string, endpoint *network.EndpointSettings) error {
	if f.networkConnectFn == nil {
		return errors.New("unexpected NetworkConnect call")
	}
	return f.networkConnectFn(ctx, netName, containerID, endpoint)
}

func (f *fakeDockerClient) ContainerStart(ctx context.Context, id string, opts containerStartOptions) error {
	if f.containerStartFn == nil {
		return errors.New("unexpected ContainerStart call")
	}
	return f.containerStartFn(ctx, id, opts)
}

func (f *fakeDockerClient) ContainerRemove(ctx context.Context, id string, opts containerRemoveOptions) error {
	if f.containerRemoveFn == nil {
		return errors.New("unexpected ContainerRemove call")
	}
	return f.containerRemoveFn(ctx, id, opts)
}

func (f *fakeDockerClient) ServiceList(ctx context.Context, opts serviceListOptions) ([]swarmtypes.Service, error) {
	if f.serviceListFn == nil {
		return nil, errors.New("unexpected ServiceList call")
	}
	return f.serviceListFn(ctx, opts)
}

func (f *fakeDockerClient) TaskList(ctx context.Context, opts taskListOptions) ([]swarmtypes.Task, error) {
	if f.taskListFn == nil {
		return nil, errors.New("unexpected TaskList call")
	}
	return f.taskListFn(ctx, opts)
}

func (f *fakeDockerClient) ImageInspectWithRaw(ctx context.Context, imageID string) (image.InspectResponse, []byte, error) {
	if f.imageInspectWithRawFn == nil {
		// Return empty response with no RepoDigests by default (simulates locally built image)
		return image.InspectResponse{}, nil, nil
	}
	return f.imageInspectWithRawFn(ctx, imageID)
}

func (f *fakeDockerClient) Close() error {
	if f.closeFn == nil {
		return nil
	}
	return f.closeFn()
}

func statsReader(t *testing.T, stats containertypes.StatsResponse) containerStatsResult {
	t.Helper()

	payload, err := json.Marshal(stats)
	if err != nil {
		t.Fatalf("marshal stats: %v", err)
	}

	return containerStatsResult{
		Body: io.NopCloser(bytes.NewReader(payload)),
	}
}

func swap[T any](t *testing.T, target *T, value T) {
	t.Helper()
	prev := *target
	*target = value
	t.Cleanup(func() {
		*target = prev
	})
}