query-orchestration/internal/test/runner_test.go

package test

import (
	"context"
	"testing"

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

func TestCreateRunner(t *testing.T) {
	t.Parallel()
	if testing.Short() {
		t.SkipNow()
	}
	ctx := context.Background()

	cfg := &TestConfig{}

	net := GetNetwork(t, ctx)

	a := CreateAWSContainer(t, ctx, cfg, net)
	SetQueueClient(t, ctx, cfg, a.ExternalEndpoint)
	_ = CreateQueue(t, ctx, cfg, QueryRunnerName)
	_ = CreateDB(t, ctx, cfg, net, &CreateDatabaseConfig{NoMigrations: true})

	qccfg := &RunnerConfig{
		Runner: QueryRunner,
	}

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

	cleanup()
}

func TestGetRunnerEnvFromName(t *testing.T) {
	assert.Equal(t, StoreEventRunnerEnv, GetRunnerEnvFromName(StoreEventRunnerName))
	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))
}