e91139d3ec
Signed-off-by: Thomas Miceli <tho.miceli@gmail.com> Co-authored-by: Qiang Zhou <zhouqiang.loaded@bytedance.com> Co-authored-by: theodoruszq <theodoruszq@gmail.com>
89 lines
2.7 KiB
Go
89 lines
2.7 KiB
Go
package index
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"strconv"
|
|
"testing"
|
|
|
|
"github.com/meilisearch/meilisearch-go"
|
|
"github.com/rs/zerolog"
|
|
)
|
|
|
|
// syncMeiliIndexer wraps MeiliIndexer to make Add/Remove synchronous for tests.
|
|
type syncMeiliIndexer struct {
|
|
*MeiliIndexer
|
|
}
|
|
|
|
func (s *syncMeiliIndexer) Add(gist *Gist) error {
|
|
if gist == nil {
|
|
return fmt.Errorf("failed to add nil gist to index")
|
|
}
|
|
doc := &meiliGist{
|
|
Gist: *gist,
|
|
ContentSplit: splitCamelCase(gist.Content),
|
|
}
|
|
primaryKey := "GistID"
|
|
taskInfo, err := s.index.AddDocuments(doc, &meilisearch.DocumentOptions{PrimaryKey: &primaryKey})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = s.client.WaitForTask(taskInfo.TaskUID, 0)
|
|
return err
|
|
}
|
|
|
|
func (s *syncMeiliIndexer) Remove(gistID uint) error {
|
|
taskInfo, err := s.index.DeleteDocument(strconv.Itoa(int(gistID)), nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = s.client.WaitForTask(taskInfo.TaskUID, 0)
|
|
return err
|
|
}
|
|
|
|
func setupMeiliIndexer(t *testing.T) (Indexer, func()) {
|
|
zerolog.SetGlobalLevel(zerolog.Disabled)
|
|
t.Helper()
|
|
|
|
host := os.Getenv("OG_TEST_MEILI_HOST")
|
|
if host == "" {
|
|
host = "http://localhost:47700"
|
|
}
|
|
apiKey := os.Getenv("OG_TEST_MEILI_API_KEY")
|
|
|
|
indexName := fmt.Sprintf("test_%d", os.Getpid())
|
|
|
|
inner := NewMeiliIndexer(host, apiKey, indexName)
|
|
err := inner.Init()
|
|
if err != nil {
|
|
t.Skipf("MeiliSearch not available at %s: %v", host, err)
|
|
}
|
|
|
|
wrapped := &syncMeiliIndexer{MeiliIndexer: inner}
|
|
|
|
// Store the inner MeiliIndexer in atomicIndexer, because MeiliIndexer.Search
|
|
// type-asserts the global to *MeiliIndexer.
|
|
var idx Indexer = inner
|
|
atomicIndexer.Store(&idx)
|
|
|
|
cleanup := func() {
|
|
atomicIndexer.Store(nil)
|
|
inner.Reset()
|
|
inner.Close()
|
|
}
|
|
|
|
return wrapped, cleanup
|
|
}
|
|
|
|
func TestMeiliAddAndSearch(t *testing.T) { testAddAndSearch(t, setupMeiliIndexer) }
|
|
func TestMeiliAccessControl(t *testing.T) { testAccessControl(t, setupMeiliIndexer) }
|
|
func TestMeiliMetadataFilters(t *testing.T) { testMetadataFilters(t, setupMeiliIndexer) }
|
|
func TestMeiliAllFieldSearch(t *testing.T) { testAllFieldSearch(t, setupMeiliIndexer) }
|
|
func TestMeiliFuzzySearch(t *testing.T) { testFuzzySearch(t, setupMeiliIndexer) }
|
|
func TestMeiliContentSearch(t *testing.T) { testContentSearch(t, setupMeiliIndexer) }
|
|
func TestMeiliPagination(t *testing.T) { testPagination(t, setupMeiliIndexer) }
|
|
func TestMeiliLanguageFacets(t *testing.T) { testLanguageFacets(t, setupMeiliIndexer) }
|
|
func TestMeiliMetadataOnlySearch(t *testing.T) { testMetadataOnlySearch(t, setupMeiliIndexer) }
|
|
func TestMeiliTitleFuzzySearch(t *testing.T) { testTitleFuzzySearch(t, setupMeiliIndexer) }
|
|
func TestMeiliMultiLanguageFacets(t *testing.T) { testMultiLanguageFacets(t, setupMeiliIndexer) }
|