testutil

type FileExpectation ¶

type FileExpectation struct {
	Path     string // relative to workspace root
	Contains string // substring that must appear in file content
	Equals   string // exact match (if non-empty, overrides Contains)
	Absent   bool   // file must NOT exist
}

FileExpectation defines expected file state after a scenario.

type ScriptEntry ¶

type ScriptEntry struct {
	Text       string
	ToolCalls  []ToolCallSpec
	StopReason ai.StopReason
	Error      string
}

ScriptEntry defines one provider response in a scripted conversation. If ToolCalls is set, the response will contain tool call content blocks. If Text is set, it will contain a text content block.

type ScriptedProvider ¶

type ScriptedProvider struct {
	CallLog []ScriptedProviderCall

	// Gates allows blocking specific provider calls until signaled.
	// Key is the 0-based call index. The provider will wait on the channel
	// before returning the response for that call.
	Gates map[int]chan struct{}
	// contains filtered or unexported fields
}

ScriptedProvider is a deterministic fake AI provider that replays a pre-defined sequence of responses. Each call to Stream/StreamSimple returns the next entry in the script. If calls exceed the script length, the last entry is repeated.

Implements ai.APIClient.

func NewScriptedProvider(clientType string, script []ScriptEntry) *ScriptedProvider

func (p *ScriptedProvider) Calls() int

func (p *ScriptedProvider) ClientType() string

func (p *ScriptedProvider) Stream(ctx context.Context, _ ai.ProviderEndpoint, model ai.Model, llmCtx ai.Context, opts ai.StreamOptions) *ai.EventStream

func (p *ScriptedProvider) StreamSimple(ctx context.Context, _ ai.ProviderEndpoint, model ai.Model, llmCtx ai.Context, opts ai.SimpleStreamOptions) *ai.EventStream

type ScriptedProviderCall ¶

type ScriptedProviderCall struct {
	Messages     []ai.Message
	Tools        []ai.Tool
	SystemPrompt string
}

ScriptedProviderCall records what the provider received.

type ToolCallSpec ¶

type ToolCallSpec struct {
	ID        string
	Name      string
	Arguments map[string]any
}

ToolCallSpec defines a tool call within a scripted provider response.

type WorkspaceSnapshot ¶

type WorkspaceSnapshot struct {
	Files map[string]string // relative path -> content
}

WorkspaceSnapshot captures all files in a directory tree.

func SnapshotWorkspace(root string) (*WorkspaceSnapshot, error)

func (s *WorkspaceSnapshot) Diff(other *WorkspaceSnapshot) string