mirror of
https://github.com/SamyRai/tercul-backend.git
synced 2025-12-27 04:01:34 +00:00
d50722dad5
Some checks failed
- Updated database models and repositories to replace uint IDs with UUIDs. - Modified test fixtures to generate and use UUIDs for authors, translations, users, and works. - Adjusted mock implementations to align with the new UUID structure. - Ensured all relevant functions and methods are updated to handle UUIDs correctly. - Added necessary imports for UUID handling in various files.
125 lines
3.7 KiB
Go
125 lines
3.7 KiB
Go
package search
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/mock"
|
|
"tercul/internal/app/localization"
|
|
"tercul/internal/domain"
|
|
domainsearch "tercul/internal/domain/search"
|
|
)
|
|
|
|
type mockLocalizationRepository struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (m *mockLocalizationRepository) GetTranslation(ctx context.Context, key string, language string) (string, error) {
|
|
args := m.Called(ctx, key, language)
|
|
return args.String(0), args.Error(1)
|
|
}
|
|
|
|
func (m *mockLocalizationRepository) GetTranslations(ctx context.Context, keys []string, language string) (map[string]string, error) {
|
|
args := m.Called(ctx, keys, language)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).(map[string]string), args.Error(1)
|
|
}
|
|
|
|
func (m *mockLocalizationRepository) GetAuthorBiography(ctx context.Context, authorID uint, language string) (string, error) {
|
|
args := m.Called(ctx, authorID, language)
|
|
return args.String(0), args.Error(1)
|
|
}
|
|
|
|
func (m *mockLocalizationRepository) GetWorkContent(ctx context.Context, workID uint, language string) (string, error) {
|
|
args := m.Called(ctx, workID, language)
|
|
return args.String(0), args.Error(1)
|
|
}
|
|
|
|
type mockWeaviateWrapper struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (m *mockWeaviateWrapper) Search(ctx context.Context, params domainsearch.SearchParams) (*domainsearch.SearchResults, error) {
|
|
args := m.Called(ctx, params)
|
|
if args.Get(0) == nil {
|
|
return nil, args.Error(1)
|
|
}
|
|
return args.Get(0).(*domainsearch.SearchResults), args.Error(1)
|
|
}
|
|
|
|
func (m *mockWeaviateWrapper) IndexWork(ctx context.Context, work *domain.Work, content string) error {
|
|
args := m.Called(ctx, work, content)
|
|
return args.Error(0)
|
|
}
|
|
|
|
func TestSearchService_Search(t *testing.T) {
|
|
localizationRepo := new(mockLocalizationRepository)
|
|
localizationService := localization.NewService(localizationRepo)
|
|
weaviateWrapper := new(mockWeaviateWrapper)
|
|
service := NewService(weaviateWrapper, localizationService)
|
|
|
|
ctx := context.Background()
|
|
testQuery := "test query"
|
|
testFilters := domainsearch.SearchFilters{
|
|
Languages: []string{"en"},
|
|
Authors: []string{"1"},
|
|
Tags: []string{"test-tag"},
|
|
Categories: []string{"test-category"},
|
|
}
|
|
expectedResults := &domainsearch.SearchResults{
|
|
Results: []domainsearch.SearchResultItem{
|
|
{Type: "Work", Entity: domain.Work{Title: "Test Work"}, Score: 0.9},
|
|
{Type: "Author", Entity: domain.Author{Name: "Test Author"}, Score: 0.8},
|
|
},
|
|
TotalResults: 2,
|
|
Limit: 10,
|
|
Offset: 0,
|
|
}
|
|
|
|
params := domainsearch.SearchParams{
|
|
Query: testQuery,
|
|
Filters: testFilters,
|
|
Limit: 10,
|
|
Offset: 0,
|
|
}
|
|
weaviateWrapper.On("Search", ctx, params).Return(expectedResults, nil)
|
|
|
|
results, err := service.Search(ctx, params)
|
|
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, expectedResults, results)
|
|
weaviateWrapper.AssertExpectations(t)
|
|
}
|
|
|
|
func TestIndexService_IndexWork(t *testing.T) {
|
|
localizationRepo := new(mockLocalizationRepository)
|
|
localizationService := localization.NewService(localizationRepo)
|
|
weaviateWrapper := new(mockWeaviateWrapper)
|
|
service := NewService(weaviateWrapper, localizationService)
|
|
|
|
ctx := context.Background()
|
|
testWork := domain.Work{
|
|
TranslatableModel: domain.TranslatableModel{
|
|
BaseModel: domain.BaseModel{ID: 1},
|
|
Language: "en",
|
|
},
|
|
Title: "Test Work",
|
|
}
|
|
testContent := "This is the test content for the work."
|
|
|
|
// Expect a call to get the work's content.
|
|
localizationRepo.On("GetWorkContent", mock.Anything, testWork.ID, testWork.Language).Return(testContent, nil)
|
|
|
|
// Expect a call to the Weaviate wrapper with the fetched content.
|
|
weaviateWrapper.On("IndexWork", mock.Anything, &testWork, testContent).Return(nil)
|
|
|
|
err := service.IndexWork(ctx, testWork)
|
|
|
|
assert.NoError(t, err)
|
|
localizationRepo.AssertExpectations(t)
|
|
weaviateWrapper.AssertExpectations(t)
|
|
}
|