wangxuming/NianAIGC
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

production runtime and mock removal

Browse Source
This commit is contained in:
inman committed 2026-08-17 23:44:14 +08:00
1 parent 2ef3da7af5
commit 6480e503eb
55 files changed
+581 -540
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

backend/internal/billing/catalog_test.go
+9 -10
View File
@@ -60,18 +60,17 @@ func TestCatalogRejectsAmbiguousWinners(t *testing.T) {
}
}
func TestCatalogExemptsDisabledMockAndSuperAdmin(t *testing.T) {
func TestCatalogExemptsDisabledBillingAndRejectsRemovedMockProvider(t *testing.T) {
rule := PriceRule{ID: "r", Provider: "bailian", Capability: "image.generate", Unit: UnitRequest, StandardUnitPriceFen: 10, MarkupMultiplier: 1, Enabled: true}
for _, input := range []QuoteInput{
{BillingDisabled: true, Provider: "bailian", Capability: "image.generate"},
{Provider: "mock", Capability: "image.generate"},
} {
quote, err := (Catalog{Rules: []PriceRule{rule}}).Quote(input)
if err != nil || quote != nil {
t.Fatalf("Quote(%#v) = %#v, %v", input, quote, err)
}
quote, err := (Catalog{Rules: []PriceRule{rule}}).Quote(QuoteInput{BillingDisabled: true, Provider: "bailian", Capability: "image.generate"})
if err != nil || quote != nil {
t.Fatalf("disabled Quote = %#v, %v", quote, err)
}
quote, err := (Catalog{Rules: []PriceRule{rule}}).Quote(QuoteInput{Provider: "bailian", Capability: "image.generate", Source: "platform", Role: "super_admin"})
quote, err = (Catalog{Rules: []PriceRule{rule}}).Quote(QuoteInput{Provider: "mock", Capability: "image.generate"})
if quote != nil || !errors.Is(err, ErrProviderUnavailable) {
t.Fatalf("removed provider Quote = %#v, %v", quote, err)
}
quote, err = (Catalog{Rules: []PriceRule{rule}}).Quote(QuoteInput{Provider: "bailian", Capability: "image.generate", Source: "platform", Role: "super_admin"})
if err != nil || quote == nil || !quote.QuotaExempt {
t.Fatalf("quote = %#v, %v", quote, err)
}