test_suite_pipeline.go

package app

import (
	"github.com/kubeshop/tracetest/server/executor"
	"github.com/kubeshop/tracetest/server/pkg/pipeline"
	"github.com/kubeshop/tracetest/server/subscription"
	"github.com/kubeshop/tracetest/server/testsuite"
)

func buildTestSuitePipeline(
	tranRepo *testsuite.Repository,
	runRepo *testsuite.RunRepository,
	testRunner *executor.TestPipeline,
	subscriptionManager *subscription.Manager,
) *executor.TestSuitesPipeline {
	tranRunner := executor.NewTestSuiteRunner(testRunner, runRepo, subscriptionManager)
	queueBuilder := executor.NewQueueConfigurer().
		WithTestSuiteGetter(tranRepo).
		WithTestSuiteRunGetter(runRepo)

	pipeline := pipeline.New(queueBuilder,
		pipeline.Step[executor.Job]{Processor: tranRunner, Driver: pipeline.NewInMemoryQueueDriver[executor.Job]("testSuiteRunner")},
	)

	return executor.NewTestSuitePipeline(
		pipeline,
		runRepo,
	)
}