package collect

import (
	"context"
	"geo/internal/collect"
	"geo/internal/config"
	"github.com/gofiber/fiber/v2/log"
	"testing"
)

var (
	cfg, _ = config.LoadConfig()

	manager = collect.NewCollectManager(context.Background(), cfg, log.DefaultLogger())
)

// TestCollectManager_Basic 测试收集管理器的基本功能
func TestCollectManager_Basic(t *testing.T) {

	// 测试列出平台
	platforms := manager.ListPlatforms()
	t.Logf("支持的平台: %v", platforms)

	if len(platforms) != 4 {
		t.Errorf("期望4个平台，实际: %d", len(platforms))
	}
	// 测试获取收集器
	for _, platform := range platforms {
		params := &collect.CollectParams{
			Headless:  true,
			RequestID: "test_req",
			Platform:  platform,
		}

		collector, err := manager.GetCollector(platform, params)
		if err != nil {
			t.Errorf("获取%s收集器失败: %v", platform, err)
			continue
		}

		if collector == nil {
			t.Errorf("%s收集器为nil", platform)
		}

		t.Logf("成功创建%s收集器", platform)
	}
}

// TestWenxinCollector_WaitLogin 测试文心一言登录功能
func TestWenxinCollector_WaitLogin(t *testing.T) {
	if testing.Short() {
		t.Skip("跳过需要浏览器交互的测试")
	}

	params := &collect.CollectParams{
		Headless:  false, // 显示浏览器窗口以便扫码登录
		RequestID: "test_wenxin_login_001",
		Platform:  "wenxin",
	}

	t.Log("开始测试文心一言登录...")
	t.Log("请在打开的浏览器窗口中完成百度账号登录（扫码或输入账号密码）")

	success, msg := manager.WaitLogin("wenxin", params)

	if !success {
		t.Errorf("文心一言登录失败: %s", msg)
		return
	}

	t.Logf("文心一言登录成功: %s", msg)
	t.Log("Cookie已保存，后续测试可以使用已登录状态")
}

// TestWenxinCollector_AskQuestion 测试文心一言提问功能
// 注意：此测试需要有效的登录状态
func TestWenxinCollector_AskQuestion(t *testing.T) {
	if testing.Short() {
		t.Skip("跳过需要浏览器交互的测试")
	}

	// 设置收集参数
	params := &collect.CollectParams{
		Headless:  true, // 显示浏览器以便调试
		RequestID: "test_wenxin_001",
		Platform:  "wenxin",
	}

	// 定义提问内容
	question := "四川房地产软件排名"
	t.Logf("向文心一言提问: %s", question)

	// 调用管理器提问并获取答案
	result, err := manager.AskQuestion("wenxin", params, question)
	if err != nil {
		t.Errorf("提问失败: %v", err)
		return
	}

	t.Logf("获取到答案:\n%s", result.Answer)
	t.Logf("分享链接: %s", result.ShareLink)

	// 验证答案非空
	if len(result.Answer) == 0 {
		t.Error("答案为空")
	}
}