query-orchestration/internal/test/runner.go

package test

import (
	"context"
	"testing"

	"queryorchestration/internal/serviceconfig/queue/clientsync"
	"queryorchestration/internal/serviceconfig/queue/documentclean"
	"queryorchestration/internal/serviceconfig/queue/documentinit"
	"queryorchestration/internal/serviceconfig/queue/documentsync"
	"queryorchestration/internal/serviceconfig/queue/documenttext"
	"queryorchestration/internal/serviceconfig/queue/storeevent"
)

type RunnerName string

const (
	StoreEventRunnerName RunnerName = "storeEventRunner"
	DocInitRunnerName    RunnerName = "docInitRunner"
	DocSyncRunnerName    RunnerName = "docSyncRunner"
	DocCleanRunnerName   RunnerName = "docCleanRunner"
	DocTextRunnerName    RunnerName = "docTextRunner"
	ClientSyncRunnerName RunnerName = "clientSyncRunner"
)

type RunnerEnv string

const (
	StoreEventRunnerEnv RunnerEnv = storeevent.EnvName
	DocInitRunnerEnv    RunnerEnv = documentinit.EnvName
	DocSyncRunnerEnv    RunnerEnv = documentsync.EnvName
	DocCleanRunnerEnv   RunnerEnv = documentclean.EnvName
	DocTextRunnerEnv    RunnerEnv = documenttext.EnvName
	ClientSyncRunnerEnv RunnerEnv = clientsync.EnvName
)

type Runner struct {
	Name             RunnerName
	DownstreamQueues []RunnerName
}

func GetRunnerEnvFromName(name RunnerName) RunnerEnv {
	switch name {
	case StoreEventRunnerName:
		return StoreEventRunnerEnv
	case DocInitRunnerName:
		return DocInitRunnerEnv
	case DocSyncRunnerName:
		return DocSyncRunnerEnv
	case DocCleanRunnerName:
		return DocCleanRunnerEnv
	case DocTextRunnerName:
		return DocTextRunnerEnv
	case ClientSyncRunnerName:
		return ClientSyncRunnerEnv
	}
	return DocInitRunnerEnv
}

type RunnerConfig struct {
	Runner   Runner
	MockHTTP string
}

func CreateRunner(t testing.TB, ctx context.Context, cfg ContainerConfigProvider, network string, config *RunnerConfig) (*Container, func()) {
	env := map[string]string{}
	url := GetQueueURL(t, cfg, config.Runner.Name)
	env["QUEUE_URL"] = url

	c, cleanup := createContainer(t, ctx, network, &containerConfig{
		Name:             string(config.Runner.Name),
		DownstreamQueues: config.Runner.DownstreamQueues,
		Cfg:              cfg,
		Env:              env,
		MockHTTP:         config.MockHTTP,
		WaitForMsg:       "Listening to queue",
	})

	return &Container{
		Container: c,
		URI:       url,
	}, cleanup
}

var StoreEventRunner = Runner{
	Name:             StoreEventRunnerName,
	DownstreamQueues: []RunnerName{DocInitRunnerName},
}

var DocInitRunner = Runner{
	Name:             DocInitRunnerName,
	DownstreamQueues: []RunnerName{DocSyncRunnerName},
}

var DocSyncRunner = Runner{
	Name:             DocSyncRunnerName,
	DownstreamQueues: []RunnerName{DocCleanRunnerName},
}

var DocCleanRunner = Runner{
	Name:             DocCleanRunnerName,
	DownstreamQueues: []RunnerName{DocTextRunnerName},
}

// DocTextRunner is the terminal runner in the main document processing pipeline.
// Query processing has been removed - pipeline ends after text extraction.
var DocTextRunner = Runner{
	Name:             DocTextRunnerName,
	DownstreamQueues: []RunnerName{},
}

var ClientSyncRunner = Runner{
	Name:             ClientSyncRunnerName,
	DownstreamQueues: []RunnerName{DocSyncRunnerName},
}

var runners = []Runner{
	StoreEventRunner,
	DocInitRunner,
	DocSyncRunner,
	DocCleanRunner,
	DocTextRunner,
	ClientSyncRunner,
}