testutils

type ExecSQLCall struct {
	DatabaseName string
	SQL          string
}

type MockDeployEngine struct {
	CreateBlueprintValidationResult *manage.BlueprintValidation
	CreateBlueprintValidationErr    error

	GetBlueprintValidationResult *manage.BlueprintValidation
	GetBlueprintValidationErr    error

	// StreamBlueprintValidationEventsFn allows full control over the streaming behaviour.
	// If set, it is called directly; otherwise the default implementation
	// sends StubValidationEvents to the streamTo channel and then closes it.
	StreamBlueprintValidationEventsFn func(ctx context.Context, validationID string, streamTo chan<- types.BlueprintValidationEvent, errChan chan<- error) error
	StreamBlueprintValidationErr      error
	StubValidationEvents              []types.BlueprintValidationEvent

	CreateChangesetResult *manage.Changeset
	CreateChangesetErr    error

	GetChangesetResult *manage.Changeset
	GetChangesetErr    error

	StreamChangeStagingEventsFn func(ctx context.Context, changesetID string, streamTo chan<- types.ChangeStagingEvent, errChan chan<- error) error
	StreamChangeStagingErr      error

	CreateBlueprintInstanceResult *state.InstanceState
	CreateBlueprintInstanceErr    error

	UpdateBlueprintInstanceResult *state.InstanceState
	UpdateBlueprintInstanceErr    error

	GetBlueprintInstanceResult *state.InstanceState
	GetBlueprintInstanceErr    error

	GetBlueprintInstanceExportsResult map[string]*state.ExportState
	GetBlueprintInstanceExportsErr    error

	DestroyBlueprintInstanceResult *state.InstanceState
	DestroyBlueprintInstanceErr    error

	StreamBlueprintInstanceEventsFn func(ctx context.Context, instanceID string, streamTo chan<- types.BlueprintInstanceEvent, errChan chan<- error) error
	StreamBlueprintInstanceErr      error
}

type MockDockerManager struct {
	CheckAvailabilityFn func(ctx context.Context) error
	EnsureImageFn       func(ctx context.Context, image string, progress chan<- docker.ImagePullProgress) error
	CreateAndStartFn    func(ctx context.Context, config *docker.ContainerConfig) (string, error)
	StreamLogsFn        func(ctx context.Context, containerID string) (io.ReadCloser, error)
	StreamLogsOptsFn    func(ctx context.Context, containerID string, opts docker.LogStreamOptions) (io.ReadCloser, error)
	RestartContainerFn  func(ctx context.Context, containerID string) error
	StopFn              func(ctx context.Context, containerID string) error
	IsRunningFn         func(ctx context.Context, containerID string) (bool, error)
	CleanupStaleFn      func(ctx context.Context, containerName string) error

	// Call records for assertions.
	StopCalls         []string
	CleanupStaleCalls []string
}

type MockNoSQLSeeder struct {
	PutItemCalls []PutItemCall
	PutItemErr   error
}

type MockSQLSeeder struct {
	ExecSQLCalls []ExecSQLCall
	ExecSQLErr   error
}

type MockStorageUploader struct {
	UploadCalls []UploadCall
	UploadErr   error
}

type PutItemCall struct {
	TableName string
	ItemJSON  []byte
}

type UploadCall struct {
	BucketName string
	ObjectKey  string
	Data       []byte
}