1. 1
2. 2
3. 3
4. 4
5. 5
6. 6
7. 7
8. 8
9. 9
10. 10
11. 11
12. 12
13. 13
14. 14
15. 15
16. 16
17. 17
18. 18
19. 19
20. 20
21. 21
22. 22
23. 23
24. 24
25. 25
26. 26
27. 27
28. 28
29. 29
30. 30
31. 31
32. 32
33. 33
34. 34
35. 35
36. 36
37. 37
38. 38
39. 39
40. 40
41. 41
42. 42
43. 43
44. 44
45. 45
46. 46
47. 47
48. 48
49. 49
50. 50
51. 51
52. 52
53. 53
54. 54
55. 55
56. 56
57. 57
58. 58
59. 59
60. 60
61. 61
62. 62
63. 63
64. 64
65. 65
66. 66
67. 67
68. 68
69. 69
70. 70
71. 71
72. 72
73. 73
74. 74
75. 75
76. 76
77. 77
78. 78
79. 79
80. 80
81. 81
82. 82
83. 83
84. 84
85. 85
86. 86
87. 87
88. 88

// 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"
	"testing"

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

func TestBleveIndexAndSearch(t *testing.T) {
	dir := t.TempDir()
	indexer := NewBleveIndexer(dir)
	defer indexer.Close()

	if _, err := indexer.Init(); err != nil {
		assert.Fail(t, "Unable to initialize 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)
	}
}