pkg/aflow/tool/codesearcher: add end-to-end tests

Update #6811

1 files changed, 7 insertions, 0 deletions

diff --git a/pkg/codesearch/codesearch.go b/pkg/codesearch/codesearch.go
index 0e8fe6014..4243f1013 100644
--- a/pkg/codesearch/codesearch.go
+++ b/pkg/codesearch/codesearch.go
@@ -13,8 +13,10 @@ import (
 	"strconv"
 	"strings"
 	"syscall"
+	"testing"
 
 	"github.com/google/syzkaller/pkg/aflow"
+	"github.com/google/syzkaller/pkg/clangtool/tooltest"
 	"github.com/google/syzkaller/pkg/osutil"
 )
 
@@ -153,6 +155,11 @@ func NewIndex(databaseFile string, srcDirs []string) (*Index, error) {
 	}, nil
 }
 
+func NewTestIndex(t *testing.T, dir string) *Index {
+	db := tooltest.LoadOutput[Database](t, dir)
+	return &Index{db, []string{dir}}
+}
+
 func (index *Index) Command(cmd string, args []string) (string, error) {
 	for _, meta := range Commands {
 		if cmd == meta.Name {