query-orchestration/internal/test/runner_test.go

package test

import (
	"context"
	"testing"

	"queryorchestration/internal/serviceconfig"

	"github.com/stretchr/testify/assert"
)

func TestCreateRunner(t *testing.T) {
	if testing.Short() {
		t.Skip("Skipping long test in short mode")
	}
	ctx := context.Background()

	ncfg, ncleanup := CreateNetwork(t, ctx)
	defer ncleanup()

	cfg := &serviceconfig.BaseConfig{}
	SetCfgProvider(t, cfg)

	_, qcleanup := CreateAWSContainer(t, ctx, &CreateAWSConfig{
		Network: ncfg,
		Cfg:     cfg,
	})
	defer qcleanup()
	err := cfg.SetQueueClient(ctx)
	assert.NoError(t, err)
	_ = CreateQueue(t, ctx, cfg, QueryRunnerName)

	_, dbcleanup := CreateDB(t, ctx, &CreateDatabaseConfig{
		Network: ncfg,
		Cfg:     cfg,
	})
	defer dbcleanup()

	qccfg := &RunnerConfig{
		Runner:  QueryRunner,
		Cfg:     cfg,
		Network: ncfg,
	}

	c, cleanup := CreateRunner(t, ctx, qccfg)
	assert.NotNil(t, cleanup)
	assert.NotNil(t, c)

	cleanup()
}

func TestGetRunnerEnvFromName(t *testing.T) {
	assert.Equal(t, DocInitRunnerEnv, GetRunnerEnvFromName(DocInitRunnerName))
	assert.Equal(t, DocSyncRunnerEnv, GetRunnerEnvFromName(DocSyncRunnerName))
	assert.Equal(t, DocCleanRunnerEnv, GetRunnerEnvFromName(DocCleanRunnerName))
	assert.Equal(t, DocTextRunnerEnv, GetRunnerEnvFromName(DocTextRunnerName))
	assert.Equal(t, QuerySyncRunnerEnv, GetRunnerEnvFromName(QuerySyncRunnerName))
	assert.Equal(t, QueryRunnerEnv, GetRunnerEnvFromName(QueryRunnerName))
	assert.Equal(t, ClientSyncRunnerEnv, GetRunnerEnvFromName(ClientSyncRunnerName))
	assert.Equal(t, QueryVersionSyncRunnerEnv, GetRunnerEnvFromName(QueryVersionSyncRunnerName))
	assert.Equal(t, DocInitRunnerEnv, GetRunnerEnvFromName(DocInitRunnerName))
}