githaven-fork/modules/indexer/issues/bleve_test.go

// Copyright 2018 The Gitea Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.

package issues

import (
	"context"
	"os"
	"testing"

	"code.gitea.io/gitea/modules/util"

	"github.com/stretchr/testify/assert"
)

func TestBleveIndexAndSearch(t *testing.T) {
	dir, err := os.MkdirTemp("", "bleve.index")
	assert.NoError(t, err)
	if err != nil {
		assert.Fail(t, "Unable to create temporary directory")
		return
	}
	defer util.RemoveAll(dir)
	indexer := NewBleveIndexer(dir)
	defer indexer.Close()

	if _, err := indexer.Init(); err != nil {
		assert.Fail(t, "Unable to initialise bleve indexer: %v", err)
		return
	}

	err = indexer.Index([]*IndexerData{
		{
			ID:      1,
			RepoID:  2,
			Title:   "Issue search should support Chinese",
			Content: "As title",
			Comments: []string{
				"test1",
				"test2",
			},
		},
		{
			ID:      2,
			RepoID:  2,
			Title:   "CJK support could be optional",
			Content: "Chinese Korean and Japanese should be supported but I would like it's not enabled by default",
			Comments: []string{
				"LGTM",
				"Good idea",
			},
		},
	})
	assert.NoError(t, err)

	keywords := []struct {
		Keyword string
		IDs     []int64
	}{
		{
			Keyword: "search",
			IDs:     []int64{1},
		},
		{
			Keyword: "test1",
			IDs:     []int64{1},
		},
		{
			Keyword: "test2",
			IDs:     []int64{1},
		},
		{
			Keyword: "support",
			IDs:     []int64{1, 2},
		},
		{
			Keyword: "chinese",
			IDs:     []int64{1, 2},
		},
		{
			Keyword: "help",
			IDs:     []int64{},
		},
	}

	for _, kw := range keywords {
		res, err := indexer.Search(context.TODO(), kw.Keyword, []int64{2}, 10, 0)
		assert.NoError(t, err)

		ids := make([]int64, 0, len(res.Hits))
		for _, hit := range res.Hits {
			ids = append(ids, hit.ID)
		}
		assert.ElementsMatch(t, kw.IDs, ids)
	}
}