44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
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())
|