初始提交: Gitea 项目代码
This commit is contained in:
@@ -0,0 +1,94 @@
|
||||
// Copyright 2023 The Gitea Authors. All rights reserved.
|
||||
// SPDX-License-Identifier: MIT
|
||||
|
||||
package meilisearch
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"os"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"gitea.dev/modules/indexer/issues/internal"
|
||||
"gitea.dev/modules/indexer/issues/internal/tests"
|
||||
"gitea.dev/modules/json"
|
||||
"gitea.dev/modules/test"
|
||||
|
||||
"github.com/meilisearch/meilisearch-go"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestMeilisearchIndexer(t *testing.T) {
|
||||
// The meilisearch instance started by pull-db-tests.yml > test-unit > services > meilisearch
|
||||
url := test.ExternalServiceHTTP(t, "TEST_MEILISEARCH_URL", "http://meilisearch:7700")
|
||||
key := os.Getenv("TEST_MEILISEARCH_KEY")
|
||||
|
||||
require.Eventually(t, func() bool {
|
||||
resp, err := http.Get(url)
|
||||
if err != nil {
|
||||
return false
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
return resp.StatusCode == http.StatusOK
|
||||
}, time.Minute, time.Second, "Expected meilisearch to be up")
|
||||
|
||||
indexer := NewIndexer(url, key, fmt.Sprintf("test_meilisearch_indexer_%d", time.Now().Unix()))
|
||||
defer indexer.Close()
|
||||
|
||||
tests.TestIndexer(t, indexer)
|
||||
}
|
||||
|
||||
func TestConvertHits(t *testing.T) {
|
||||
convert := func(d any) []byte {
|
||||
b, _ := json.Marshal(d)
|
||||
return b
|
||||
}
|
||||
|
||||
_, err := convertHits(&meilisearch.SearchResponse{
|
||||
Hits: []meilisearch.Hit{
|
||||
{
|
||||
"aa": convert(1),
|
||||
"bb": convert(2),
|
||||
"cc": convert(3),
|
||||
"dd": convert(4),
|
||||
},
|
||||
},
|
||||
})
|
||||
assert.ErrorIs(t, err, ErrMalformedResponse)
|
||||
|
||||
validResponse := &meilisearch.SearchResponse{
|
||||
Hits: []meilisearch.Hit{
|
||||
{
|
||||
"id": convert(float64(11)),
|
||||
"title": convert("a title"),
|
||||
"content": convert("issue body with no match"),
|
||||
"comments": convert([]any{"hey whats up?", "I'm currently bowling", "nice"}),
|
||||
},
|
||||
{
|
||||
"id": convert(float64(22)),
|
||||
"title": convert("Bowling as title"),
|
||||
"content": convert(""),
|
||||
"comments": convert([]any{}),
|
||||
},
|
||||
{
|
||||
"id": convert(float64(33)),
|
||||
"title": convert("Bowl-ing as fuzzy match"),
|
||||
"content": convert(""),
|
||||
"comments": convert([]any{}),
|
||||
},
|
||||
},
|
||||
}
|
||||
hits, err := convertHits(validResponse)
|
||||
assert.NoError(t, err)
|
||||
assert.Equal(t, []internal.Match{{ID: 11}, {ID: 22}, {ID: 33}}, hits)
|
||||
}
|
||||
|
||||
func TestDoubleQuoteKeyword(t *testing.T) {
|
||||
assert.Empty(t, doubleQuoteKeyword(""))
|
||||
assert.Equal(t, `"a" "b" "c"`, doubleQuoteKeyword("a b c"))
|
||||
assert.Equal(t, `"a" "d" "g"`, doubleQuoteKeyword("a d g"))
|
||||
assert.Equal(t, `"a" "d" "g"`, doubleQuoteKeyword("a d g"))
|
||||
assert.Equal(t, `"a" "d" "g"`, doubleQuoteKeyword(`a "" "d" """g`))
|
||||
}
|
||||
Reference in New Issue
Block a user