Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/migrations/channel_pricing_multipliers_migration_test.go

@@ -0,0 +1,43 @@
+package migrations
+
+import (
+	"strings"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestChannelPricingMultipliersMigration(t *testing.T) {
+	content, err := FS.ReadFile("228_channel_pricing_multipliers.sql")
+	require.NoError(t, err)
+
+	sql := strings.Join(strings.Fields(string(content)), " ")
+	for _, column := range []string{
+		"fast_multiplier NUMERIC(12,6)",
+		"flex_multiplier NUMERIC(12,6)",
+		"input_multiplier NUMERIC(12,6)",
+		"output_multiplier NUMERIC(12,6)",
+		"cache_write_multiplier NUMERIC(12,6)",
+		"cache_read_multiplier NUMERIC(12,6)",
+	} {
+		require.Contains(t, sql, "ADD COLUMN IF NOT EXISTS "+column)
+	}
+
+	constraints := []struct {
+		table  string
+		name   string
+		column string
+	}{
+		{"channel_model_pricing", "channel_model_pricing_fast_multiplier_positive", "fast_multiplier"},
+		{"channel_model_pricing", "channel_model_pricing_flex_multiplier_positive", "flex_multiplier"},
+		{"channel_pricing_intervals", "channel_pricing_intervals_input_multiplier_positive", "input_multiplier"},
+		{"channel_pricing_intervals", "channel_pricing_intervals_output_multiplier_positive", "output_multiplier"},
+		{"channel_pricing_intervals", "channel_pricing_intervals_cache_write_multiplier_positive", "cache_write_multiplier"},
+		{"channel_pricing_intervals", "channel_pricing_intervals_cache_read_multiplier_positive", "cache_read_multiplier"},
+	}
+	for _, constraint := range constraints {
+		require.Contains(t, sql, "conname = '"+constraint.name+"' AND conrelid = '"+constraint.table+"'::regclass")
+		require.Contains(t, sql, "ALTER TABLE "+constraint.table+" ADD CONSTRAINT "+constraint.name+
+			" CHECK ("+constraint.column+" IS NULL OR "+constraint.column+" > 0)")
+	}
+}