query-orchestration/internal/test/queuecontainer_test.go

package test

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

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

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

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

	t.Setenv("BASE_PATH", path.Join(os.Getenv("PWD"), "../.."))
	cfg := &serviceconfig.BaseConfig{}
	dbcfg, dbcleanup := CreateDB(t, ctx, &CreateDatabaseConfig{
		Network:       ncfg,
		Cfg:           cfg,
		RunMigrations: true,
	})
	defer dbcleanup()

	qcfg, qcleanup := CreateQueue(t, ctx, &CreateQueueConfig{
		Network: ncfg,
	})
	defer qcleanup()

	extcfg := &queueConfig{
		URL:         qcfg.URL,
		Region:      qcfg.External.Region,
		Endpoint:    qcfg.External.Endpoint,
		Credentials: qcfg.External.Credentials,
	}
	qccfg := &QueueContainerConfig{
		ServiceName: "queryRunner",
		DB:          dbcfg.External,
		Network:     ncfg,
		Queue:       extcfg,
	}

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

	cleanup()
}

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

	conn, cleanup := CreateQueueWithDependencies(t, ctx, "queryRunner")

	assert.NotNil(t, conn)
	assert.NotNil(t, cleanup)

	cleanup()
}