mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-03-19 13:47:43 +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>
43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
|
|
from playwright.sync_api import TimeoutError as PlaywrightTimeoutError
|
|
|
|
try:
|
|
from test.playwright.helpers._next_apps_helpers import (
|
|
RESULT_TIMEOUT_MS as DEFAULT_TIMEOUT_MS,
|
|
)
|
|
except Exception:
|
|
DEFAULT_TIMEOUT_MS = 15000
|
|
|
|
|
|
def wait_for_login_complete(page, timeout_ms: int | None = None) -> None:
|
|
if timeout_ms is None:
|
|
timeout_ms = DEFAULT_TIMEOUT_MS
|
|
wait_js = """
|
|
() => {
|
|
const path = window.location.pathname || '';
|
|
if (path.includes('/login')) return false;
|
|
const token = localStorage.getItem('Token');
|
|
const auth = localStorage.getItem('Authorization');
|
|
return Boolean((token && token.length) || (auth && auth.length));
|
|
}
|
|
"""
|
|
try:
|
|
page.wait_for_function(wait_js, timeout=timeout_ms)
|
|
except PlaywrightTimeoutError as exc:
|
|
url = page.url
|
|
testids = []
|
|
try:
|
|
testids = page.evaluate(
|
|
"""
|
|
() => Array.from(document.querySelectorAll('[data-testid]'))
|
|
.map((el) => el.getAttribute('data-testid'))
|
|
.filter((val) => val && /auth/i.test(val))
|
|
.slice(0, 30)
|
|
"""
|
|
)
|
|
except Exception:
|
|
testids = []
|
|
raise AssertionError(
|
|
f"Login did not complete within {timeout_ms}ms. url={url} auth_testids={testids}"
|
|
) from exc
|