package agentnode

import (
	"strings"
	"testing"

	agentllm "github.com/LoveLosita/smartflow/backend/agent2/llm"
	agentmodel "github.com/LoveLosita/smartflow/backend/agent2/model"
)

// TestExtractExplicitLimitFromUser_Number 验证阿拉伯数字条数可以被正确提取。
func TestExtractExplicitLimitFromUser_Number(t *testing.T) {
	limit, ok := extractExplicitLimitFromUser("给我3个优先级低的任务")
	if !ok {
		t.Fatalf("期望识别到显式数量")
	}
	if limit != 3 {
		t.Fatalf("数量识别错误，期望=3 实际=%d", limit)
	}
}

// TestExtractExplicitLimitFromUser_ChineseNumber 验证中文数字也可以被正确提取。
func TestExtractExplicitLimitFromUser_ChineseNumber(t *testing.T) {
	limit, ok := extractExplicitLimitFromUser("前五个简单任务给我看看")
	if !ok {
		t.Fatalf("期望识别到中文数量")
	}
	if limit != 5 {
		t.Fatalf("数量识别错误，期望=5 实际=%d", limit)
	}
}

// TestExtractExplicitLimitFromUser_LaiYiGe 验证“来一个”这类口语表达也能命中数量提取。
func TestExtractExplicitLimitFromUser_LaiYiGe(t *testing.T) {
	limit, ok := extractExplicitLimitFromUser("来一个我的简单任务")
	if !ok {
		t.Fatalf("期望识别到‘来一个’的显式数量")
	}
	if limit != 1 {
		t.Fatalf("数量识别错误，期望=1 实际=%d", limit)
	}
}

// TestBuildTaskQueryFinalReply_RespectsLimit 验证最终回复严格遵守 plan.limit。
func TestBuildTaskQueryFinalReply_RespectsLimit(t *testing.T) {
	items := []agentmodel.TaskQueryItem{
		{ID: 1, Title: "任务1", PriorityLabel: "简单不重要", DeadlineAt: "2026-03-16 10:00"},
		{ID: 2, Title: "任务2", PriorityLabel: "简单不重要", DeadlineAt: "2026-03-17 10:00"},
		{ID: 3, Title: "任务3", PriorityLabel: "简单不重要", DeadlineAt: "2026-03-18 10:00"},
	}
	reply := buildTaskQueryFinalReply(items, agentmodel.TaskQueryPlan{Limit: 2}, "好的")
	if !strings.Contains(reply, "整理了 2 条任务") {
		t.Fatalf("回复未体现 limit=2，reply=%s", reply)
	}
	if strings.Contains(reply, "3. ") {
		t.Fatalf("回复中不应出现第 3 条，reply=%s", reply)
	}
}

// TestBuildTaskQueryFinalReply_NoDuplicateList 验证 llmReply 自带列表时不会与后端列表重复拼接。
func TestBuildTaskQueryFinalReply_NoDuplicateList(t *testing.T) {
	items := []agentmodel.TaskQueryItem{{ID: 1, Title: "任务1", PriorityLabel: "简单不重要", DeadlineAt: "2026-03-16 10:00"}}
	llmReply := "以下是你的任务：\n#1 任务1"
	reply := buildTaskQueryFinalReply(items, agentmodel.TaskQueryPlan{Limit: 1}, llmReply)
	if strings.Contains(reply, "以下是你的任务") {
		t.Fatalf("不应保留 llm 列表头，reply=%s", reply)
	}
	if !strings.Contains(reply, "整理了 1 条任务") {
		t.Fatalf("应保留后端确定性列表头，reply=%s", reply)
	}
}

// TestApplyRetryPatch_RespectExplicitLimit 验证显式数量存在时，反思补丁不能覆盖 limit。
func TestApplyRetryPatch_RespectExplicitLimit(t *testing.T) {
	plan := agentmodel.TaskQueryPlan{Limit: 1, SortBy: "deadline", Order: "asc"}
	limit := 10
	next := applyRetryPatch(plan, agentllm.TaskQueryRetryPatch{Limit: &limit}, 1)
	if next.Limit != 1 {
		t.Fatalf("显式数量锁应生效，期望=1 实际=%d", next.Limit)
	}
}