wehub-resource-sync
a06f331eb8
CI / benchmark (push) Has been skipped
install-script / posix-syntax (push) Successful in 6m1s
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
install-script / powershell-syntax (push) Has been cancelled
install-script / install (macos-14) (push) Has been cancelled
install-script / install (ubuntu-latest) (push) Has been cancelled
74 lines
2.6 KiB
Go
74 lines
2.6 KiB
Go
package hooks
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestPromptQuery(t *testing.T) {
|
|
cases := []struct {
|
|
name string
|
|
prompt string
|
|
want string
|
|
}{
|
|
{"empty", "", ""},
|
|
{"whitespace", " \n ", ""},
|
|
{"slash command", "/clear", ""},
|
|
{"short ack", "ok", ""},
|
|
{"two-letter word", "go", ""},
|
|
{"normal task", "fix the auth bug", "fix the auth bug"},
|
|
{"collapses whitespace", "fix\n\n the bug", "fix the bug"},
|
|
}
|
|
for _, tc := range cases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
require.Equal(t, tc.want, promptQuery(tc.prompt))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestPromptQueryTruncatesLongPrompts(t *testing.T) {
|
|
long := strings.Repeat("token ", 100) // ~600 chars
|
|
got := promptQuery(long)
|
|
require.LessOrEqual(t, len([]rune(got)), 240)
|
|
require.NotEmpty(t, got)
|
|
}
|
|
|
|
func TestBuildPromptInjection(t *testing.T) {
|
|
require.Equal(t, "", buildPromptInjection(nil), "no hits → no block")
|
|
|
|
hits := []grepSymbolHit{
|
|
{Name: "ValidateToken", Kind: "function", FilePath: "internal/auth/token.go", Line: 42},
|
|
{Name: "AuthMiddleware", Kind: "type", FilePath: "internal/auth/mw.go"}, // no line
|
|
}
|
|
block := buildPromptInjection(hits)
|
|
require.Contains(t, block, "relevant indexed symbols")
|
|
require.Contains(t, block, "`ValidateToken` (function) — internal/auth/token.go:42")
|
|
require.Contains(t, block, "`AuthMiddleware` (type) — internal/auth/mw.go")
|
|
// The follow-up routing points at the one-shot localization verb first,
|
|
// with the granular readers as the single-symbol fallback.
|
|
require.Contains(t, block, "`explore`")
|
|
require.Contains(t, block, "batch_symbols")
|
|
}
|
|
|
|
func TestBuildPromptInjectionCapsHits(t *testing.T) {
|
|
var hits []grepSymbolHit
|
|
for i := 0; i < 20; i++ {
|
|
hits = append(hits, grepSymbolHit{Name: "Sym", Kind: "function", FilePath: "f.go", Line: i + 1})
|
|
}
|
|
block := buildPromptInjection(hits)
|
|
require.Equal(t, maxInjectedHits, strings.Count(block, "- `Sym`"), "injection is capped")
|
|
}
|
|
|
|
// TestRunUserPromptSubmitTrivialPromptIsNoOp confirms a trivial prompt never
|
|
// even probes the daemon (and so produces no output).
|
|
func TestRunUserPromptSubmitTrivialPromptIsNoOp(t *testing.T) {
|
|
// A slash command produces an empty query → early return, no panic, no
|
|
// daemon dial. We assert it doesn't panic and returns cleanly.
|
|
runUserPromptSubmit([]byte(`{"hook_event_name":"UserPromptSubmit","prompt":"/clear"}`))
|
|
runUserPromptSubmit([]byte(`{"hook_event_name":"UserPromptSubmit","prompt":"ok"}`))
|
|
// Wrong event name is also a no-op.
|
|
runUserPromptSubmit([]byte(`{"hook_event_name":"SessionStart","prompt":"do something big"}`))
|
|
}
|