test_suites_pipeline.go

package executor

import (
	"context"

	"github.com/kubeshop/tracetest/server/executor/testrunner"
	"github.com/kubeshop/tracetest/server/pkg/pipeline"
	"github.com/kubeshop/tracetest/server/test"
	"github.com/kubeshop/tracetest/server/testsuite"
	"github.com/kubeshop/tracetest/server/variableset"
)

type TestSuitesPipeline struct {
	*pipeline.Pipeline[Job]
	runs testSuiteRunRepo
}

type testSuiteRunRepo interface {
	CreateRun(context.Context, testsuite.TestSuiteRun) (testsuite.TestSuiteRun, error)
}

func NewTestSuitePipeline(
	pipeline *pipeline.Pipeline[Job],
	runs testSuiteRunRepo,
) *TestSuitesPipeline {
	return &TestSuitesPipeline{
		Pipeline: pipeline,
		runs:     runs,
	}
}

func (p *TestSuitesPipeline) Run(ctx context.Context, tran testsuite.TestSuite, metadata test.RunMetadata, variableSet variableset.VariableSet, requiredGates *[]testrunner.RequiredGate) testsuite.TestSuiteRun {
	tranRun := tran.NewRun()
	tranRun.Metadata = metadata
	tranRun.VariableSet = variableSet
	tranRun.RequiredGates = requiredGates

	tranRun, _ = p.runs.CreateRun(ctx, tranRun)

	job := NewJob()
	job.TestSuite = tran
	job.TestSuiteRun = tranRun

	p.Pipeline.Begin(ctx, job)

	return tranRun
}