@@ -29,6 +29,7 @@ import (
29
29
type codeSearchResult struct {
30
30
Filename string
31
31
Content string
32
+ Indexer string
32
33
}
33
34
34
35
func TestMain (m * testing.M ) {
@@ -179,6 +180,7 @@ func testIndexer(name string, t *testing.T, indexer internal.Indexer) {
179
180
{
180
181
Filename : "ham.md" ,
181
182
Content : "This is also not cheese" ,
183
+ Indexer : "elastic_search" ,
182
184
},
183
185
},
184
186
},
@@ -249,8 +251,18 @@ func testIndexer(name string, t *testing.T, indexer internal.Indexer) {
249
251
250
252
hits := make ([]codeSearchResult , 0 , len (res ))
251
253
254
+ expectedResults := make ([]codeSearchResult , 0 , len (kw .Results ))
255
+ for _ , expected := range kw .Results {
256
+ if expected .Indexer == "" || expected .Indexer == name {
257
+ expectedResults = append (expectedResults , codeSearchResult {
258
+ Filename : expected .Filename ,
259
+ Content : expected .Content ,
260
+ })
261
+ }
262
+ }
263
+
252
264
if total > 0 {
253
- assert .NotEmpty (t , kw . Results , "The given scenario does not provide any expected results" )
265
+ assert .NotEmpty (t , expectedResults , "The given scenario does not provide any expected results" )
254
266
}
255
267
256
268
for _ , hit := range res {
@@ -262,7 +274,7 @@ func testIndexer(name string, t *testing.T, indexer internal.Indexer) {
262
274
263
275
lastIndex := - 1
264
276
265
- for _ , expected := range kw . Results {
277
+ for _ , expected := range expectedResults {
266
278
index := slices .Index (hits , expected )
267
279
if index == - 1 {
268
280
assert .Failf (t , "Result not found" , "Expected %v in %v" , expected , hits )
0 commit comments