@@ -29,6 +29,7 @@ import (
2929type codeSearchResult struct {
3030 Filename string
3131 Content string
32+ Indexer string
3233}
3334
3435func TestMain (m * testing.M ) {
@@ -179,6 +180,7 @@ func testIndexer(name string, t *testing.T, indexer internal.Indexer) {
179180 {
180181 Filename : "ham.md" ,
181182 Content : "This is also not cheese" ,
183+ Indexer : "elastic_search" ,
182184 },
183185 },
184186 },
@@ -249,8 +251,18 @@ func testIndexer(name string, t *testing.T, indexer internal.Indexer) {
249251
250252 hits := make ([]codeSearchResult , 0 , len (res ))
251253
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+
252264 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" )
254266 }
255267
256268 for _ , hit := range res {
@@ -262,7 +274,7 @@ func testIndexer(name string, t *testing.T, indexer internal.Indexer) {
262274
263275 lastIndex := - 1
264276
265- for _ , expected := range kw . Results {
277+ for _ , expected := range expectedResults {
266278 index := slices .Index (hits , expected )
267279 if index == - 1 {
268280 assert .Failf (t , "Result not found" , "Expected %v in %v" , expected , hits )
0 commit comments