170 lines
4.3 KiB
Go
170 lines
4.3 KiB
Go
package collect
|
||
|
||
import (
|
||
"context"
|
||
"geo/internal/collect"
|
||
"geo/internal/config"
|
||
"testing"
|
||
"time"
|
||
|
||
"github.com/gofiber/fiber/v2/log"
|
||
)
|
||
|
||
var (
|
||
doubaoCfg, _ = config.LoadConfig()
|
||
|
||
doubaoManager = collect.NewCollectManager(context.Background(), doubaoCfg, log.DefaultLogger())
|
||
)
|
||
|
||
// TestDoubaoCollector_WaitLogin 测试豆包登录功能
|
||
func TestDoubaoCollector_WaitLogin(t *testing.T) {
|
||
if testing.Short() {
|
||
t.Skip("跳过需要浏览器交互的测试")
|
||
}
|
||
|
||
params := &collect.CollectParams{
|
||
Headless: false, // 显示浏览器窗口以便扫码登录
|
||
RequestID: "test_doubao_login_001",
|
||
Platform: "doubao",
|
||
}
|
||
|
||
t.Log("开始测试豆包登录...")
|
||
t.Log("请在打开的浏览器窗口中完成豆包账号登录（扫码或输入账号密码）")
|
||
|
||
success, msg := doubaoManager.WaitLogin("doubao", params)
|
||
|
||
if !success {
|
||
t.Errorf("豆包登录失败: %s", msg)
|
||
return
|
||
}
|
||
|
||
t.Logf("豆包登录成功: %s", msg)
|
||
t.Log("Cookie已保存，后续测试可以使用已登录状态")
|
||
}
|
||
|
||
// TestDoubaoCollector_AskQuestion 测试豆包提问功能
|
||
// 注意：此测试需要有效的登录状态
|
||
func TestDoubaoCollector_AskQuestion(t *testing.T) {
|
||
if testing.Short() {
|
||
t.Skip("跳过需要浏览器交互的测试")
|
||
}
|
||
|
||
// 设置收集参数
|
||
params := &collect.CollectParams{
|
||
Headless: false, // 显示浏览器以便调试
|
||
RequestID: "test_doubao_001",
|
||
Platform: "doubao",
|
||
}
|
||
|
||
// 定义提问内容
|
||
question := "云案场怎么样"
|
||
t.Logf("向豆包提问: %s", question)
|
||
|
||
// 调用管理器提问并获取答案
|
||
result, err := doubaoManager.AskQuestion("doubao", params, question)
|
||
if err != nil {
|
||
t.Errorf("提问失败: %v", err)
|
||
return
|
||
}
|
||
|
||
t.Logf("获取到答案:\n%s", result.Answer)
|
||
|
||
// 验证答案非空
|
||
if len(result.Answer) == 0 {
|
||
t.Error("答案为空")
|
||
}
|
||
}
|
||
|
||
// TestDoubaoCollector_MultipleQuestions 测试豆包多次提问
|
||
func TestDoubaoCollector_MultipleQuestions(t *testing.T) {
|
||
if testing.Short() {
|
||
t.Skip("跳过需要浏览器交互的测试")
|
||
}
|
||
|
||
params := &collect.CollectParams{
|
||
Headless: false,
|
||
RequestID: "test_doubao_multi_001",
|
||
Platform: "doubao",
|
||
}
|
||
|
||
questions := []string{
|
||
"什么是人工智能？",
|
||
"Python和Go的区别是什么？",
|
||
"如何学习编程？",
|
||
"中国有哪些著名的旅游景点？",
|
||
"健康饮食的基本原则是什么？",
|
||
"区块链技术的主要应用场景有哪些？",
|
||
"如何提高英语口语能力？",
|
||
"云计算的优势和劣势分别是什么？",
|
||
"环境保护的重要性体现在哪些方面？",
|
||
"未来十年最有前景的行业有哪些？",
|
||
}
|
||
|
||
for i, question := range questions {
|
||
t.Logf("========== 第 %d/%d 个问题 ==========", i+1, len(questions))
|
||
t.Logf("问题: %s", question)
|
||
|
||
result, err := doubaoManager.AskQuestion("doubao", params, question)
|
||
if err != nil {
|
||
t.Errorf("第 %d 个问题提问失败: %v", i+1, err)
|
||
continue
|
||
}
|
||
|
||
t.Logf("✓ 第 %d 个问题的答案长度: %d 字符", i+1, len(result.Answer))
|
||
t.Logf("答案预览: %s...", result.Answer[:min(100, len(result.Answer))])
|
||
|
||
// 每个问题之间等待一下
|
||
if i < len(questions)-1 {
|
||
t.Log("等待3秒后继续下一个问题...")
|
||
time.Sleep(3 * time.Second)
|
||
}
|
||
}
|
||
|
||
t.Logf("========== 测试完成 ==========")
|
||
t.Logf("成功回答了 %d 个问题", len(questions))
|
||
}
|
||
|
||
// TestDoubaoCollector_SpeedTest 测试优化后的速度
|
||
func TestDoubaoCollector_SpeedTest(t *testing.T) {
|
||
if testing.Short() {
|
||
t.Skip("跳过需要浏览器交互的测试")
|
||
}
|
||
|
||
params := &collect.CollectParams{
|
||
Headless: false,
|
||
RequestID: "test_doubao_speed",
|
||
Platform: "doubao",
|
||
}
|
||
|
||
question := "1+1等于几？"
|
||
|
||
startTime := time.Now()
|
||
t.Logf("开始提问: %s", question)
|
||
|
||
result, err := doubaoManager.AskQuestion("doubao", params, question)
|
||
if err != nil {
|
||
t.Fatalf("提问失败: %v", err)
|
||
}
|
||
|
||
elapsed := time.Since(startTime)
|
||
t.Logf("✓ 完成时间: %v", elapsed)
|
||
t.Logf("答案长度: %d 字符", len(result.Answer))
|
||
t.Logf("分享链接: %s", result.ShareLink)
|
||
t.Logf("答案预览: %s...", result.Answer[:min(100, len(result.Answer))])
|
||
|
||
// 如果超过60秒，说明还有优化空间
|
||
if elapsed > 60*time.Second {
|
||
t.Logf("⚠️ 警告: 耗时较长 (%v)，可能需要进一步优化", elapsed)
|
||
} else {
|
||
t.Logf("✅ 速度正常")
|
||
}
|
||
}
|
||
|
||
// min 辅助函数
|
||
func min(a, b int) int {
|
||
if a < b {
|
||
return a
|
||
}
|
||
return b
|
||
}
|