Fix tests.

Author: Andreas Wagner

internal/handlers/admin_test.go

@@ -113,6 +113,6 @@ func TestAdminFunc(t *testing.T) {
 		shutDownServer()
 	})
 
-	fmt.Printf("\n\n\n\n")
+	fmt.Printf("\n")
 
 }

internal/handlers/api_standards_test.go

@@ -176,5 +176,5 @@ func TestAPIStandardFunc(t *testing.T) {
 		shutDownServer()
 	})
 
-	fmt.Printf("\n\n\n\n")
+	fmt.Printf("\n")
 }

internal/handlers/embeddings_test.go

@@ -334,5 +334,5 @@ func TestEmbeddingsFunc(t *testing.T) {
 		shutDownServer()
 	})
 
-	fmt.Printf("\n\n\n\n")
+	fmt.Printf("\n")
 }

internal/handlers/llm_services_sharing_test.go

@@ -376,5 +376,5 @@ func TestDefinitionSharingFunc(t *testing.T) {
 		shutDownServer()
 	})
 
-	fmt.Printf("\n\n\n\n")
+	fmt.Printf("\n")
 }

internal/handlers/llm_services_test.go

@@ -14,7 +14,7 @@ import (
 
 func TestInstancesFunc(t *testing.T) {
 
-	fmt.Printf("\n\n\n\n")
+	fmt.Printf("\n")
 
 	// Get the database connection pool from package variable
 	pool := connPool

internal/handlers/projects_test.go

@@ -417,6 +417,6 @@ func TestProjectTransactionRollback(t *testing.T) {
 		shutDownServer()
 	})
 
-	fmt.Printf("\n\n\n\n")
+	fmt.Printf("\n")
 }
 */

internal/handlers/projects_transfer_ownership_test.go

@@ -183,7 +183,7 @@ func TestProjectTransferOwnershipFunc(t *testing.T) {
 
 			// Build the request URL
 			requestURL := fmt.Sprintf("http://%s:%d%s", options.Host, options.Port, tc.requestPath)
-			
+
 			// Create the request
 			req, err := http.NewRequest(tc.method, requestURL, bodyReader)
 			if err != nil {
@@ -319,8 +319,24 @@ func TestProjectTransferOwnershipFunc(t *testing.T) {
 		assert.Equal(t, "owner", response["role"])
 	})
 
-	// Cleanup - shutdown the server
-	defer shutDownServer()
+	// Cleanup
+	t.Cleanup(func() {
+		fmt.Print("\n\nRunning cleanup ...\n\n")
+
+		requestURL := fmt.Sprintf("http://%s:%d/v1/admin/footgun", options.Host, options.Port)
+		req, err := http.NewRequest(http.MethodGet, requestURL, nil)
+		assert.NoError(t, err)
+		req.Header.Set("Authorization", "Bearer "+options.AdminKey)
+		_, err = http.DefaultClient.Do(req)
+		if err != nil && err.Error() != "no rows in result set" {
+			t.Fatalf("Error sending request: %v\n", err)
+		}
+		assert.NoError(t, err)
+
+		fmt.Print("Shutting down server\n\n")
+		shutDownServer()
+	})
+
 }
 
 // Helper function to share a project

internal/handlers/public_access_test.go

@@ -209,7 +209,7 @@ func TestPublicAccess(t *testing.T) {
 		shutDownServer()
 	})
 
-	fmt.Printf("\n\n\n\n")
+	fmt.Printf("\n")
 }
 
 // Helper function to post embeddings

internal/handlers/similars_test.go

@@ -91,7 +91,7 @@ func TestSimilarsFunc(t *testing.T) {
 			requestPath:  "/v1/similars/alice/test1/https%3A%2F%2Fid.salamanca.school%2Ftexts%2FW0001%3Avol1.1.1.1.1",
 			bodyPath:     "",
 			apiKey:       aliceAPIKey,
-			expectBody:   "",  // Will validate structure programmatically
+			expectBody:   "", // Will validate structure programmatically
 			expectStatus: http.StatusOK,
 		},
 		{
@@ -100,7 +100,7 @@ func TestSimilarsFunc(t *testing.T) {
 			requestPath:  "/v1/similars/alice/test1/https%3A%2F%2Fid.salamanca.school%2Ftexts%2FW0001%3Avol1.1.1.1.1?metadata_path=author&metadata_value=Immanuel%20Kant",
 			bodyPath:     "",
 			apiKey:       options.AdminKey,
-			expectBody:   "",  // Will validate structure programmatically
+			expectBody:   "", // Will validate structure programmatically
 			expectStatus: http.StatusOK,
 		},
 	}
@@ -145,14 +145,14 @@ func TestSimilarsFunc(t *testing.T) {
 
 			respBody, err := io.ReadAll(resp.Body) // response body is []byte
 			assert.NoError(t, err)
-			
+
 			// Parse and validate JSON structure
 			if v.expectBody == "" && resp.StatusCode == http.StatusOK {
 				// Validate that response has correct structure with results array
 				var result map[string]interface{}
 				err = json.Unmarshal(respBody, &result)
 				assert.NoError(t, err)
-				
+
 				// Check results field exists and is an array
 				results, ok := result["results"].([]interface{})
 				if !ok {
@@ -212,7 +212,7 @@ func TestSimilarsFunc(t *testing.T) {
 		shutDownServer()
 	})
 
-	fmt.Printf("\n\n\n\n")
+	fmt.Printf("\n")
 }
 
 // TestPostSimilar tests the POST similar functionality.
@@ -296,11 +296,11 @@ func TestPostSimilar(t *testing.T) {
 		expectError  bool
 	}{
 		{
-			name:        "POST similar with valid 5D vector",
-			method:      http.MethodPost,
-			requestPath: "/v1/similars/alice/test1",
-			body:        `{"vector": [-0.02085085, 0.01852216, 0.05327000, 0.07138438, 0.02000308]}`,
-			apiKey:      aliceAPIKey,
+			name:         "POST similar with valid 5D vector",
+			method:       http.MethodPost,
+			requestPath:  "/v1/similars/alice/test1",
+			body:         `{"vector": [-0.02085085, 0.01852216, 0.05327000, 0.07138438, 0.02000308]}`,
+			apiKey:       aliceAPIKey,
 			expectStatus: http.StatusOK,
 			expectIDs: []string{
 				"https%3A%2F%2Fid.salamanca.school%2Ftexts%2FW0001%3Avol1.1.1.1.1",
@@ -310,11 +310,11 @@ func TestPostSimilar(t *testing.T) {
 			expectError: false,
 		},
 		{
-			name:        "POST similar with valid 5D vector and metadata filter",
-			method:      http.MethodPost,
-			requestPath: "/v1/similars/alice/test1?metadata_path=author&metadata_value=Immanuel%20Kant",
-			body:        `{"vector": [-0.02085085, 0.01852216, 0.05327000, 0.07138438, 0.02000308]}`,
-			apiKey:      aliceAPIKey,
+			name:         "POST similar with valid 5D vector and metadata filter",
+			method:       http.MethodPost,
+			requestPath:  "/v1/similars/alice/test1?metadata_path=author&metadata_value=Immanuel%20Kant",
+			body:         `{"vector": [-0.02085085, 0.01852216, 0.05327000, 0.07138438, 0.02000308]}`,
+			apiKey:       aliceAPIKey,
 			expectStatus: http.StatusOK,
 			expectIDs: []string{
 				"https%3A%2F%2Fid.salamanca.school%2Ftexts%2FW0001%3Avol2",
@@ -425,7 +425,7 @@ func TestPostSimilar(t *testing.T) {
 							continue
 						}
 						actualIDs[i] = id.(string)
-						
+
 						// Verify similarity field exists and is a number
 						similarity, hasSim := resultItem["similarity"]
 						if !hasSim {
@@ -434,7 +434,7 @@ func TestPostSimilar(t *testing.T) {
 							t.Errorf("Result item %d 'similarity' is not a number", i)
 						}
 					}
-					
+
 					// Check that all expected IDs are present (order doesn't matter for similar items)
 					for _, expectedID := range v.expectIDs {
 						found := false
@@ -474,5 +474,5 @@ func TestPostSimilar(t *testing.T) {
 		shutDownServer()
 	})
 
-	fmt.Printf("\n\n\n\n")
+	fmt.Printf("\n")
 }

internal/handlers/system_user_restrictions_test.go

@@ -24,7 +24,6 @@ func TestSystemUserRestrictions(t *testing.T) {
 	// Start the server
 	err, shutDownServer := startTestServer(t, pool, mockKeyGen)
 	assert.NoError(t, err)
-	defer shutDownServer()
 
 	// Create a regular user (alice) for testing
 	aliceJSON := `{"user_handle": "alice", "name": "Alice Doe", "email": "alice@foo.bar"}`
@@ -195,4 +194,23 @@ func TestSystemUserRestrictions(t *testing.T) {
 			}
 		})
 	}
+
+	// Cleanup
+	t.Cleanup(func() {
+		fmt.Print("\n\nRunning cleanup ...\n\n")
+
+		requestURL := fmt.Sprintf("http://%s:%d/v1/admin/footgun", options.Host, options.Port)
+		req, err := http.NewRequest(http.MethodGet, requestURL, nil)
+		assert.NoError(t, err)
+		req.Header.Set("Authorization", "Bearer "+options.AdminKey)
+		_, err = http.DefaultClient.Do(req)
+		if err != nil && err.Error() != "no rows in result set" {
+			t.Fatalf("Error sending request: %v\n", err)
+		}
+		assert.NoError(t, err)
+
+		fmt.Print("Shutting down server\n\n")
+		shutDownServer()
+	})
+
 }