wehub-resource-sync
72 lines
1.7 KiB
Go
72 lines
1.7 KiB
Go
package main
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestDetectClaudeCodeWebSearchFromFixture(t *testing.T) {
|
|
root := filepath.Join("..", "..", "..", "..", "temp", "1.json")
|
|
raw, errRead := os.ReadFile(root)
|
|
if errRead != nil {
|
|
t.Skipf("fixture not found: %v", errRead)
|
|
}
|
|
// Fixture is HTTP capture; extract JSON request body between first blank line after headers.
|
|
body := extractHTTPJSONBody(raw)
|
|
if len(body) == 0 {
|
|
t.Fatal("empty JSON body in fixture")
|
|
}
|
|
if !hasClaudeTypedWebSearchTool(body) {
|
|
t.Fatal("fixture should declare web_search_20250305")
|
|
}
|
|
if !looksLikeClaudeCodeWebSearchAssistant(body) {
|
|
t.Fatal("fixture should match Claude Code web search assistant heuristics")
|
|
}
|
|
if !isClaudeCodeBuiltinWebSearchRequest(body, true) {
|
|
t.Fatal("expected match with require_web_search_only=true")
|
|
}
|
|
query := extractClaudeWebSearchQuery(body)
|
|
if query == "" {
|
|
t.Fatal("expected non-empty search query")
|
|
}
|
|
if want := "北京天气 2026年6月16日"; query != want {
|
|
t.Fatalf("query = %q, want %q", query, want)
|
|
}
|
|
}
|
|
|
|
func extractHTTPJSONBody(raw []byte) []byte {
|
|
text := string(raw)
|
|
idx := 0
|
|
for {
|
|
next := findDoubleNewline(text, idx)
|
|
if next < 0 {
|
|
return nil
|
|
}
|
|
rest := trimLeft(text[next:])
|
|
if len(rest) > 0 && rest[0] == '{' {
|
|
return []byte(rest)
|
|
}
|
|
idx = next + 1
|
|
}
|
|
}
|
|
|
|
func findDoubleNewline(s string, from int) int {
|
|
for i := from; i+1 < len(s); i++ {
|
|
if s[i] == '\n' && s[i+1] == '\n' {
|
|
return i + 2
|
|
}
|
|
if s[i] == '\r' && i+3 < len(s) && s[i+1] == '\n' && s[i+2] == '\r' && s[i+3] == '\n' {
|
|
return i + 4
|
|
}
|
|
}
|
|
return -1
|
|
}
|
|
|
|
func trimLeft(s string) string {
|
|
for len(s) > 0 && (s[0] == '\r' || s[0] == '\n' || s[0] == ' ') {
|
|
s = s[1:]
|
|
}
|
|
return s
|
|
}
|