Updated all code to use SafeConn and SafeTX

middleware/pageprotection_test.go

@@ -5,6 +5,7 @@ import (
 	"net/http/httptest"
 	"testing"
 
+	"projectreshoot/db"
 	"projectreshoot/tests"
 
 	"github.com/stretchr/testify/assert"
@@ -13,13 +14,14 @@ import (
 
 func TestPageLoginRequired(t *testing.T) {
 	// Basic setup
+	conn, err := tests.SetupTestDB()
+	require.NoError(t, err)
+	sconn := db.MakeSafe(conn)
+	defer sconn.Close()
+
 	cfg, err := tests.TestConfig()
 	require.NoError(t, err)
-	logger := tests.NilLogger()
-	conn, err := tests.SetupTestDB(t.Context())
-	require.NoError(t, err)
-	require.NotNil(t, conn)
-	defer tests.DeleteTestDB()
+	logger := tests.DebugLogger(t)
 
 	// Handler to check outcome of Authentication middleware
 	testHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
@@ -28,7 +30,7 @@ func TestPageLoginRequired(t *testing.T) {
 
 	// Add the middleware and create the server
 	loginRequiredHandler := RequiresLogin(testHandler)
-	authHandler := Authentication(logger, cfg, conn, loginRequiredHandler)
+	authHandler := Authentication(logger, cfg, sconn, loginRequiredHandler)
 	server := httptest.NewServer(authHandler)
 	defer server.Close()