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

backend/migrations/group_usage_rollup_migration_test.go

@@ -0,0 +1,52 @@
+//go:build unit
+
+package migrations
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestMigration222CreatesGroupUsageRollups(t *testing.T) {
+	content, err := FS.ReadFile("222_group_usage_daily_rollups.sql")
+	require.NoError(t, err)
+
+	sql := string(content)
+	require.Contains(t, sql, "CREATE TABLE IF NOT EXISTS usage_group_daily_rollups")
+	require.Contains(t, sql, "actual_cost DECIMAL(20, 10)")
+	require.Contains(t, sql, "PRIMARY KEY (bucket_date, group_id)")
+	require.Contains(t, sql, "CREATE TABLE IF NOT EXISTS usage_group_rollup_state")
+	require.Contains(t, sql, "CHECK (id = 1)")
+	require.Contains(t, sql, "TIMESTAMPTZ '1970-01-01 00:00:00+00'")
+	require.Contains(t, sql, "ON CONFLICT (id) DO NOTHING")
+}
+
+func TestMigration222InvalidatesClosedBucketsWhenUsageLogsChange(t *testing.T) {
+	content, err := FS.ReadFile("222_group_usage_daily_rollups.sql")
+	require.NoError(t, err)
+
+	sql := string(content)
+	require.Contains(t, sql, "CREATE OR REPLACE FUNCTION invalidate_group_usage_rollup_state")
+	require.Contains(t, sql, "SELECT closed_before")
+	require.Contains(t, sql, "FOR UPDATE")
+	require.Contains(t, sql, "FOR KEY SHARE")
+	require.Contains(t, sql, "REFERENCING NEW TABLE AS inserted_usage_logs")
+	require.Contains(t, sql, "closed_before = LEAST(closed_before, affected_date)")
+	require.Contains(t, sql, "CREATE TRIGGER usage_logs_group_rollup_invalidate_insert")
+	require.Contains(t, sql, "CREATE TRIGGER usage_logs_group_rollup_invalidate_delete")
+	require.Contains(t, sql, "CREATE TRIGGER usage_logs_group_rollup_invalidate_update")
+	require.Contains(t, sql, "AFTER UPDATE OF created_at, group_id, actual_cost")
+}
+
+func TestMigration223TracksConfiguredTimezone(t *testing.T) {
+	content, err := FS.ReadFile("223_group_usage_rollup_timezone.sql")
+	require.NoError(t, err)
+
+	sql := string(content)
+	require.Contains(t, sql, "ADD COLUMN IF NOT EXISTS timezone_name TEXT")
+	require.Contains(t, sql, "DEFAULT 'Asia/Shanghai'")
+	require.Contains(t, sql, "current_setting('TimeZone')")
+	require.Contains(t, sql, "CREATE OR REPLACE FUNCTION invalidate_group_usage_rollup_state")
+	require.Contains(t, sql, "CREATE OR REPLACE FUNCTION invalidate_group_usage_rollup_state_after_insert")
+}