mirror of
https://github.com/QuantumNous/new-api.git
synced 2026-09-01 19:41:57 +00:00
20 lines
908 B
Go
20 lines
908 B
Go
package billingexpr
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestTaskUsageExpressionUsesFactsAndTaskQuotaConversion(t *testing.T) {
|
|
expression := `tier("1080p", u("seconds") * (u("resolution") == "1080p" ? 0.4 : 0.2))`
|
|
cost, trace, err := RunExprWithRequest(expression, TokenParams{}, RequestInput{Usage: map[string]any{"seconds": 10.0, "resolution": "1080p"}})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 4.0, cost)
|
|
assert.Equal(t, "1080p", trace.MatchedTier)
|
|
result, err := ComputeTieredQuotaWithRequest(&BillingSnapshot{ExprString: expression, ExprHash: ExprHashString(expression), GroupRatio: 2, QuotaPerUnit: 500000, ExprVersion: 1, TaskUsageBilling: true}, TokenParams{}, RequestInput{Usage: map[string]any{"seconds": 10.0, "resolution": "1080p"}})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 4_000_000, result.ActualQuotaAfterGroup)
|
|
}
|