From dca7773519ab0f946293a371ee331ffd009ba8fa Mon Sep 17 00:00:00 2001
From: linkeyu <798648091@qq.com>
Date: Sat, 1 Aug 2026 16:59:11 +0800
Subject: [PATCH] test(marquee): isolate test account filter data

---
 tests/test_ops_marquee.py | 97 ++++++++++++++++++++-------------------
 1 file changed, 50 insertions(+), 47 deletions(-)

diff --git a/tests/test_ops_marquee.py b/tests/test_ops_marquee.py
index c529b6b..637506b 100644
--- a/tests/test_ops_marquee.py
+++ b/tests/test_ops_marquee.py
@@ -19,54 +19,57 @@ def test_real_marquee_records_exclude_configured_test_account(monkeypatch) -> No
     monkeypatch.setattr(settings, "TEST_ACCOUNT_PHONES", listed_test_phone)
     _clear_real_rows_cache()
 
-    with SessionLocal() as db:
-        legacy_test_user = User(
-            phone=legacy_test_phone,
-            username="29900009991",
-            register_channel="sms",
-        )
-        listed_test_user = User(
-            phone=listed_test_phone,
-            username="29900009992",
-            register_channel="sms",
-        )
-        real_user = User(
-            phone=real_phone,
-            username="29900009993",
-            register_channel="sms",
-        )
-        db.add_all([legacy_test_user, listed_test_user, real_user])
-        db.flush()
-        db.add_all([
-            ComparisonRecord(
-                user_id=legacy_test_user.id,
-                trace_id="marquee-legacy-test-account-trace",
-                status="success",
-                saved_amount_cents=29_991,
-                items=[],
-                comparison_results=[],
-            ),
-            ComparisonRecord(
-                user_id=listed_test_user.id,
-                trace_id="marquee-listed-test-account-trace",
-                status="success",
-                saved_amount_cents=29_992,
-                items=[],
-                comparison_results=[],
-            ),
-            ComparisonRecord(
-                user_id=real_user.id,
-                trace_id="marquee-real-account-trace",
-                status="success",
-                saved_amount_cents=29_993,
-                items=[],
-                comparison_results=[],
-            ),
-        ])
-        db.commit()
+    try:
+        with SessionLocal() as db:
+            legacy_test_user = User(
+                phone=legacy_test_phone,
+                username="29900009991",
+                register_channel="sms",
+            )
+            listed_test_user = User(
+                phone=listed_test_phone,
+                username="29900009992",
+                register_channel="sms",
+            )
+            real_user = User(
+                phone=real_phone,
+                username="29900009993",
+                register_channel="sms",
+            )
+            db.add_all([legacy_test_user, listed_test_user, real_user])
+            db.flush()
+            db.add_all([
+                ComparisonRecord(
+                    user_id=legacy_test_user.id,
+                    trace_id="marquee-legacy-test-account-trace",
+                    status="success",
+                    saved_amount_cents=29_991,
+                    items=[],
+                    comparison_results=[],
+                ),
+                ComparisonRecord(
+                    user_id=listed_test_user.id,
+                    trace_id="marquee-listed-test-account-trace",
+                    status="success",
+                    saved_amount_cents=29_992,
+                    items=[],
+                    comparison_results=[],
+                ),
+                ComparisonRecord(
+                    user_id=real_user.id,
+                    trace_id="marquee-real-account-trace",
+                    status="success",
+                    saved_amount_cents=29_993,
+                    items=[],
+                    comparison_results=[],
+                ),
+            ])
+            db.flush()
 
-        app_rows = ops_marquee._recent_real_rows(db)
-        browse_rows, _ = ops_marquee.list_real_records(db, mode="real", limit=1_000)
+            app_rows = ops_marquee._recent_real_rows(db)
+            browse_rows, _ = ops_marquee.list_real_records(db, mode="real", limit=1_000)
+    finally:
+        _clear_real_rows_cache()
 
     assert all(saved not in {29_991, 29_992} for _uid, saved, _nickname in app_rows)
     assert any(saved == 29_993 for _uid, saved, _nickname in app_rows)