framework

type Defaults struct {
	Client     string            `yaml:"client"`      // e.g., "jsonrpc.mcp_assistant" (hint to pick generated client)
	Headers    map[string]string `yaml:"headers"`     // default headers for all steps
	ClientMode string            `yaml:"client_mode"` // http | cli (optional)
}

type Expect struct {
	Status string         `yaml:"status"` // success | error | no_response
	Error  *ExpectedError `yaml:"error"`
	Result map[string]any `yaml:"result"`
}

type ExpectRetry struct {
	PromptContains string   `yaml:"prompt_contains"`
	Contains       []string `yaml:"contains"`
}

type ExpectedError struct {
	Code    int    `yaml:"code"`
	Message string `yaml:"message"`
}

type Pre struct {
	AutoInitialize *bool `yaml:"auto_initialize"` // default true
}

type Runner struct {
	// contains filtered or unexported fields
}

type Scenario struct {
	Name     string    `yaml:"name"`
	Defaults *Defaults `yaml:"defaults"`
	Pre      *Pre      `yaml:"pre"`
	Steps    []Step    `yaml:"steps"`
}

type Step struct {
	Name         string            `yaml:"name"`
	Client       string            `yaml:"client"`       // overrides defaults.client
	Op           string            `yaml:"op"`           // generated endpoint method name, e.g., "ToolsCall"
	Input        map[string]any    `yaml:"input"`        // maps to payload fields
	Headers      map[string]string `yaml:"headers"`      // per-step headers (e.g., Accept)
	Notification bool              `yaml:"notification"` // send as JSON-RPC notification (no id)
	Expect       *Expect           `yaml:"expect"`
	StreamExpect *StreamExpect     `yaml:"stream_expect"`
	ExpectRetry  *ExpectRetry      `yaml:"expect_retry"` // generated client retry expectation
}

type StreamEventExp struct {
	Event string         `yaml:"event"`
	Data  map[string]any `yaml:"data"`
}

type StreamExpect struct {
	MinEvents int              `yaml:"min_events"`
	TimeoutMS int              `yaml:"timeout_ms"`
	Events    []StreamEventExp `yaml:"events"`
}