refactor: singleton database instance for tests

tapservicego/internal/tapsync/alerce_suite_test.go

@@ -0,0 +1,108 @@
+package tapsync
+
+import (
+	"ataps/internal/testhelpers"
+	"context"
+	"database/sql"
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/suite"
+	"github.com/testcontainers/testcontainers-go/modules/postgres"
+)
+
+type AlerceTestSuite struct {
+	suite.Suite
+	DB            *sql.DB
+	connUrl       string
+	PsqlContainer *postgres.PostgresContainer
+	ctx           context.Context
+	Service       *TapSyncService
+}
+
+func (suite *AlerceTestSuite) SetupSuite() {
+	if os.Getenv("ENV") == "DEV" || os.Getenv("ENV") == "" {
+		suite.InitializeLocalDB()
+	} else if os.Getenv("ENV") == "CI" {
+		suite.InitializeDaggerDB()
+	} else {
+		suite.T().Fatal("Unknown environment")
+	}
+	suite.Service = NewTapSyncService()
+}
+
+func (suite *AlerceTestSuite) TearDownSuite() {
+	if os.Getenv("ENV") == "DEV" || os.Getenv("ENV") == "" {
+		testhelpers.CleanUpContainer(suite.ctx, suite.PsqlContainer)
+	}
+	suite.DB.Close()
+}
+
+func (suite *AlerceTestSuite) SetupTest() {
+	suite.T().Log("Populating database")
+	err := testhelpers.PopulateALeRCEDB(suite.DB)
+	if err != nil {
+		suite.T().Fatal(err)
+	}
+}
+
+func (suite *AlerceTestSuite) TearDownTest() {
+	err := testhelpers.ClearALeRCEDB(suite.DB)
+	if err != nil {
+		suite.T().Fatal(err)
+	}
+}
+
+func TestAlerceTestSuite(t *testing.T) {
+	suite.Run(t, new(AlerceTestSuite))
+}
+
+func (suite *AlerceTestSuite) InitializeLocalDB() {
+	suite.T().Log("Using local database")
+	var err error
+	suite.ctx = context.Background()
+	suite.PsqlContainer, err = testhelpers.CreatePostgresContainer(suite.ctx, "alerce")
+	if err != nil {
+		suite.T().Fatal(err)
+	}
+	var connStr string
+	connStr, err = suite.PsqlContainer.ConnectionString(suite.ctx)
+	if err != nil {
+		suite.T().Fatal(err)
+	}
+	suite.connUrl = connStr
+	os.Setenv("DATABASE_URL", connStr)
+	db, err := GetDB(connStr)
+	if err != nil {
+		suite.T().Log("Could not connect")
+		suite.T().Fatal(err)
+	}
+	suite.DB = db
+}
+
+func (suite *AlerceTestSuite) InitializeDaggerDB() {
+	suite.T().Log("Using Dagger database")
+	connUrl := "host=db user=testuser password=testpassword port=5432"
+	// create alerce database
+	db, err := GetDB(connUrl)
+	if err != nil {
+		suite.T().Log("Could not connect")
+		suite.T().Fatal(err)
+	}
+	_, err = db.Exec("CREATE DATABASE alerce")
+	if err != nil {
+		suite.T().Log("Could not create alerce database")
+		suite.T().Fatal(err)
+	}
+	db.Close()
+	// connect to alerce database
+	connUrl = connUrl + " dbname=alerce"
+	suite.connUrl = connUrl
+	os.Setenv("DATABASE_URL", connUrl)
+	db, err = GetDB(connUrl)
+	if err != nil {
+		suite.T().Log("Could not connect")
+		suite.T().Fatal(err)
+	}
+	suite.DB = db
+}