87 lines
3.2 KiB
Go
87 lines
3.2 KiB
Go
package app
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"infinite-canvas/backend/internal/model"
|
|
)
|
|
|
|
func TestAllVideoProtocolsQuoteReserveAndSettleWithoutProviderUsage(t *testing.T) {
|
|
for _, protocol := range []model.ChannelInterfaceType{"newapi-channel-2", "newapi-video", "custom-video"} {
|
|
t.Run(string(protocol), func(t *testing.T) {
|
|
svc, db, channel, channelModel, _ := newVideoTokenQuoteFixture(t)
|
|
if err := db.AutoMigrate(&model.ApiCallLog{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := db.Model(&model.ChannelModel{}).Where("id = ?", channelModel.ID).Update("protocol", protocol).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
intent := ModelRequestIntent{Capability: "video", Options: map[string]any{"vquality": "720p", "size": "16:9", "videoSeconds": 5}}
|
|
quote, err := svc.QuoteChannelModel(ChannelModelQuoteRequest{ChannelID: channel.ID, ModelKey: channelModel.ModelKey, Intent: intent})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assertVideoTokenQuote(t, quote, 1_900_800)
|
|
input := quoteInput(intent, channelModel.ModelKey)
|
|
input["config"].(map[string]any)["channelId"] = channel.ID
|
|
input, err = svc.resolveSystemChannelModelSelection(input, "canvas_video", "")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
order, err := svc.taskBillingOrder("quote-user", &model.Task{ID: "generic-video", Type: "canvas_video"}, input)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if order.VideoFormulaTokens != 108_000 || order.AmountMicrocredits != quote.AmountMicrocredits {
|
|
t.Fatalf("quote/order disagree: %#v / %#v", quote, order)
|
|
}
|
|
if err := svc.repo.ReserveBillingOrder(order); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for i := 0; i < 2; i++ {
|
|
if err := svc.SettleBilling(order.ID, "completed-video"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
settled, err := svc.repo.BillingOrder(order.ID)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if settled.Status != model.BillingStatusSettled || settled.UsageSource != "video_formula" || settled.UsageAvailable || settled.OutputTokens != 108_000 || settled.ActualAmountMicrocredits != 1_728_000 || settled.RefundedAmountMicrocredits != 172_800 {
|
|
t.Fatalf("wrong formula settlement: %#v", settled)
|
|
}
|
|
var account model.CreditAccount
|
|
if err := db.First(&account, "user_id = ?", "quote-user").Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if account.AvailableMicrocredits != 10_000_000-1_728_000 || account.ReservedMicrocredits != 250_000 {
|
|
t.Fatalf("incorrect balance after idempotent settlement: %#v", account)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestGenericVideoTokenDimensions(t *testing.T) {
|
|
for _, test := range []struct {
|
|
resolution, size string
|
|
width, height int64
|
|
}{
|
|
{"1440p", "16:9", 2560, 1440},
|
|
{"2K", "9:16", 1440, 2560},
|
|
{"960p", "4:3", 1280, 960},
|
|
{"768p竖", "9:16", 768, 1366},
|
|
{"720p", "2:1", 1440, 720},
|
|
{"", "1280x720", 1280, 720},
|
|
{"", "", 1280, 720},
|
|
} {
|
|
t.Run(test.resolution+"/"+test.size, func(t *testing.T) {
|
|
got := estimateArkVideoTokens(map[string]any{"config": map[string]any{
|
|
"model": "generic-video", "vquality": test.resolution, "size": test.size, "videoSeconds": 5,
|
|
}})
|
|
if got.Err != nil || got.Video.OutputWidth != test.width || got.Video.OutputHeight != test.height {
|
|
t.Fatalf("dimensions=%#v error=%v", got.Video, got.Err)
|
|
}
|
|
})
|
|
}
|
|
}
|