Files
ai-agent/internal/pkg/toolx/builtin_tools_test.go
T
2026-05-25 12:06:15 +08:00

65 lines
2.2 KiB
Go

package toolx
import (
"testing"
"cs-agent/internal/pkg/i18nx"
)
func TestResolveToolMetadata(t *testing.T) {
item := ResolveToolMetadata("builtin/create_ticket_with_confirmation", "")
if item.ToolCode != GraphCreateTicketConfirm.Code {
t.Fatalf("unexpected tool code: %s", item.ToolCode)
}
if item.ServerCode != GraphCreateTicketConfirm.ServerCode {
t.Fatalf("unexpected server code: %s", item.ServerCode)
}
if item.ToolName != GraphCreateTicketConfirm.Name {
t.Fatalf("unexpected tool name: %s", item.ToolName)
}
if item.SourceType != GraphCreateTicketConfirm.SourceType {
t.Fatalf("unexpected source type: %s", item.SourceType)
}
}
func TestResolveToolMetadataFallsBackToName(t *testing.T) {
item := ResolveToolMetadata("mcp/demo_tool", "demo_tool")
if item.ToolCode != "mcp/demo_tool" {
t.Fatalf("unexpected tool code: %s", item.ToolCode)
}
if item.ServerCode != "" {
t.Fatalf("unexpected server code: %s", item.ServerCode)
}
if item.ToolName != "demo_tool" {
t.Fatalf("unexpected tool name: %s", item.ToolName)
}
if item.SourceType != "mcp" {
t.Fatalf("unexpected source type: %s", item.SourceType)
}
}
func TestRegisteredToolTextUsesEnglishLocale(t *testing.T) {
title := GetRegisteredToolTitleLocale(GraphCreateTicketConfirm.Code, i18nx.LocaleEnUS)
if title != "Create Ticket With Confirmation" {
t.Fatalf("unexpected english title: %q", title)
}
description := GetRegisteredToolDescriptionLocale(GraphCreateTicketConfirm.Code, i18nx.LocaleEnUS)
want := "Guides ticket creation with parameter preparation, customer confirmation, actual ticket creation, and final result delivery."
if description != want {
t.Fatalf("unexpected english description: %q", description)
}
}
func TestRegisteredToolTextKeepsChineseLocale(t *testing.T) {
title := GetRegisteredToolTitleLocale(GraphCreateTicketConfirm.Code, i18nx.LocaleZhCN)
if title != GraphCreateTicketConfirm.Title {
t.Fatalf("unexpected chinese title: %q", title)
}
description := GetRegisteredToolDescriptionLocale(GraphCreateTicketConfirm.Code, i18nx.LocaleZhCN)
if description != GraphCreateTicketConfirm.Description {
t.Fatalf("unexpected chinese description: %q", description)
}
}