feat: publish v1.0.0-rc.36-custom.1

pkg/billingexpr/settle_clamp_test.go

@@ -0,0 +1,52 @@
+package billingexpr_test
+
+import (
+	"testing"
+
+	"github.com/QuantumNous/new-api/common"
+	"github.com/QuantumNous/new-api/pkg/billingexpr"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+// TestComputeTieredQuota_ClampOnOverflow guards the billing-safety invariant
+// that an oversized tiered settlement clamps to the single-request max instead of
+// wrapping into a credit, and that the saturation event is surfaced on the
+// result so callers can record it for admin auditing.
+func TestComputeTieredQuota_ClampOnOverflow(t *testing.T) {
+	// exprOutput = p * 1e12 = 1e21; quotaBeforeGroup = 1e21 / 1e6 * 5e5 = 5e20,
+	// which far exceeds the supported single-request range and must saturate.
+	exprStr := `tier("base", p * 1000000000000)`
+	snap := &billingexpr.BillingSnapshot{
+		BillingMode:  "tiered_expr",
+		ExprString:   exprStr,
+		ExprHash:     billingexpr.ExprHashString(exprStr),
+		GroupRatio:   1.0,
+		QuotaPerUnit: 500_000,
+	}
+
+	result, err := billingexpr.ComputeTieredQuota(snap, billingexpr.TokenParams{P: 1_000_000_000})
+	require.NoError(t, err)
+
+	assert.Equal(t, common.MaxQuota, result.ActualQuotaAfterGroup, "oversized quota must clamp, never wrap negative")
+	require.NotNil(t, result.Clamp, "clamp event must be surfaced so it can be audited")
+	assert.Equal(t, common.QuotaClampOverflow, result.Clamp.Kind)
+	assert.Equal(t, common.MaxQuota, result.Clamp.Clamped)
+}
+
+// TestComputeTieredQuota_NoClampInRange confirms an in-range settlement leaves
+// Clamp nil, so the audit path is a no-op in the common case.
+func TestComputeTieredQuota_NoClampInRange(t *testing.T) {
+	exprStr := `tier("base", p * 2 + c * 10)`
+	snap := &billingexpr.BillingSnapshot{
+		BillingMode:  "tiered_expr",
+		ExprString:   exprStr,
+		ExprHash:     billingexpr.ExprHashString(exprStr),
+		GroupRatio:   1.0,
+		QuotaPerUnit: 500_000,
+	}
+
+	result, err := billingexpr.ComputeTieredQuota(snap, billingexpr.TokenParams{P: 1000, C: 500})
+	require.NoError(t, err)
+	assert.Nil(t, result.Clamp, "in-range settlement must not report a clamp")
+}