testrunner

type AssertionError struct {
	Field    string
	Expected any
	Actual   any
	Message  string
}

type BuildConfig struct {
	Package string   `yaml:"package"`
	Timeout Duration `yaml:"timeout"`
}

type CC2Config struct {
	Timeout  Duration `yaml:"timeout"`   // Default timeout per test (default: 2m)
	CacheDir string   `yaml:"cache_dir"` // Shared OCI cache
	Image    string   `yaml:"image"`     // Default image (alpine:latest)
}

type CC2Expectation struct {
	ExitCode       int                   `yaml:"exit_code"`
	StdoutEquals   string                `yaml:"stdout_equals"`
	StdoutContains string                `yaml:"stdout_contains"`
	StderrEquals   string                `yaml:"stderr_equals"`
	StderrContains string                `yaml:"stderr_contains"`
	Files          map[string]FileExpect `yaml:"files"`
}

type CC2Fixtures struct {
	Files map[string]string `yaml:"files"` // path -> content
	Dirs  []string          `yaml:"dirs"`
}

type CC2Flags struct {
	Memory     uint64   `yaml:"memory"`
	CPUs       int      `yaml:"cpus"`
	Timeout    Duration `yaml:"timeout"`
	Workdir    string   `yaml:"workdir"`
	User       string   `yaml:"user"`
	Dmesg      bool     `yaml:"dmesg"`
	Exec       bool     `yaml:"exec"`
	Packetdump string   `yaml:"packetdump"`
	GPU        bool     `yaml:"gpu"`
	Arch       string   `yaml:"arch"`
	CacheDir   string   `yaml:"cache_dir"`
	Build      string   `yaml:"build"`
	Env        []string `yaml:"env"`
	Mounts     []string `yaml:"mounts"`
}

type CC2TestCase struct {
	Name   string `yaml:"name"`
	Skip   bool   `yaml:"skip"`
	SkipCI bool   `yaml:"skip_ci"`

	// Source (one of - uses cc2.Image default if none specified)
	Image      string `yaml:"image"`
	Dockerfile string `yaml:"dockerfile"`
	Bundle     string `yaml:"bundle"`

	// cc2 flags
	Flags CC2Flags `yaml:"flags"`

	// Command and input
	Command []string `yaml:"command"`
	Stdin   string   `yaml:"stdin"`

	// Fixtures
	Fixtures CC2Fixtures `yaml:"fixtures"`

	// Expected results
	Expect CC2Expectation `yaml:"expect"`
}

type CLIConfig struct {
	// Timeout for each CLI test execution.
	Timeout Duration `yaml:"timeout"`
	// Env sets environment variables for all CLI tests.
	Env map[string]string `yaml:"env"`
}

type CLIExpectation struct {
	ExitCode       int    `yaml:"exit_code"`
	StdoutContains string `yaml:"stdout_contains"`
	StdoutEquals   string `yaml:"stdout_equals"`
	StderrContains string `yaml:"stderr_contains"`
	StderrEquals   string `yaml:"stderr_equals"`
}

type CLITestCase struct {
	Name   string            `yaml:"name"`
	Args   []string          `yaml:"args"`
	Stdin  string            `yaml:"stdin"`
	Env    map[string]string `yaml:"env"`
	Expect CLIExpectation    `yaml:"expect"`
}

type Duration time.Duration

type ExampleResult struct {
	Name     string
	Tests    []TestResult
	Total    int
	Passed   int
	Failed   int
	Error    string
	Duration time.Duration
}

type Expectation struct {
	Status       int               `yaml:"status"`
	Headers      map[string]string `yaml:"headers"`
	BodyContains string            `yaml:"body_contains"`
	BodyEquals   string            `yaml:"body_equals"`
	JSON         map[string]any    `yaml:"json"`
}

type FileExpect struct {
	Exists   bool   `yaml:"exists"`
	Contains string `yaml:"contains"`
}

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

type Response struct {
	StatusCode int
	Body       []byte
	Headers    http.Header
}

type Results struct {
	Examples []ExampleResult
	Total    int
	Passed   int
	Failed   int
	Duration time.Duration
}

type Runner struct {
	Verbose   bool
	KeepAlive bool
	Parallel  int
	Output    *Output
	// CC2Binary is the path to the cc2 binary for CC2 tests.
	// If empty, CC2 tests will attempt to build cc2 automatically.
	CC2Binary string
}

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

type ServerConfig struct {
	Port            int               `yaml:"port"`
	StartupTimeout  Duration          `yaml:"startup_timeout"`
	ShutdownTimeout Duration          `yaml:"shutdown_timeout"`
	Env             map[string]string `yaml:"env"`
}

type TestCase struct {
	Name       string            `yaml:"name"`
	Method     string            `yaml:"method"`
	Path       string            `yaml:"path"`
	Headers    map[string]string `yaml:"headers"`
	Body       any               `yaml:"body"`
	BodyRaw    string            `yaml:"body_raw"`
	BodyBase64 string            `yaml:"body_base64"`
	Expect     Expectation       `yaml:"expect"`
}

type TestResult struct {
	Name     string
	Passed   bool
	Error    string
	Duration time.Duration
	// Details contains additional context for debugging failures.
	Details *TestResultDetails
}

type TestResultDetails struct {
	// For CLI tests
	Stdout   string
	Stderr   string
	ExitCode int
	Args     []string
	// For HTTP tests
	Method     string
	Path       string
	StatusCode int
	Body       string
}

type TestSpec struct {
	Name        string       `yaml:"name"`
	Description string       `yaml:"description"`
	Build       BuildConfig  `yaml:"build"`
	Server      ServerConfig `yaml:"server"`
	Tests       []TestCase   `yaml:"tests"`
	// CLI specifies this is a CLI test (no server).
	CLI      *CLIConfig    `yaml:"cli,omitempty"`
	CLITests []CLITestCase `yaml:"cli_tests,omitempty"`
	// CC2 specifies this is a cc2 test (container VM).
	CC2      *CC2Config    `yaml:"cc2,omitempty"`
	CC2Tests []CC2TestCase `yaml:"cc2_tests,omitempty"`
}