//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)
		}
	}
}