package console

import (
	"testing"

	"yunyan/comm"
)

// normalizeAgent 是落库前最后一道清洗：解绑过的 MCP 服务不能把工具白名单留在库里，
// 否则重新勾上该服务时会冒出上一轮的选择，运营看不见、客户端却会照着执行。
func TestNormalizeAgentMcpTools(t *testing.T) {
	a := &AgentConfig{
		McpSvcIds: []string{" mcp_map ", "mcp_music", "mcp_map", ""},
		McpTools: map[string][]string{
			"mcp_map":   {"geocode", " geocode ", "", "route"},
			"mcp_music": {},         // 空列表 = 全放开，不该留在库里
			"mcp_gone":  {"search"}, // 已解绑的服务，残留必须清掉
		},
	}
	normalizeAgent(a)

	if len(a.McpSvcIds) != 2 || a.McpSvcIds[0] != "mcp_map" || a.McpSvcIds[1] != "mcp_music" {
		t.Fatalf("MCP 引用清洗错：%v", a.McpSvcIds)
	}
	if _, ok := a.McpTools["mcp_gone"]; ok {
		t.Fatalf("解绑服务的工具白名单没清掉：%v", a.McpTools)
	}
	if _, ok := a.McpTools["mcp_music"]; ok {
		t.Fatalf("空白名单应整条删掉（空=全放开）：%v", a.McpTools)
	}
	got := a.McpTools["mcp_map"]
	if len(got) != 2 || got[0] != "geocode" || got[1] != "route" {
		t.Fatalf("工具名去空白/去重错：%v", got)
	}
}

// 没配过工具白名单的老 agent（mcp_tools 为 nil）不能因为清洗炸掉，也不该凭空长出条目。
func TestNormalizeAgentMcpToolsNil(t *testing.T) {
	a := &AgentConfig{McpSvcIds: []string{"mcp_map"}}
	normalizeAgent(a)
	if a.McpTools == nil || len(a.McpTools) != 0 {
		t.Fatalf("nil 白名单应清洗成空 map：%v", a.McpTools)
	}
}

// id 前缀按类型走，肉眼能看出这是哪类 agent；未知类型兜底成 agent。
func TestAgentIdPrefix(t *testing.T) {
	for typ, want := range map[string]string{
		comm.AgentTypeChat:    "chat",
		comm.AgentTypeStsChat: "sts",
		"":                    "agent",
		"whatever":            "agent",
	} {
		if got := agentIdPrefix(typ); got != want {
			t.Fatalf("type=%q 前缀应为 %s，实际 %s", typ, want, got)
		}
	}
}