query-orchestration/internal/test/api_test.go

package test_test

import (
	"context"
	"os"
	"path"
	"queryorchestration/internal/database"
	"queryorchestration/internal/test"
	"testing"

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

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

	ncfg, ncleanup := test.CreateNetwork(t, ctx)
	defer ncleanup()
	dbcfg, dbcleanup := test.CreateDB(t, ctx, &test.CreateDatabaseConfig{
		Network: ncfg,
		Migrations: &database.MigrationConfig{
			BasePath: path.Join(os.Getenv("PWD"), "../.."),
		},
	})
	defer dbcleanup()

	cfg := &test.APIContainerConfig{
		ServiceName: "queryService",
		DB:          dbcfg.External,
		Network:     ncfg,
	}

	conn, cleanup := test.CreateAPIContainer(t, ctx, cfg)
	assert.NotNil(t, conn)
	assert.NotNil(t, cleanup)

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

	conn, cleanup := test.CreateAPIWithDependencies(t, ctx, "queryService")

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

	cleanup()
}