testutil

package
v2.13.0 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Aug 10, 2026 License: GPL-3.0 Imports: 8 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func StartContextRunner 

func StartContextRunner(t *testing.T, runner ContextRunner) func()

StartContextRunner starts a context-driven runner and returns a shutdown function that cancels the context and waits for the runner to exit.

Types

type ContextRunner 

type ContextRunner interface {
	Run(context.Context)
}

ContextRunner is a minimal interface for background services that are driven by a cancellable context.

type MockDAGRunAttempt added in v2.13.0 

type MockDAGRunAttempt struct {
	mock.Mock
	Status *ir.DAGRunStatus
}

MockDAGRunAttempt is a configurable DAG-run attempt for tests.

func (*MockDAGRunAttempt) Abort added in v2.13.0 

func (m *MockDAGRunAttempt) Abort(ctx context.Context) error

func (*MockDAGRunAttempt) Close added in v2.13.0 

func (m *MockDAGRunAttempt) Close(ctx context.Context) error

func (*MockDAGRunAttempt) Hidden added in v2.13.0 

func (m *MockDAGRunAttempt) Hidden() bool

func (*MockDAGRunAttempt) Hide added in v2.13.0 

func (m *MockDAGRunAttempt) Hide(ctx context.Context) error

func (*MockDAGRunAttempt) ID added in v2.13.0 

func (m *MockDAGRunAttempt) ID() string

func (*MockDAGRunAttempt) IsAborting added in v2.13.0 

func (m *MockDAGRunAttempt) IsAborting(ctx context.Context) (bool, error)

func (*MockDAGRunAttempt) Open added in v2.13.0 

func (m *MockDAGRunAttempt) Open(ctx context.Context) error

func (*MockDAGRunAttempt) ReadDAG added in v2.13.0 

func (m *MockDAGRunAttempt) ReadDAG(ctx context.Context) (*ir.DAG, error)

func (*MockDAGRunAttempt) ReadOutputs added in v2.13.0 

func (m *MockDAGRunAttempt) ReadOutputs(ctx context.Context) (*ir.DAGRunOutputs, error)

func (*MockDAGRunAttempt) ReadStatus added in v2.13.0 

func (m *MockDAGRunAttempt) ReadStatus(ctx context.Context) (*ir.DAGRunStatus, error)

func (*MockDAGRunAttempt) ReadStepMessages added in v2.13.0 

func (m *MockDAGRunAttempt) ReadStepMessages(ctx context.Context, stepName string) ([]ir.LLMMessage, error)

func (*MockDAGRunAttempt) SetDAG added in v2.13.0 

func (m *MockDAGRunAttempt) SetDAG(dag *ir.DAG)

func (*MockDAGRunAttempt) WorkDir added in v2.13.0 

func (m *MockDAGRunAttempt) WorkDir() string

func (*MockDAGRunAttempt) Write added in v2.13.0 

func (m *MockDAGRunAttempt) Write(ctx context.Context, status ir.DAGRunStatus) error

func (*MockDAGRunAttempt) WriteOutputs added in v2.13.0 

func (m *MockDAGRunAttempt) WriteOutputs(ctx context.Context, outputs *ir.DAGRunOutputs) error

func (*MockDAGRunAttempt) WriteStepMessages added in v2.13.0 

func (m *MockDAGRunAttempt) WriteStepMessages(ctx context.Context, stepName string, messages []ir.LLMMessage) error

type MockQueueStore added in v2.13.0 

type MockQueueStore struct {
	mock.Mock
}

MockQueueStore is a configurable queue store for tests.

func (*MockQueueStore) All added in v2.13.0 

func (m *MockQueueStore) All(ctx context.Context) ([]queue.QueuedItemData, error)

func (*MockQueueStore) DeleteByItemIDs added in v2.13.0 

func (m *MockQueueStore) DeleteByItemIDs(ctx context.Context, name string, itemIDs []string) (int, error)

func (*MockQueueStore) DequeueByDAGRunID added in v2.13.0 

func (m *MockQueueStore) DequeueByDAGRunID(ctx context.Context, name string, run ir.DAGRunRef) ([]queue.QueuedItemData, error)

func (*MockQueueStore) Enqueue added in v2.13.0 

func (m *MockQueueStore) Enqueue(ctx context.Context, name string, priority queue.QueuePriority, run ir.DAGRunRef) error

func (*MockQueueStore) Len added in v2.13.0 

func (m *MockQueueStore) Len(ctx context.Context, name string) (int, error)

func (*MockQueueStore) List added in v2.13.0 

func (m *MockQueueStore) List(ctx context.Context, name string) ([]queue.QueuedItemData, error)

func (*MockQueueStore) ListByDAGName added in v2.13.0 

func (m *MockQueueStore) ListByDAGName(ctx context.Context, name, dagName string) ([]queue.QueuedItemData, error)

func (*MockQueueStore) ListCursor added in v2.13.0 

func (m *MockQueueStore) ListCursor(ctx context.Context, name, cursor string, limit int) (pagination.CursorResult[queue.QueuedItemData], error)

func (*MockQueueStore) QueueList added in v2.13.0 

func (m *MockQueueStore) QueueList(ctx context.Context) ([]string, error)

func (*MockQueueStore) QueueWatcher added in v2.13.0 

func (m *MockQueueStore) QueueWatcher(ctx context.Context) queue.QueueWatcher

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.