90 lines
3.9 KiB
Go
90 lines
3.9 KiB
Go
|
//go:build llm_script
|
|||
|
|
|
|||
|
|
package main
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"bytes"
|
|||
|
|
"os"
|
|||
|
|
"path/filepath"
|
|||
|
|
"strings"
|
|||
|
|
"testing"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func TestResolveMobileCloudPricingArticle(t *testing.T) {
|
|||
|
|
raw := `{"code":200,"data":{"children":[{"articleTitle":"其他文档","articleId":1,"articleContentPublished":"x"},{"children":[{"articleTitle":"预置模型服务-token按量计费","articleId":91592,"articleContentPublished":"64ec46cbfd7c535db501aff43df5a788"}]}]}}`
|
|||
|
|
articleID, contentPublished, err := resolveMobileCloudPricingArticle(raw)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("resolveMobileCloudPricingArticle 返回错误: %v", err)
|
|||
|
|
}
|
|||
|
|
if articleID != 91592 || contentPublished != "64ec46cbfd7c535db501aff43df5a788" {
|
|||
|
|
t.Fatalf("解析价格文章失败: articleID=%d contentPublished=%q", articleID, contentPublished)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestParseMobileCloudPricingHTMLBuildsRecords(t *testing.T) {
|
|||
|
|
raw, err := os.ReadFile(filepath.Join("testdata", "mobile_cloud_pricing_sample.html"))
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("读取 fixture 失败: %v", err)
|
|||
|
|
}
|
|||
|
|
records, err := parseMobileCloudPricingHTML(string(raw), "https://ecloud.10086.cn/op-help-center/doc/article/91592")
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("parseMobileCloudPricingHTML 返回错误: %v", err)
|
|||
|
|
}
|
|||
|
|
if len(records) != 8 {
|
|||
|
|
t.Fatalf("期望 8 条移动云价格记录，实际 %d", len(records))
|
|||
|
|
}
|
|||
|
|
recordMap := make(map[string]officialPricingRecord, len(records))
|
|||
|
|
for _, record := range records {
|
|||
|
|
recordMap[record.ModelID] = record
|
|||
|
|
}
|
|||
|
|
if recordMap["mobile-cloud-huabei-huhehaote-minimax-m2-5"].Region != "华北-呼和浩特" {
|
|||
|
|
t.Fatalf("华北 MiniMax region 错误: %+v", recordMap["mobile-cloud-huabei-huhehaote-minimax-m2-5"])
|
|||
|
|
}
|
|||
|
|
if recordMap["mobile-cloud-huabei-huhehaote-deepseek-v3-0324"].InputPrice != 2 || recordMap["mobile-cloud-huabei-huhehaote-deepseek-v3-0324"].OutputPrice != 8 {
|
|||
|
|
t.Fatalf("DeepSeek-V3-0324 价格错误: %+v", recordMap["mobile-cloud-huabei-huhehaote-deepseek-v3-0324"])
|
|||
|
|
}
|
|||
|
|
if recordMap["mobile-cloud-huabei-huhehaote-qwq-32b"].ProviderName != "Qwen" || recordMap["mobile-cloud-huabei-huhehaote-qwq-32b"].OutputPrice != 6 {
|
|||
|
|
t.Fatalf("QwQ-32B provider/价格错误: %+v", recordMap["mobile-cloud-huabei-huhehaote-qwq-32b"])
|
|||
|
|
}
|
|||
|
|
if recordMap["mobile-cloud-huabei-huhehaote-bge-m3"].ProviderName != "BAAI" || recordMap["mobile-cloud-huabei-huhehaote-bge-m3"].InputPrice != 0.5 || recordMap["mobile-cloud-huabei-huhehaote-bge-m3"].OutputPrice != 0.5 {
|
|||
|
|
t.Fatalf("bge-m3 平价 token 记录错误: %+v", recordMap["mobile-cloud-huabei-huhehaote-bge-m3"])
|
|||
|
|
}
|
|||
|
|
cosyVoice, ok := recordMap["mobile-cloud-huabei-huhehaote-cosyvoice"]
|
|||
|
|
if !ok {
|
|||
|
|
t.Fatalf("缺少 CosyVoice 语音计费记录")
|
|||
|
|
}
|
|||
|
|
if cosyVoice.ProviderName != "Alibaba" || cosyVoice.Modality != "audio" || cosyVoice.PricingMode != "flat" || cosyVoice.FlatPrice != 2 || cosyVoice.PriceUnit != "10k_characters" {
|
|||
|
|
t.Fatalf("CosyVoice 语音计费记录错误: %+v", cosyVoice)
|
|||
|
|
}
|
|||
|
|
senseVoice, ok := recordMap["mobile-cloud-huabei-huhehaote-sensevoice"]
|
|||
|
|
if !ok {
|
|||
|
|
t.Fatalf("缺少 SenseVoice 语音计费记录")
|
|||
|
|
}
|
|||
|
|
if senseVoice.ProviderName != "Alibaba" || senseVoice.Modality != "audio" || senseVoice.PricingMode != "flat" || senseVoice.FlatPrice != 0.0007 || senseVoice.PriceUnit != "second" {
|
|||
|
|
t.Fatalf("SenseVoice 语音计费记录错误: %+v", senseVoice)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func TestRunMobileCloudPricingImportDryRunPrintsSummary(t *testing.T) {
|
|||
|
|
var out bytes.Buffer
|
|||
|
|
err := runMobileCloudPricingImport(mobileCloudPricingImportConfig{
|
|||
|
|
OutlineTreeURL: defaultMobileCloudOutlineTreeURL,
|
|||
|
|
Fixture: filepath.Join("testdata", "mobile_cloud_pricing_sample.html"),
|
|||
|
|
DryRun: true,
|
|||
|
|
}, nil, &out)
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("runMobileCloudPricingImport 返回错误: %v", err)
|
|||
|
|
}
|
|||
|
|
output := out.String()
|
|||
|
|
for _, want := range []string{
|
|||
|
|
"source=mobile-cloud-pricing-import",
|
|||
|
|
"models=8",
|
|||
|
|
"operator=Mobile Cloud",
|
|||
|
|
"dry_run=true",
|
|||
|
|
} {
|
|||
|
|
if !strings.Contains(output, want) {
|
|||
|
|
t.Fatalf("输出缺少 %q，实际: %q", want, output)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|