package search

import (
	"context"
	"testing"
	"tercul/internal/domain"
	domainsearch "tercul/internal/domain/search"
	"github.com/stretchr/testify/assert"
)

func TestWeaviateWrapper_Search(t *testing.T) {
	mockWrapper := new(MockWeaviateWrapper)
	expectedResults := &domain.SearchResults{
		Works: []domain.Work{
			{Title: "Work 1", Description: "alpha beta", TranslatableModel: domain.TranslatableModel{Language: "en"}},
		},
	}
	params := domainsearch.SearchParams{
		Query: "alpha",
		Mode:  domainsearch.SearchModeHybrid,
		Filters: domain.SearchFilters{
			Languages: []string{"en"},
		},
		Limit:  1,
		Offset: 0,
	}

	mockWrapper.On("Search", context.Background(), params).Return(expectedResults, nil)

	results, err := mockWrapper.Search(context.Background(), params)

	assert.NoError(t, err)
	assert.NotNil(t, results)
	assert.Equal(t, 1, len(results.Works))
	assert.Equal(t, "Work 1", results.Works[0].Title)

	mockWrapper.AssertExpectations(t)
}