fix(tencent): wait for real-name verification

2026-08-10 02:09:51 +08:00
parent 8ee7bcdb8a
commit ad2419ef16
2 changed files with 71 additions and 0 deletions
@@ -0,0 +1,43 @@
+import asyncio
+from pathlib import Path
+
+from patchright.async_api import async_playwright
+
+from uploader.tencent_uploader.main import TencentBaseUploader
+
+
+def test_wait_for_realtime_verification_saves_qr_and_resumes(tmp_path: Path):
+    async def scenario():
+        async with async_playwright() as playwright:
+            browser = await playwright.chromium.launch(headless=True, channel="chrome")
+            page = await browser.new_page()
+            await page.set_content(
+                """
+                <div class="weui-desktop-dialog__wrp">
+                  <div>为保障账号安全，需用管理员微信扫码完成实名验证后才能继续发表</div>
+                  <div class="qr">QR</div>
+                </div>
+                """
+            )
+            uploader = TencentBaseUploader(publish_date=0, account_file="unused")
+            qr_path = tmp_path / "verification.png"
+
+            async def dismiss_dialog():
+                await page.wait_for_timeout(150)
+                await page.locator("div.weui-desktop-dialog__wrp").evaluate("el => el.remove()")
+
+            dismiss_task = asyncio.create_task(dismiss_dialog())
+            result = await uploader.wait_for_realtime_verification(
+                page,
+                qr_path=qr_path,
+                timeout_seconds=2,
+                poll_interval_seconds=0.05,
+            )
+            await dismiss_task
+            await browser.close()
+
+            assert result == qr_path
+            assert qr_path.is_file()
+            assert qr_path.stat().st_size > 0
+
+    asyncio.run(scenario())