fix: repair generated focused tests

This commit is contained in:
Aaron Liang
2026-07-15 00:48:45 +08:00
parent 7be06330f8
commit 3dd96b31ef
+46
View File
@@ -0,0 +1,46 @@
#!/usr/bin/env python3
from pathlib import Path
script = Path(__file__).resolve().with_name("apply_final_flow_fixes.py")
text = script.read_text(encoding="utf-8")
old_import = '''remote_tests = r\'''import unittest
from unittest.mock import patch
import grok_register_ttk as app
'''
new_import = '''remote_tests = r\'''import sys
import types
import unittest
from unittest.mock import patch
# Keep this unit test independent from optional browser/network dependencies.
drission = types.ModuleType("DrissionPage")
drission.Chromium = type("Chromium", (), {})
drission.ChromiumOptions = type("ChromiumOptions", (), {})
drission_errors = types.ModuleType("DrissionPage.errors")
drission_errors.PageDisconnectedError = type("PageDisconnectedError", (Exception,), {})
curl_cffi = types.ModuleType("curl_cffi")
curl_cffi.requests = types.SimpleNamespace()
sys.modules.setdefault("DrissionPage", drission)
sys.modules.setdefault("DrissionPage.errors", drission_errors)
sys.modules.setdefault("curl_cffi", curl_cffi)
import grok_register_ttk as app
'''
if old_import not in text:
if new_import not in text:
raise RuntimeError("remote test import block not found")
else:
text = text.replace(old_import, new_import, 1)
old_cleanup = ' interval_cleanups = [event for event in fake.events if isinstance(event, tuple) and "已成功" in event[1]]\n'
new_cleanup = ' interval_cleanups = [\n event for event in fake.events\n if isinstance(event, tuple)\n and len(event) > 1\n and isinstance(event[1], str)\n and "已成功" in event[1]\n ]\n'
if old_cleanup not in text:
if new_cleanup not in text:
raise RuntimeError("cleanup test expression not found")
else:
text = text.replace(old_cleanup, new_cleanup, 1)
script.write_text(text, encoding="utf-8")
print("final flow test generation repaired")