mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-03-19 05:37:51 +08:00
860c4bd0bb
### What problem does this PR solve? This PR helps automate the testing of the ui interface using pytest Playwright ### Type of change - [x] New Feature (non-breaking change which adds functionality) - [x] Other (please describe): test automation infrastructure --------- Co-authored-by: Liu An <asiro@qq.com>
40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
|
|
try:
|
|
from test.playwright.helpers._auth_helpers import RESULT_TIMEOUT_MS as DEFAULT_TIMEOUT_MS
|
|
except Exception:
|
|
# Fallback for standalone usage when helper constants are unavailable.
|
|
DEFAULT_TIMEOUT_MS = 30_000
|
|
|
|
|
|
def capture_response(page, trigger, predicate, timeout_ms: int = DEFAULT_TIMEOUT_MS):
|
|
if hasattr(page, "expect_response"):
|
|
with page.expect_response(predicate, timeout=timeout_ms) as response_info:
|
|
trigger()
|
|
return response_info.value
|
|
if hasattr(page, "expect_event"):
|
|
with page.expect_event(
|
|
"response", predicate=predicate, timeout=timeout_ms
|
|
) as response_info:
|
|
trigger()
|
|
return response_info.value
|
|
if hasattr(page, "wait_for_event"):
|
|
trigger()
|
|
return page.wait_for_event("response", predicate=predicate, timeout=timeout_ms)
|
|
raise RuntimeError("Playwright Page lacks expect_response/expect_event/wait_for_event.")
|
|
|
|
|
|
def capture_response_json(
|
|
page, trigger, predicate, timeout_ms: int = DEFAULT_TIMEOUT_MS
|
|
) -> dict:
|
|
response = capture_response(page, trigger, predicate, timeout_ms)
|
|
info: dict = {"__url__": response.url, "__status__": response.status}
|
|
try:
|
|
data = response.json()
|
|
if isinstance(data, dict):
|
|
info.update(data)
|
|
else:
|
|
info["__parse_error__"] = "non-dict response body"
|
|
except Exception as exc:
|
|
info["__parse_error__"] = str(exc)
|
|
return info
|