wait_test.go

package wait

import (
	"context"
	"errors"
	"io"

	"github.com/docker/docker/api/types"
	"github.com/docker/go-connections/nat"

	tcexec "github.com/testcontainers/testcontainers-go/exec"
)

var ErrPortNotFound = errors.New("port not found")

type MockStrategyTarget struct {
	HostImpl       func(context.Context) (string, error)
	PortsImpl      func(context.Context) (nat.PortMap, error)
	MappedPortImpl func(context.Context, nat.Port) (nat.Port, error)
	LogsImpl       func(context.Context) (io.ReadCloser, error)
	ExecImpl       func(context.Context, []string, ...tcexec.ProcessOption) (int, io.Reader, error)
	StateImpl      func(context.Context) (*types.ContainerState, error)
}

func (st MockStrategyTarget) Host(ctx context.Context) (string, error) {
	return st.HostImpl(ctx)
}

func (st MockStrategyTarget) Ports(ctx context.Context) (nat.PortMap, error) {
	return st.PortsImpl(ctx)
}

func (st MockStrategyTarget) MappedPort(ctx context.Context, port nat.Port) (nat.Port, error) {
	return st.MappedPortImpl(ctx, port)
}

func (st MockStrategyTarget) Logs(ctx context.Context) (io.ReadCloser, error) {
	return st.LogsImpl(ctx)
}

func (st MockStrategyTarget) Exec(ctx context.Context, cmd []string, options ...tcexec.ProcessOption) (int, io.Reader, error) {
	return st.ExecImpl(ctx, cmd, options...)
}

func (st MockStrategyTarget) State(ctx context.Context) (*types.ContainerState, error) {
	return st.StateImpl(ctx)
}