Add database tests

codecov.yml

@@ -0,0 +1,8 @@
+ignore:
+  - "main.go"
+  - "commands/"
+  - "config/"
+  - "jobs/"
+  - "model/"
+  - "controller/"
+  - "service/"

database/db.go

@@ -15,17 +15,17 @@ var DB *gorm.DB
 
 var dbRetries = 0
 
-func InitializeDB() {
+func InitializeDB() error {
 	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=UTC", config.DatabaseUser, config.DatabasePassword, config.DatabaseHost, config.DatabasePort, config.DatabaseName)
 	db, err := gorm.Open(singlestore.Open(dsn), &gorm.Config{})
 	if err != nil {
-		if dbRetries < 15 {
+		if dbRetries < 5 {
 			dbRetries++
 			utils.SugarLogger.Errorln("failed to connect database, retrying in 5s... ")
 			time.Sleep(time.Second * 5)
 			InitializeDB()
 		} else {
-			utils.SugarLogger.Fatalln("failed to connect database after 15 attempts, terminating program...")
+			return fmt.Errorf("failed to connect database after 5 attempts")
 		}
 	} else {
 		utils.SugarLogger.Infoln("Connected to database")
@@ -44,4 +44,5 @@ func InitializeDB() {
 		utils.SugarLogger.Infoln("AutoMigration complete")
 		DB = db
 	}
+	return nil
 }

database/db_test.go

@@ -0,0 +1,79 @@
+package database
+
+import (
+	"context"
+	"log"
+	"sentinel/config"
+	"sentinel/utils"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/testcontainers/testcontainers-go/modules/mysql"
+	"go.uber.org/zap"
+	"gorm.io/gorm"
+)
+
+func TestInitializeDB(t *testing.T) {
+	// Initialize logger
+	logger, _ := zap.NewDevelopment()
+	utils.SugarLogger = logger.Sugar()
+
+	// Start MySQL container
+	ctx := context.Background()
+
+	mysqlContainer, err := mysql.Run(ctx,
+		"mysql:8.0.36",
+		mysql.WithDatabase(config.DatabaseName),
+		mysql.WithUsername(config.DatabaseUser),
+		mysql.WithPassword(config.DatabasePassword),
+	)
+	if err != nil {
+		log.Fatalf("failed to start container: %s", err)
+	}
+
+	port, err := mysqlContainer.MappedPort(ctx, "3306/tcp")
+	if err != nil {
+		log.Fatalf("failed to get container port: %s", err)
+	}
+
+	host, err := mysqlContainer.Host(ctx)
+	if err != nil {
+		log.Fatalf("failed to get container host: %s", err)
+	}
+
+	config.DatabaseHost = host
+	config.DatabasePort = port.Port()
+
+	// Clean up the container
+	defer func() {
+		if err := mysqlContainer.Terminate(ctx); err != nil {
+			log.Fatalf("failed to terminate container: %s", err)
+		}
+	}()
+
+	// Test successful connection
+	t.Run("Successful Connection", func(t *testing.T) {
+		InitializeDB()
+		assert.NotNil(t, DB, "DB should not be nil after successful connection")
+		assert.IsType(t, &gorm.DB{}, DB, "DB should be of type *gorm.DB")
+	})
+
+	// Test retry mechanism
+	t.Run("Retry Mechanism", func(t *testing.T) {
+		config.DatabaseHost = "non-existent-host"
+
+		done := make(chan bool)
+		go func() {
+			InitializeDB()
+			done <- true
+		}()
+
+		select {
+		case <-done:
+			t.Error("InitializeDB should not have succeeded with non-existent host")
+		case <-time.After(10 * time.Second):
+			// Test passes if it times out after 10 seconds
+		}
+	})
+}

go.mod

@@ -10,6 +10,8 @@ require (
 	github.com/google/uuid v1.6.0
 	github.com/lithammer/fuzzysearch v1.1.8
 	github.com/singlestore-labs/gorm-singlestore v1.2.0
+	github.com/stretchr/testify v1.9.0
+	github.com/testcontainers/testcontainers-go/modules/mysql v0.33.0
 	go.uber.org/zap v1.27.0
 	golang.org/x/oauth2 v0.21.0
 	gorm.io/gorm v1.25.7
@@ -19,26 +21,62 @@ require (
 	cloud.google.com/go/auth v0.6.1 // indirect
 	cloud.google.com/go/auth/oauth2adapt v0.2.2 // indirect
 	cloud.google.com/go/compute/metadata v0.3.0 // indirect
+	dario.cat/mergo v1.0.0 // indirect
+	github.com/Azure/go-ansiterm v0.0.0-20210617225240-d185dfc1b5a1 // indirect
+	github.com/Microsoft/go-winio v0.6.2 // indirect
 	github.com/bytedance/sonic/loader v0.1.1 // indirect
+	github.com/cenkalti/backoff/v4 v4.2.1 // indirect
 	github.com/cloudwego/base64x v0.1.4 // indirect
 	github.com/cloudwego/iasm v0.2.0 // indirect
+	github.com/containerd/containerd v1.7.18 // indirect
+	github.com/containerd/log v0.1.0 // indirect
+	github.com/containerd/platforms v0.2.1 // indirect
+	github.com/cpuguy83/dockercfg v0.3.1 // indirect
+	github.com/davecgh/go-spew v1.1.1 // indirect
+	github.com/distribution/reference v0.6.0 // indirect
+	github.com/docker/docker v27.1.1+incompatible // indirect
+	github.com/docker/go-connections v0.5.0 // indirect
+	github.com/docker/go-units v0.5.0 // indirect
 	github.com/felixge/httpsnoop v1.0.4 // indirect
 	github.com/go-logr/logr v1.4.1 // indirect
 	github.com/go-logr/stdr v1.2.2 // indirect
-	github.com/go-sql-driver/mysql v1.7.0 // indirect
+	github.com/go-ole/go-ole v1.2.6 // indirect
+	github.com/go-sql-driver/mysql v1.7.1 // indirect
+	github.com/gogo/protobuf v1.3.2 // indirect
 	github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect
 	github.com/golang/protobuf v1.5.4 // indirect
 	github.com/google/s2a-go v0.1.7 // indirect
 	github.com/googleapis/enterprise-certificate-proxy v0.3.2 // indirect
 	github.com/googleapis/gax-go/v2 v2.12.5 // indirect
+	github.com/klauspost/compress v1.17.4 // indirect
+	github.com/lufia/plan9stats v0.0.0-20211012122336-39d0f177ccd0 // indirect
+	github.com/magiconair/properties v1.8.7 // indirect
 	github.com/mattn/go-colorable v0.1.13 // indirect
+	github.com/moby/docker-image-spec v1.3.1 // indirect
+	github.com/moby/patternmatcher v0.6.0 // indirect
+	github.com/moby/sys/sequential v0.5.0 // indirect
+	github.com/moby/sys/user v0.1.0 // indirect
+	github.com/moby/term v0.5.0 // indirect
+	github.com/morikuni/aec v1.0.0 // indirect
+	github.com/opencontainers/go-digest v1.0.0 // indirect
+	github.com/opencontainers/image-spec v1.1.0 // indirect
+	github.com/pkg/errors v0.9.1 // indirect
+	github.com/pmezard/go-difflib v1.0.0 // indirect
+	github.com/power-devops/perfstat v0.0.0-20210106213030-5aafc221ea8c // indirect
+	github.com/shirou/gopsutil/v3 v3.23.12 // indirect
+	github.com/shoenig/go-m1cpu v0.1.6 // indirect
+	github.com/sirupsen/logrus v1.9.3 // indirect
+	github.com/testcontainers/testcontainers-go v0.33.0 // indirect
+	github.com/tklauser/go-sysconf v0.3.12 // indirect
+	github.com/tklauser/numcpus v0.6.1 // indirect
+	github.com/yusufpapurcu/wmi v1.2.3 // indirect
 	go.opencensus.io v0.24.0 // indirect
 	go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.49.0 // indirect
 	go.opentelemetry.io/otel v1.24.0 // indirect
 	go.opentelemetry.io/otel/metric v1.24.0 // indirect
 	go.opentelemetry.io/otel/trace v1.24.0 // indirect
 	google.golang.org/genproto/googleapis/rpc v0.0.0-20240624140628-dc46fd24d27d // indirect
-	google.golang.org/grpc v1.64.0 // indirect
+	google.golang.org/grpc v1.64.1 // indirect
 )
 
 require (