Курнат Андрей
64 lines
2.1 KiB
Go
64 lines
2.1 KiB
Go
package opensearch_test
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/tidwall/sjson"
|
|
|
|
"github.com/qsfera/server/services/search/pkg/opensearch"
|
|
"github.com/qsfera/server/services/search/pkg/opensearch/internal/test"
|
|
)
|
|
|
|
func TestIndexManager(t *testing.T) {
|
|
t.Run("index plausibility", func(t *testing.T) {
|
|
tests := []opensearchtest.TableTest[opensearch.IndexManager, struct{}]{
|
|
{
|
|
Name: "empty",
|
|
Got: opensearch.IndexManagerLatest,
|
|
},
|
|
}
|
|
tc := opensearchtest.NewDefaultTestClient(t, defaultConfig.Engine.OpenSearch.Client)
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.Name, func(t *testing.T) {
|
|
indexName := "qsfera-test-resource"
|
|
tc.Require.IndicesReset([]string{indexName})
|
|
|
|
body, err := test.Got.MarshalJSON()
|
|
require.NoError(t, err)
|
|
require.NotEmpty(t, body)
|
|
require.NotEmpty(t, test.Got.String())
|
|
require.JSONEq(t, test.Got.String(), string(body))
|
|
require.NoError(t, test.Got.Apply(t.Context(), indexName, tc.Client()))
|
|
})
|
|
}
|
|
})
|
|
|
|
t.Run("does not create index if it already exists and is up to date", func(t *testing.T) {
|
|
indexManager := opensearch.IndexManagerLatest
|
|
indexName := "qsfera-test-resource"
|
|
|
|
tc := opensearchtest.NewDefaultTestClient(t, defaultConfig.Engine.OpenSearch.Client)
|
|
tc.Require.IndicesReset([]string{indexName})
|
|
tc.Require.IndicesCreate(indexName, strings.NewReader(indexManager.String()))
|
|
|
|
require.NoError(t, indexManager.Apply(t.Context(), indexName, tc.Client()))
|
|
})
|
|
|
|
t.Run("fails to create index if it already exists but is not up to date", func(t *testing.T) {
|
|
indexManager := opensearch.IndexManagerLatest
|
|
indexName := "qsfera-test-resource"
|
|
|
|
tc := opensearchtest.NewDefaultTestClient(t, defaultConfig.Engine.OpenSearch.Client)
|
|
tc.Require.IndicesReset([]string{indexName})
|
|
|
|
body, err := sjson.Set(indexManager.String(), "settings.number_of_shards", "2")
|
|
require.NoError(t, err)
|
|
tc.Require.IndicesCreate(indexName, strings.NewReader(body))
|
|
|
|
require.ErrorIs(t, indexManager.Apply(t.Context(), indexName, tc.Client()), opensearch.ErrManualActionRequired)
|
|
})
|
|
}
|