fix: unit tests

Former-commit-id: d369354
Former-commit-id: a955ae2ce4698a06c48270bab16ef6a688d755ed

Author: linhdmn

mcp-server.REMOVED.git-id

@@ -1 +1 @@
-64cc0aa69d3ec38779a297042faf233beea5c405
+63a866e3c565fc326675a4051e3074bf542a9b0f

pkg/dbtools/performance_test.go

@@ -7,8 +7,14 @@ import (
 )
 
 func TestPerformanceAnalyzer(t *testing.T) {
-	// Create a new performance analyzer
-	analyzer := NewPerformanceAnalyzer()
+	// Get the global performance analyzer and reset it to ensure clean state
+	analyzer := GetPerformanceAnalyzer()
+	analyzer.Reset()
+
+	// Ensure we restore previous state after test
+	defer func() {
+		analyzer.Reset()
+	}()
 
 	// Test tracking a query
 	ctx := context.Background()
@@ -27,6 +33,9 @@ func TestPerformanceAnalyzer(t *testing.T) {
 		t.Errorf("Expected result to be 'test result', got %v", result)
 	}
 
+	// Add a small delay to ensure async metrics update completes
+	time.Sleep(10 * time.Millisecond)
+
 	// Check metrics were collected
 	metrics := analyzer.GetAllMetrics()
 	if len(metrics) == 0 {
@@ -36,7 +45,7 @@ func TestPerformanceAnalyzer(t *testing.T) {
 	// Find the test query in metrics
 	var foundMetrics *QueryMetrics
 	for _, m := range metrics {
-		if m.Query == "SELECT * FROM test_table" {
+		if normalizeQuery(m.Query) == normalizeQuery("SELECT * FROM test_table") {
 			foundMetrics = m
 			break
 		}