mirror of
https://github.com/sourcegraph/sourcegraph.git
synced 2026-02-06 18:51:59 +00:00
d3df71ef98
Closes https://linear.app/sourcegraph/issue/GRAPH-726/test-syntactic-and-search-based-usages Testing just the search-based usages just requires mocking the SearchClient, which works out nicely. ## Test plan The whole PR is just a test
96 lines
2.9 KiB
Go
96 lines
2.9 KiB
Go
// This file does not contain tests for utils.go, instead it contains utils
|
|
// for setting up the test environment for our codenav tests.
|
|
package codenav
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
genslices "github.com/life4/genesis/slices"
|
|
"github.com/sourcegraph/scip/bindings/go/scip"
|
|
|
|
"github.com/sourcegraph/sourcegraph/internal/search"
|
|
searchClient "github.com/sourcegraph/sourcegraph/internal/search/client"
|
|
"github.com/sourcegraph/sourcegraph/internal/search/result"
|
|
"github.com/sourcegraph/sourcegraph/internal/search/streaming"
|
|
)
|
|
|
|
// Generates a fake scip.Range that is easy to tell from other ranges
|
|
func scipRange(x int32) scip.Range {
|
|
return scip.NewRangeUnchecked([]int32{x, x, x})
|
|
}
|
|
|
|
func scipToResultPosition(p scip.Position) result.Location {
|
|
return result.Location{
|
|
Line: int(p.Line),
|
|
Column: int(p.Character),
|
|
}
|
|
}
|
|
|
|
func scipToResultRange(r scip.Range) result.Range {
|
|
return result.Range{
|
|
Start: scipToResultPosition(r.Start),
|
|
End: scipToResultPosition(r.End),
|
|
}
|
|
}
|
|
|
|
// scipToSymbolMatch "reverse engineers" the lsp.Range function on result.Symbol
|
|
func scipToSymbolMatch(r scip.Range) *result.SymbolMatch {
|
|
return &result.SymbolMatch{
|
|
Symbol: result.Symbol{
|
|
Line: int(r.Start.Line + 1),
|
|
Character: int(r.Start.Character),
|
|
Name: strings.Repeat("a", int(r.End.Character-r.Start.Character)),
|
|
}}
|
|
}
|
|
|
|
type FakeSearchBuilder struct {
|
|
fileMatches []result.Match
|
|
symbolMatches []result.Match
|
|
}
|
|
|
|
func FakeSearchClient() FakeSearchBuilder {
|
|
return FakeSearchBuilder{
|
|
fileMatches: []result.Match{},
|
|
symbolMatches: make([]result.Match, 0),
|
|
}
|
|
}
|
|
|
|
func (b FakeSearchBuilder) WithFile(file string, ranges ...scip.Range) FakeSearchBuilder {
|
|
b.fileMatches = append(b.fileMatches, &result.FileMatch{
|
|
File: result.File{Path: file},
|
|
ChunkMatches: result.ChunkMatches{{
|
|
Ranges: genslices.Map(ranges, scipToResultRange),
|
|
}},
|
|
})
|
|
return b
|
|
}
|
|
|
|
func (b FakeSearchBuilder) WithSymbols(file string, ranges ...scip.Range) FakeSearchBuilder {
|
|
b.symbolMatches = append(b.symbolMatches, &result.FileMatch{
|
|
File: result.File{Path: file},
|
|
Symbols: genslices.Map(ranges, scipToSymbolMatch),
|
|
})
|
|
return b
|
|
}
|
|
|
|
func (b FakeSearchBuilder) Build() searchClient.SearchClient {
|
|
mockSearchClient := searchClient.NewMockSearchClient()
|
|
mockSearchClient.PlanFunc.SetDefaultHook(func(_ context.Context, _ string, _ *string, query string, _ search.Mode, _ search.Protocol, _ *int32) (*search.Inputs, error) {
|
|
return &search.Inputs{OriginalQuery: query}, nil
|
|
})
|
|
mockSearchClient.ExecuteFunc.SetDefaultHook(func(_ context.Context, s streaming.Sender, i *search.Inputs) (*search.Alert, error) {
|
|
if strings.Contains(i.OriginalQuery, "type:file") {
|
|
s.Send(streaming.SearchEvent{
|
|
Results: b.fileMatches,
|
|
})
|
|
} else if strings.Contains(i.OriginalQuery, "type:symbol") {
|
|
s.Send(streaming.SearchEvent{
|
|
Results: b.symbolMatches,
|
|
})
|
|
}
|
|
return nil, nil
|
|
})
|
|
return mockSearchClient
|
|
}
|