75 lines
3.2 KiB
Python
75 lines
3.2 KiB
Python
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import functools
|
|
import threading
|
|
import unittest
|
|
from contextlib import contextmanager
|
|
from http.server import SimpleHTTPRequestHandler, ThreadingHTTPServer
|
|
from pathlib import Path
|
|
|
|
from playwright.async_api import async_playwright
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
@contextmanager
|
|
def local_server():
|
|
handler = functools.partial(SimpleHTTPRequestHandler, directory=str(ROOT))
|
|
server = ThreadingHTTPServer(("127.0.0.1", 0), handler)
|
|
thread = threading.Thread(target=server.serve_forever, daemon=True)
|
|
thread.start()
|
|
try:
|
|
yield f"http://127.0.0.1:{server.server_port}"
|
|
finally:
|
|
server.shutdown()
|
|
server.server_close()
|
|
thread.join(timeout=2)
|
|
|
|
|
|
class GuanghanTopologyE2ETests(unittest.TestCase):
|
|
def test_rendered_topology_and_modal_interactions(self) -> None:
|
|
async def scenario(base_url: str) -> None:
|
|
async with async_playwright() as pw:
|
|
browser = await pw.chromium.launch(headless=True)
|
|
page = await browser.new_page(viewport={"width": 1440, "height": 1100})
|
|
errors: list[str] = []
|
|
page.on("console", lambda msg: errors.append(msg.text) if msg.type == "error" else None)
|
|
await page.goto(
|
|
f"{base_url}/data/wiki/guanghan_topology_demo.html",
|
|
wait_until="networkidle",
|
|
)
|
|
|
|
self.assertEqual(await page.locator("[data-module]").count(), 14)
|
|
self.assertEqual(await page.locator('[data-connection="pressurized"]').count(), 2)
|
|
strokes = await page.locator('[data-connection="pressurized"]').evaluate_all(
|
|
"(els) => els.map((el) => getComputedStyle(el).stroke)"
|
|
)
|
|
self.assertEqual(strokes, ["rgb(81, 232, 156)", "rgb(81, 232, 156)"])
|
|
|
|
for forbidden in ("未来北部十字结构", "未来东部十字结构", "未来西部十字结构"):
|
|
self.assertEqual(await page.get_by_text(forbidden, exact=False).count(), 0)
|
|
|
|
await page.locator('[data-module="core"]').click()
|
|
self.assertEqual(await page.locator("#detail-modal").count(), 1)
|
|
self.assertEqual(await page.locator("#detail-modal").get_attribute("aria-hidden"), "false")
|
|
self.assertEqual(await page.locator("#detail-title").inner_text(), "广寒核心舱")
|
|
await page.keyboard.press("Escape")
|
|
self.assertEqual(await page.locator("#detail-modal").get_attribute("aria-hidden"), "true")
|
|
|
|
await page.locator('[data-module="habitat-1"]').click()
|
|
self.assertIn("运输：高2.7 m", await page.locator("#detail-dimensions").inner_text())
|
|
await page.locator("#detail-modal").click(position={"x": 5, "y": 5})
|
|
self.assertEqual(await page.locator("#detail-modal").get_attribute("aria-hidden"), "true")
|
|
|
|
self.assertEqual(errors, [])
|
|
await browser.close()
|
|
|
|
with local_server() as base_url:
|
|
asyncio.run(scenario(base_url))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|