forked from niuniu/llm-intelligence
99 lines
2.6 KiB
Go
99 lines
2.6 KiB
Go
|
//go:build llm_script
|
|||
|
|
|
|||
|
|
package main
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"net/http"
|
|||
|
|
"net/http/httptest"
|
|||
|
|
"os"
|
|||
|
|
"path/filepath"
|
|||
|
|
"strings"
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func TestParseTencentCatalogExtractsPlansAndModels(t *testing.T) {
|
|||
|
|
raw, err := os.ReadFile(filepath.Join("testdata", "tencent_token_plan_sample.txt"))
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("读取样例失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
catalog, err := parseTencentCatalog(string(raw))
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("parseTencentCatalog 失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if catalog.UpdatedAt != "2026-04-27 17:18:02" {
|
|||
|
|
t.Fatalf("更新时间错误: %q", catalog.UpdatedAt)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if len(catalog.Plans) != 8 {
|
|||
|
|
t.Fatalf("期望 8 个套餐，实际 %d", len(catalog.Plans))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if len(catalog.Models) != 11 {
|
|||
|
|
t.Fatalf("期望 11 个模型，实际 %d", len(catalog.Models))
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
firstPlan := catalog.Plans[0]
|
|||
|
|
if firstPlan.Series != "通用 Token Plan" {
|
|||
|
|
t.Fatalf("套餐系列错误: %q", firstPlan.Series)
|
|||
|
|
}
|
|||
|
|
if firstPlan.Tier != "Lite" {
|
|||
|
|
t.Fatalf("套餐档位错误: %q", firstPlan.Tier)
|
|||
|
|
}
|
|||
|
|
if firstPlan.Price != "39元/月" {
|
|||
|
|
t.Fatalf("套餐价格错误: %q", firstPlan.Price)
|
|||
|
|
}
|
|||
|
|
if firstPlan.Quota != "3500万 Tokens" {
|
|||
|
|
t.Fatalf("套餐额度错误: %q", firstPlan.Quota)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
lastModel := catalog.Models[len(catalog.Models)-1]
|
|||
|
|
if lastModel.Name != "Hy3 preview" {
|
|||
|
|
t.Fatalf("最后一个模型错误: %q", lastModel.Name)
|
|||
|
|
}
|
|||
|
|
if lastModel.ModelID != "hy3-preview" {
|
|||
|
|
t.Fatalf("最后一个模型 ID 错误: %q", lastModel.ModelID)
|
|||
|
|
}
|
|||
|
|
if lastModel.ContextLength != 262144 {
|
|||
|
|
t.Fatalf("Hy3 preview 上下文长度错误: %d", lastModel.ContextLength)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestRunTencentCatalogDryRunPrintsSummary(t *testing.T) {
|
|||
|
|
raw, err := os.ReadFile(filepath.Join("testdata", "tencent_token_plan_sample.txt"))
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("读取样例失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|||
|
|
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
|||
|
|
_, _ = w.Write([]byte("<html><body><main><pre>" + string(raw) + "</pre></main></body></html>"))
|
|||
|
|
}))
|
|||
|
|
defer server.Close()
|
|||
|
|
|
|||
|
|
var out bytes.Buffer
|
|||
|
|
err = runTencentCatalog(fetchTencentCatalogConfig{
|
|||
|
|
URL: server.URL,
|
|||
|
|
DryRun: true,
|
|||
|
|
Timeout: defaultTencentCatalogTimeout,
|
|||
|
|
}, server.Client(), &out)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("runTencentCatalog 失败: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
output := out.String()
|
|||
|
|
for _, want := range []string{
|
|||
|
|
"source=tencent-public-catalog",
|
|||
|
|
"plans=8",
|
|||
|
|
"models=11",
|
|||
|
|
"series=Hy Token Plan:4,通用 Token Plan:4",
|
|||
|
|
"updated_at=2026-04-27 17:18:02",
|
|||
|
|
} {
|
|||
|
|
if !strings.Contains(output, want) {
|
|||
|
|
t.Fatalf("输出缺少 %q，实际: %q", want, output)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|