feat(dir): add integration tests for server

Signed-off-by: Bendegúz Csirmaz <csirmazbendeguz@gmail.com>

Author: csirmazbendeguz

server/database/database.go

@@ -37,22 +37,22 @@ func New(config config.Config) (types.DatabaseAPI, error) {
 	}
 }
 
-func newCustomLogger() gormlogger.Interface {
-	// Create a custom logger configuration that ignores "record not found" errors
-	// since these are expected during normal operation (checking if records exist)
-	return gormlogger.New(
-		log.New(os.Stdout, "\r\n", log.LstdFlags),
-		gormlogger.Config{
-			SlowThreshold:             200 * time.Millisecond, //nolint:mnd
-			LogLevel:                  gormlogger.Warn,
-			IgnoreRecordNotFoundError: true,
-			Colorful:                  true,
-		},
-	)
-}
-
 // newPostgres creates a new database connection using PostgreSQL driver.
 func newPostgres(cfg config.PostgresConfig) (*gormdb.DB, error) {
+	db, err := NewPostgresGormDb(cfg)
+	if err != nil {
+		return nil, fmt.Errorf("failed to connect to PostgreSQL database: %w", err)
+	}
+
+	gdb, err := gormdb.New(db)
+	if err != nil {
+		return nil, fmt.Errorf("failed to initialize PostgreSQL database: %w", err)
+	}
+
+	return gdb, nil
+}
+
+func NewPostgresGormDb(cfg config.PostgresConfig) (*gorm.DB, error) {
 	host := cfg.Host
 	if host == "" {
 		host = config.DefaultPostgresHost
@@ -76,17 +76,21 @@ func newPostgres(cfg config.PostgresConfig) (*gormdb.DB, error) {
 	dsn := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=%s",
 		host, port, cfg.Username, cfg.Password, database, sslMode)
 
-	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{
+	return gorm.Open(postgres.Open(dsn), &gorm.Config{
 		Logger: newCustomLogger(),
 	})
-	if err != nil {
-		return nil, fmt.Errorf("failed to connect to PostgreSQL database: %w", err)
-	}
-
-	gdb, err := gormdb.New(db)
-	if err != nil {
-		return nil, fmt.Errorf("failed to initialize PostgreSQL database: %w", err)
-	}
+}
 
-	return gdb, nil
+func newCustomLogger() gormlogger.Interface {
+	// Create a custom logger configuration that ignores "record not found" errors
+	// since these are expected during normal operation (checking if records exist)
+	return gormlogger.New(
+		log.New(os.Stdout, "\r\n", log.LstdFlags),
+		gormlogger.Config{
+			SlowThreshold:             200 * time.Millisecond, //nolint:mnd
+			LogLevel:                  gormlogger.Warn,
+			IgnoreRecordNotFoundError: true,
+			Colorful:                  true,
+		},
+	)
 }

server/server.go

@@ -63,7 +63,7 @@ type Server struct {
 	authzService       *authz.Service
 	publicationService *publication.Service
 	health             *healthcheck.Checker
-	grpcServer         *grpc.Server
+	GrpcServer         *grpc.Server
 	metricsServer      *metrics.Server
 }
 
@@ -114,7 +114,7 @@ func buildConnectionOptions(cfg config.ConnectionConfig) []grpc.ServerOption {
 func Run(ctx context.Context, cfg *config.Config) error {
 	errCh := make(chan error)
 
-	server, err := New(ctx, cfg)
+	server, err := New(ctx, cfg, nil)
 	if err != nil {
 		return fmt.Errorf("failed to create server: %w", err)
 	}
@@ -153,7 +153,7 @@ func configureOASFValidation(cfg *config.Config) error {
 }
 
 //nolint:cyclop // This function has been at the limit; refactoring is out of scope.
-func New(ctx context.Context, cfg *config.Config) (*Server, error) {
+func New(ctx context.Context, cfg *config.Config, databaseAPI types.DatabaseAPI) (*Server, error) {
 	logger.Debug("Creating server with config", "config", cfg, "version", version.String())
 
 	if err := configureOASFValidation(cfg); err != nil {
@@ -227,9 +227,11 @@ func New(ctx context.Context, cfg *config.Config) (*Server, error) {
 		return nil, fmt.Errorf("failed to create routing: %w", err)
 	}
 
-	databaseAPI, err := database.New(cfg.Database)
-	if err != nil {
-		return nil, fmt.Errorf("failed to create database API: %w", err)
+	if databaseAPI == nil {
+		databaseAPI, err = database.New(cfg.Database)
+		if err != nil {
+			return nil, fmt.Errorf("failed to create database API: %w", err)
+		}
 	}
 
 	// Create JWT authentication service if enabled
@@ -312,7 +314,7 @@ func New(ctx context.Context, cfg *config.Config) (*Server, error) {
 		authzService:       authzService,
 		publicationService: publicationService,
 		health:             healthChecker,
-		grpcServer:         grpcServer,
+		GrpcServer:         grpcServer,
 		metricsServer:      metricsServer,
 	}, nil
 }
@@ -385,7 +387,7 @@ func (s Server) Close(ctx context.Context) {
 		}
 	}
 
-	s.grpcServer.GracefulStop()
+	s.GrpcServer.GracefulStop()
 }
 
 func (s Server) start(ctx context.Context) error {
@@ -428,7 +430,7 @@ func (s Server) start(ctx context.Context) error {
 	go func() {
 		logger.Info("Server starting", "address", s.Options().Config().ListenAddress)
 
-		if err := s.grpcServer.Serve(listen); err != nil {
+		if err := s.GrpcServer.Serve(listen); err != nil {
 			logger.Error("Failed to start server", "error", err)
 		}
 	}()

tests/Taskfile.yml

@@ -596,3 +596,10 @@ tasks:
         else
           echo "No pods directory found"
         fi
+
+  test:integration:server:
+    desc: Run integration tests for the API server
+    dir: "{{ .ROOT_DIR }}/tests/integration-server"
+    cmds:
+      - docker compose up -d --build --wait
+      - ginkgo --randomize-all -p