opencode/picoclaw
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update mock tool registry tests for consistency

Browse source
This commit is contained in:
stevef 2026-04-14 10:23:56 +02:00
parent bfdc9734f5
commit 10f568f87e
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
pkg/tools/registry_test.go
View file

@ -764,14 +764,14 @@ func TestToolRegistry_Filter_SupportsPrefix(t *testing.T) {
r := NewToolRegistry() r := NewToolRegistry()
r.Register(newMockTool("read_file", "core tool")) r.Register(newMockTool("read_file", "core tool"))
r.Register(newMockTool("write_file", "core tool")) r.Register(newMockTool("write_file", "core tool"))
r.Register(newMockTool("mcp_monday_get_items", "mcp tool")) r.Register(newMockTool("mcp_github_get_items", "mcp tool"))
r.Register(newMockTool("mcp_harvest_get_entries", "mcp tool")) r.Register(newMockTool("mcp_google_get_entries", "mcp tool"))
r.Register(newMockTool("tool_search_regex", "discovery tool")) r.Register(newMockTool("tool_search_regex", "discovery tool"))
whitelist := []string{"read_file", "monday", "search"} whitelist := []string{"read_file", "github", "search"}
r.Filter(whitelist, true) r.Filter(whitelist, true)
// expected: read_file (exact), mcp_monday_get_items (mcp_monday_ prefix), tool_search_regex (tool_search_ prefix) // expected: read_file (exact), mcp_github_get_items (mcp_github_ prefix), tool_search_regex (tool_search_ prefix)
if r.Count() != 3 { if r.Count() != 3 {
t.Errorf("expected 3 tools after filtering, got %d: %v", r.Count(), r.List()) t.Errorf("expected 3 tools after filtering, got %d: %v", r.Count(), r.List())
} }
@ -779,7 +779,7 @@ func TestToolRegistry_Filter_SupportsPrefix(t *testing.T) {
allowed := r.List() allowed := r.List()
expected := map[string]bool{ expected := map[string]bool{
"read_file": true, "read_file": true,
"mcp_monday_get_items": true, "mcp_github_get_items": true,
"tool_search_regex": true, "tool_search_regex": true,
} }