Added tests to ensure SIGUSR1 and SIGUSR2 will toggle db lock

Makefile

@@ -23,6 +23,8 @@ test:
 	go mod tidy && \
    	templ generate && \
 	go generate && \
+	go test .
+	go test ./db
 	go test ./middleware
 
 clean:

db/connection.go

@@ -4,7 +4,6 @@ import (
 	"context"
 	"database/sql"
 	"fmt"
-	"os"
 	"time"
 
 	"github.com/pkg/errors"

main.go

@@ -67,16 +67,14 @@ func handleMaintSignals(
 			case syscall.SIGUSR1:
 				if atomic.LoadUint32(&maint) != 1 {
 					atomic.StoreUint32(&maint, 1)
-					log := logger.With().Logger().Output(os.Stdout)
+					logger.Info().Msg("Signal received: Starting maintenance")
-					log.Info().Msg("Signal received: Starting maintenance")
+					logger.Info().Msg("Attempting to acquire database lock")
-					log.Info().Msg("Attempting to acquire database lock")
 					conn.Pause(config.DBLockTimeout * time.Second)
 				}
 			case syscall.SIGUSR2:
 				if atomic.LoadUint32(&maint) != 0 {
-					log := logger.With().Logger().Output(os.Stdout)
+					logger.Info().Msg("Signal received: Maintenance over")
-					log.Info().Msg("Signal received: Maintenance over")
+					logger.Info().Msg("Releasing database lock")
-					log.Info().Msg("Releasing database lock")
 					conn.Resume()
 					atomic.StoreUint32(&maint, 0)
 				}

main_test.go

@@ -1,12 +1,17 @@
 package main
 
 import (
+	"bytes"
 	"context"
 	"fmt"
 	"net/http"
 	"os"
+	"strings"
+	"syscall"
 	"testing"
 	"time"
+
+	"github.com/stretchr/testify/require"
 )
 
 func Test_main(t *testing.T) {
@@ -14,13 +19,60 @@ func Test_main(t *testing.T) {
 	ctx, cancel := context.WithCancel(ctx)
 	t.Cleanup(cancel)
 	args := map[string]string{}
-	go run(ctx, os.Stdout, args)
+	var stdout bytes.Buffer
+	go run(ctx, &stdout, args)
 
-	// wait for the server to become available
 	waitForReady(ctx, 10*time.Second, "http://localhost:3333/healthz")
 
-	// do tests
+	t.Run("SIGUSR1 puts database into global lock", func(t *testing.T) {
-	fmt.Println("Tests starting")
+		done := make(chan bool)
+		go func() {
+			expected := "Global database lock acquired"
+			for {
+				if strings.Contains(stdout.String(), expected) {
+					done <- true
+					return
+				}
+				time.Sleep(100 * time.Millisecond)
+			}
+		}()
+
+		proc, err := os.FindProcess(os.Getpid())
+		require.NoError(t, err)
+		proc.Signal(syscall.SIGUSR1)
+
+		select {
+		case <-done:
+			t.Log("found")
+		case <-time.After(250 * time.Millisecond):
+			t.Errorf("Not found")
+		}
+	})
+
+	t.Run("SIGUSR2 releases database global lock", func(t *testing.T) {
+		done := make(chan bool)
+		go func() {
+			expected := "Global database lock released"
+			for {
+				if strings.Contains(stdout.String(), expected) {
+					done <- true
+					return
+				}
+				time.Sleep(100 * time.Millisecond)
+			}
+		}()
+
+		proc, err := os.FindProcess(os.Getpid())
+		require.NoError(t, err)
+		proc.Signal(syscall.SIGUSR2)
+
+		select {
+		case <-done:
+			t.Log("found")
+		case <-time.After(250 * time.Millisecond):
+			t.Errorf("Not found")
+		}
+	})
 }
 
 func waitForReady(