harness

type Expectation struct {
	RunStatus     string              `yaml:"runStatus,omitempty"`
	ErrorContains string              `yaml:"errorContains,omitempty"`
	Tasks         []TaskExpectation   `yaml:"tasks,omitempty"`
	Metrics       []MetricExpectation `yaml:"metrics,omitempty"`
	Lineage       *LineageExpectation `yaml:"lineage,omitempty"`
}

type File struct {
	APIVersion string     `yaml:"apiVersion"`
	Kind       string     `yaml:"kind"`
	Scenarios  []Scenario `yaml:"scenarios"`
}

type ImpactExpectation struct {
	// Dataset is the root dataset name whose downstream impact is queried
	// (e.g. "<job-alias>.<step>.output").
	Dataset string `yaml:"dataset"`
	// Namespace overrides the dataset namespace; defaults to the harness
	// namespace when empty.
	Namespace string `yaml:"namespace,omitempty"`
	// MaxDepth bounds the downstream BFS (0 = unbounded).
	MaxDepth int `yaml:"maxDepth,omitempty"`
	// Downstream lists dataset names that must appear in the impact result.
	Downstream []string `yaml:"downstream"`
}

type LineageExpectation struct {
	TotalEvents *int                `yaml:"totalEvents,omitempty"`
	EventTypes  map[string]int      `yaml:"eventTypes,omitempty"`
	JobNames    []string            `yaml:"jobNames,omitempty"`
	Impact      []ImpactExpectation `yaml:"impact,omitempty"`
}

type MetricExpectation struct {
	Name   string            `yaml:"name"`
	Labels map[string]string `yaml:"labels,omitempty"`
	Value  *float64          `yaml:"value,omitempty"`
	Delta  *float64          `yaml:"delta,omitempty"`
}

type MetricObservation struct {
	Name   string
	Labels map[string]string
	Value  float64
	Delta  float64
}

type ResolvedScenario struct {
	Scenario   Scenario
	SourcePath string
}

type Result struct {
	Scenario           ResolvedScenario
	Run                *localrun.RunResult
	ExecutionError     error
	MetricObservations []MetricObservation
	LineageEvents      []lineage.RunEvent
	Failures           []string
}

type Scenario struct {
	Name        string        `yaml:"name"`
	Path        string        `yaml:"path"`
	Alias       string        `yaml:"alias,omitempty"`
	MaxParallel int           `yaml:"maxParallel,omitempty"`
	TaskTimeout time.Duration `yaml:"taskTimeout,omitempty"`
	RunTimeout  time.Duration `yaml:"runTimeout,omitempty"`
	Expect      Expectation   `yaml:"expect"`
}

type TaskExpectation struct {
	Name                 string            `yaml:"name"`
	Status               string            `yaml:"status,omitempty"`
	Output               map[string]string `yaml:"output,omitempty"`
	LogContains          []string          `yaml:"logContains,omitempty"`
	SchemaViolationCount *int              `yaml:"schemaViolationCount,omitempty"`
	CacheHit             *bool             `yaml:"cacheHit,omitempty"`
	ErrorContains        string            `yaml:"errorContains,omitempty"`
}