Курнат Андрей
95 lines
2.3 KiB
Go
95 lines
2.3 KiB
Go
package osu_test
|
|
|
|
import (
|
|
"io"
|
|
"testing"
|
|
|
|
opensearchgoAPI "github.com/opensearch-project/opensearch-go/v4/opensearchapi"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/qsfera/server/services/search/pkg/opensearch/internal/osu"
|
|
opensearchtest "github.com/qsfera/server/services/search/pkg/opensearch/internal/test"
|
|
)
|
|
|
|
func TestRequestBody(t *testing.T) {
|
|
tests := []opensearchtest.TableTest[osu.Builder, map[string]any]{
|
|
{
|
|
Name: "simple",
|
|
Got: osu.NewQueryReqBody[any](osu.NewTermQuery[string]("name").Value("tom")),
|
|
Want: map[string]any{
|
|
"query": map[string]any{
|
|
"term": map[string]any{
|
|
"name": map[string]any{
|
|
"value": "tom",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.Name, func(t *testing.T) {
|
|
assert.JSONEq(t, opensearchtest.JSONMustMarshal(t, test.Want), opensearchtest.JSONMustMarshal(t, test.Got))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestBuildSearchReq(t *testing.T) {
|
|
tests := []opensearchtest.TableTest[io.Reader, map[string]any]{
|
|
{
|
|
Name: "highlight",
|
|
Got: func() io.Reader {
|
|
req, _ := osu.BuildSearchReq(
|
|
&opensearchgoAPI.SearchReq{},
|
|
osu.NewTermQuery[string]("content").Value("content"),
|
|
osu.SearchBodyParams{
|
|
Highlight: &osu.BodyParamHighlight{
|
|
HighlightOptions: osu.HighlightOptions{
|
|
PreTags: []string{"<b>"},
|
|
PostTags: []string{"</b>"},
|
|
},
|
|
Fields: map[string]osu.HighlightOptions{
|
|
"content": {
|
|
PreTags: []string{"<strong>"},
|
|
PostTags: []string{"</strong>"},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
)
|
|
|
|
return req.Body
|
|
}(),
|
|
Want: map[string]any{
|
|
"query": map[string]any{
|
|
"term": map[string]any{
|
|
"content": map[string]any{
|
|
"value": "content",
|
|
},
|
|
},
|
|
},
|
|
"highlight": map[string]any{
|
|
"pre_tags": []string{"<b>"},
|
|
"post_tags": []string{"</b>"},
|
|
"fields": map[string]any{
|
|
"content": map[string]any{
|
|
"pre_tags": []string{"<strong>"},
|
|
"post_tags": []string{"</strong>"},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.Name, func(t *testing.T) {
|
|
body, err := io.ReadAll(test.Got)
|
|
require.NoError(t, err)
|
|
assert.JSONEq(t, opensearchtest.JSONMustMarshal(t, test.Want), string(body))
|
|
})
|
|
}
|
|
}
|