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>
88 lines
2.7 KiB
Go
88 lines
2.7 KiB
Go
package index
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// setupBleveIndexer creates a new BleveIndexer for testing
|
|
func setupBleveIndexer(t *testing.T) (Indexer, func()) {
|
|
zerolog.SetGlobalLevel(zerolog.Disabled)
|
|
t.Helper()
|
|
|
|
tmpDir, err := os.MkdirTemp("", "bleve-test-*")
|
|
require.NoError(t, err)
|
|
|
|
indexPath := filepath.Join(tmpDir, "test.index")
|
|
indexer := NewBleveIndexer(indexPath)
|
|
|
|
err = indexer.Init()
|
|
if err != nil {
|
|
os.RemoveAll(tmpDir)
|
|
t.Fatalf("Failed to initialize BleveIndexer: %v", err)
|
|
}
|
|
|
|
var idx Indexer = indexer
|
|
atomicIndexer.Store(&idx)
|
|
|
|
cleanup := func() {
|
|
atomicIndexer.Store(nil)
|
|
indexer.Close()
|
|
os.RemoveAll(tmpDir)
|
|
}
|
|
|
|
return indexer, cleanup
|
|
}
|
|
|
|
func TestBleveAddAndSearch(t *testing.T) { testAddAndSearch(t, setupBleveIndexer) }
|
|
func TestBleveAccessControl(t *testing.T) { testAccessControl(t, setupBleveIndexer) }
|
|
func TestBleveMetadataFilters(t *testing.T) { testMetadataFilters(t, setupBleveIndexer) }
|
|
func TestBleveAllFieldSearch(t *testing.T) { testAllFieldSearch(t, setupBleveIndexer) }
|
|
func TestBleveFuzzySearch(t *testing.T) { testFuzzySearch(t, setupBleveIndexer) }
|
|
func TestBleveContentSearch(t *testing.T) { testContentSearch(t, setupBleveIndexer) }
|
|
func TestBlevePagination(t *testing.T) { testPagination(t, setupBleveIndexer) }
|
|
func TestBleveLanguageFacets(t *testing.T) { testLanguageFacets(t, setupBleveIndexer) }
|
|
func TestBleveWildcardSearch(t *testing.T) { testWildcardSearch(t, setupBleveIndexer) }
|
|
func TestBleveMetadataOnlySearch(t *testing.T) { testMetadataOnlySearch(t, setupBleveIndexer) }
|
|
func TestBleveTitleFuzzySearch(t *testing.T) { testTitleFuzzySearch(t, setupBleveIndexer) }
|
|
func TestBleveMultiLanguageFacets(t *testing.T) { testMultiLanguageFacets(t, setupBleveIndexer) }
|
|
|
|
func TestBlevePersistence(t *testing.T) {
|
|
tmpDir, err := os.MkdirTemp("", "bleve-persist-test-*")
|
|
require.NoError(t, err)
|
|
defer os.RemoveAll(tmpDir)
|
|
|
|
indexPath := filepath.Join(tmpDir, "test.index")
|
|
|
|
// Create and populate index
|
|
indexer1 := NewBleveIndexer(indexPath)
|
|
require.NoError(t, indexer1.Init())
|
|
|
|
var idx Indexer = indexer1
|
|
atomicIndexer.Store(&idx)
|
|
|
|
g := newGist(1, 1, 0, "persistent data survives restart")
|
|
require.NoError(t, indexer1.Add(g))
|
|
|
|
indexer1.Close()
|
|
atomicIndexer.Store(nil)
|
|
|
|
// Reopen at same path
|
|
indexer2 := NewBleveIndexer(indexPath)
|
|
require.NoError(t, indexer2.Init())
|
|
defer indexer2.Close()
|
|
|
|
idx = indexer2
|
|
atomicIndexer.Store(&idx)
|
|
defer atomicIndexer.Store(nil)
|
|
|
|
ids, total, _, err := indexer2.Search(SearchGistMetadata{Content: "persistent"}, 1, 1)
|
|
require.NoError(t, err)
|
|
require.Equal(t, uint64(1), total, "data should survive close+reopen")
|
|
require.Equal(t, uint(1), ids[0])
|
|
}
|