query-orchestration/internal/test/runner_test.go

package test

import (
	"context"
	"queryorchestration/internal/serviceconfig"
	"testing"

	"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)
	url := CreateQueue(t, ctx, cfg, "myname")

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

	qccfg := &RunnerConfig{
		Name:     QueryRunner,
		Cfg:      cfg,
		Network:  ncfg,
		QueueURL: url,
		Env: map[string]string{
			"QUERY_URL": "/i/am/here",
		},
	}

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

	cleanup()
}