package bot

import (
	"testing"

	"myaibot/internal/tokens"

	"github.com/openai/openai-go"

	"myaibot/internal/config"
)

func TestEstimateTokensEmpty(t *testing.T) {
	if n := tokens.Count(""); n != 0 {
		t.Errorf("空串应为 0, got %d", n)
	}
}

func TestEstimateTokensKnown(t *testing.T) {
	cases := []struct {
		text string
		want int64
	}{
		{"hello", 1},
		{"hello world", 2},
		{"你好", 1},
		{"你是一个乐于助人的 AI 助手。", 12},
	}
	for _, c := range cases {
		if n := tokens.Count(c.text); n != c.want {
			t.Errorf("tokens.Count(%q) = %d, want %d", c.text, n, c.want)
		}
	}
}

func TestTokenize(t *testing.T) {
	ids := tokens.Tokenize("hello hello world")
	if len(ids) < 2 {
		t.Errorf("应有多个 token, got %v", ids)
	}
	seen := make(map[int64]bool)
	for _, tok := range ids {
		if tok.Text == "" {
			t.Errorf("token 文本不应为空: %+v", tok)
		}
		if seen[tok.ID] {
			t.Errorf("token 应去重: %v", ids)
		}
		seen[tok.ID] = true
	}
	if len(tokens.Tokenize("")) != 0 {
		t.Error("空串应返回空")
	}
	chinese := tokens.Tokenize("用户喜欢喝咖啡")
	if len(chinese) == 0 {
		t.Error("中文文本应产生 token")
	}
	if got := tokens.TokenIDs(ids); len(got) != len(ids) {
		t.Errorf("TokenIDs 数量 = %d, want %d", len(got), len(ids))
	}
}

func TestTokenizeCaseInsensitive(t *testing.T) {
	upper := tokens.Tokenize("Kevin 的生日")
	lower := tokens.Tokenize("kevin 的生日")
	if len(upper) != len(lower) {
		t.Fatalf("大小写 token 数量不一致: %v vs %v", upper, lower)
	}
	for i := range upper {
		if upper[i].ID != lower[i].ID {
			t.Errorf("token %d: %d != %d（大小写应编码为相同 id）", i, upper[i].ID, lower[i].ID)
		}
	}
	chinese := tokens.Tokenize("用户喜欢喝咖啡")
	if len(chinese) == 0 {
		t.Error("中文文本应产生 token")
	}
}

func TestContextStats(t *testing.T) {
	b := &Bot{
		systemPrompt: "你是一个乐于助人的 AI 助手。",
		provider: &config.Provider{
			Name: "p",
			Models: []config.ModelConfig{
				{Name: "m", ContextWindow: 1000},
			},
		},
		model: "m",
		history: []openai.ChatCompletionMessageParamUnion{
			openai.UserMessage("hello"),
			openai.AssistantMessage("world"),
		},
	}
	used, total := b.ContextStats()
	if total != 1000 {
		t.Errorf("total = %d, want 1000", total)
	}
	if used != 14 { // 系统提示 12 + hello 1 + world 1
		t.Errorf("used = %d, want 14", used)
	}
}