Files
NianAIGC/backend/internal/billing/defaults_test.go
2026-08-25 14:06:04 +08:00

38 lines
1.2 KiB
Go

package billing
import "testing"
func TestDefaultBillingPriceRulesFreezeOfficialCatalogMatchKeys(t *testing.T) {
rules := DefaultBillingPriceRules()
want := map[string]bool{
"base-volcengine-jimeng-seedream46": false,
"base-evolink-gpt-image-2": false,
"base-bailian-wan27-image-pro": false,
"base-seedream-5-0-pro": false,
"base-seedream-5-0-pro-layers": false,
"base-bailian-wan27-i2v-720p": false,
"base-bailian-wan27-i2v-1080p": false,
"base-seedance-2-0-480p": false,
"base-seedance-2-0-720p": false,
"base-seedance-2-0-1080p": false,
"base-seedance-2-0-4k": false,
"base-seedance-2-5-480p": false,
"base-seedance-2-5-720p": false,
"base-seedance-2-5-1080p": false,
}
for _, rule := range rules {
if _, ok := want[rule.ID]; !ok {
t.Fatalf("unexpected default rule %q", rule.ID)
}
want[rule.ID] = true
if !rule.Enabled || rule.MarkupMultiplier != 1.2 || rule.StandardUnitPriceFen <= 0 {
t.Fatalf("invalid default rule %#v", rule)
}
}
for id, found := range want {
if !found {
t.Errorf("missing default rule %q", id)
}
}
}