wehub-resource-sync
a06f331eb8
install-script / powershell-syntax (push) Waiting to run
CI / benchmark (push) Has been skipped
install-script / posix-syntax (push) Successful in 6m1s
install-script / install (macos-14) (push) Waiting to run
install-script / install (ubuntu-latest) (push) Waiting to run
CI / build-onnx (push) Failing after 6m43s
init-smoke / dry-run (push) Failing after 15m57s
security / govulncheck (push) Has been cancelled
security / trivy-fs (push) Has been cancelled
CI / test (1.26, ubuntu-latest) (push) Has been cancelled
Scorecard supply-chain security / Scorecard analysis (push) Has been cancelled
CI / test (1.26, macos-latest) (push) Has been cancelled
CI / build-windows (push) Has been cancelled
CI / lint (push) Has been cancelled
62 lines
2.2 KiB
Go
62 lines
2.2 KiB
Go
package mcp
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
mcplib "github.com/mark3labs/mcp-go/mcp"
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/zzet/gortex/internal/config"
|
|
"github.com/zzet/gortex/internal/graph"
|
|
"github.com/zzet/gortex/internal/indexer"
|
|
"github.com/zzet/gortex/internal/query"
|
|
)
|
|
|
|
// TestSymbolNotFound_DidYouMean pins the recovery hint on a missed symbol
|
|
// id: a stale/typo'd name surfaces the nearest symbol in the same file, a
|
|
// right-name/wrong-path id surfaces where the symbol actually lives, and a
|
|
// genuinely unknown id in an empty path still errors cleanly with no hint.
|
|
func TestSymbolNotFound_DidYouMean(t *testing.T) {
|
|
dir := t.TempDir()
|
|
require.NoError(t, os.WriteFile(filepath.Join(dir, "main.go"),
|
|
[]byte("package app\n\nfunc AlphaHandler() {}\n\nfunc BetaHandler() {}\n"), 0o644))
|
|
|
|
g := graph.New()
|
|
reg := testRegistry()
|
|
cfg := config.Default()
|
|
idx := indexer.New(g, reg, cfg.Index, zap.NewNop())
|
|
_, err := idx.Index(dir)
|
|
require.NoError(t, err)
|
|
eng := query.NewEngine(g)
|
|
srv := NewServer(eng, g, idx, nil, zap.NewNop(), nil)
|
|
|
|
errText := func(res *mcplib.CallToolResult) string {
|
|
require.True(t, res.IsError)
|
|
return res.Content[0].(mcplib.TextContent).Text
|
|
}
|
|
|
|
// Typo'd name in the right file → nearest symbol in that file.
|
|
typo := errText(callTool(t, srv, "get_symbol_source", map[string]any{"id": "main.go::AlphaHandlr"}))
|
|
require.Contains(t, typo, "symbol not found")
|
|
require.Contains(t, typo, "did you mean")
|
|
require.Contains(t, typo, "AlphaHandler")
|
|
|
|
// Right name, wrong path → the id where the symbol actually lives.
|
|
wrongPath := errText(callTool(t, srv, "get_symbol_source", map[string]any{"id": "nope.go::BetaHandler"}))
|
|
require.Contains(t, wrongPath, "BetaHandler")
|
|
|
|
// edit_symbol shares the hint.
|
|
editMiss := errText(callTool(t, srv, "edit_symbol", map[string]any{
|
|
"id": "main.go::AlphaHandlr", "old_source": "a", "new_source": "b",
|
|
}))
|
|
require.Contains(t, editMiss, "did you mean")
|
|
|
|
// Unknown name in a path with no indexed symbols → clean error, no hint.
|
|
clean := errText(callTool(t, srv, "get_symbol_source", map[string]any{"id": "ghost.go::Nonexistent"}))
|
|
require.Contains(t, clean, "symbol not found")
|
|
require.NotContains(t, clean, "did you mean")
|
|
}
|