fix: resolve data races in test mocks under -race detector

Author: razvan

internal/service/engine/engine_exact_search_test.go

@@ -3,6 +3,7 @@ package engine
 import (
 "context"
 "errors"
+"sync"
 "testing"
 
 "github.com/doITmagic/rag-code-mcp/internal/config"
@@ -22,6 +23,7 @@ type testStore struct {
 storage.VectorStore
 existing         map[string]bool
 exactSearchFunc  func(coll string, filters map[string]interface{}) ([]storage.SearchResult, error)
+mu               sync.Mutex
 recordedFilters  []capturedCall
 exactSearchErr   error
 }
@@ -36,7 +38,9 @@ return s.existing[name], nil
 }
 
 func (s *testStore) ExactSearch(_ context.Context, coll string, filters map[string]interface{}, _ int) ([]storage.SearchResult, error) {
+s.mu.Lock()
 s.recordedFilters = append(s.recordedFilters, capturedCall{collection: coll, filters: filters})
+s.mu.Unlock()
 if s.exactSearchErr != nil {
 return nil, s.exactSearchErr
 }

internal/service/tools/tests/mocks_test.go

@@ -2,6 +2,7 @@ package tests
 
 import (
 	"context"
+	"sync"
 
 	"github.com/doITmagic/rag-code-mcp/internal/config"
 	"github.com/doITmagic/rag-code-mcp/internal/service/engine"
@@ -15,6 +16,7 @@ import (
 
 type mockVectorStore struct {
 	storage.VectorStore
+	mu                   sync.Mutex
 	SearchFunc           func(ctx context.Context, collection string, query storage.SearchQuery) ([]storage.SearchResult, error)
 	SearchCodeOnlyFunc   func(ctx context.Context, collection string, query storage.SearchQuery) ([]storage.SearchResult, error)
 	ExactSearchFunc      func(ctx context.Context, collection string, filters map[string]interface{}, limit int) ([]storage.SearchResult, error)
@@ -37,6 +39,8 @@ func (m *mockVectorStore) SearchCodeOnly(ctx context.Context, collection string,
 }
 
 func (m *mockVectorStore) ExactSearch(ctx context.Context, collection string, filters map[string]interface{}, limit int) ([]storage.SearchResult, error) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
 	if m.ExactSearchFunc != nil {
 		return m.ExactSearchFunc(ctx, collection, filters, limit)
 	}