mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-03-26 00:47:44 +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>
87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
import os
|
|
|
|
import pytest
|
|
from playwright.sync_api import expect
|
|
|
|
RESULT_TIMEOUT_MS = 15000
|
|
|
|
|
|
def _wait_for_login_complete(page, timeout_ms: int = RESULT_TIMEOUT_MS) -> None:
|
|
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));
|
|
}
|
|
"""
|
|
page.wait_for_function(wait_js, timeout=timeout_ms)
|
|
|
|
|
|
def ensure_authed(
|
|
page,
|
|
login_url: str,
|
|
active_auth_context,
|
|
auth_click,
|
|
seeded_user_credentials=None,
|
|
timeout_ms: int = RESULT_TIMEOUT_MS,
|
|
) -> None:
|
|
if seeded_user_credentials:
|
|
email, password = seeded_user_credentials
|
|
else:
|
|
email = os.getenv("SEEDED_USER_EMAIL")
|
|
password = os.getenv("SEEDED_USER_PASSWORD")
|
|
if not email or not password:
|
|
pytest.skip("SEEDED_USER_EMAIL/SEEDED_USER_PASSWORD not set.")
|
|
|
|
token_wait_js = """
|
|
() => {
|
|
const token = localStorage.getItem('Token');
|
|
const auth = localStorage.getItem('Authorization');
|
|
return Boolean((token && token.length) || (auth && auth.length));
|
|
}
|
|
"""
|
|
|
|
try:
|
|
if "/login" not in page.url:
|
|
if (
|
|
page.locator(
|
|
"input[data-testid='auth-email'], [data-testid='auth-email'] input"
|
|
).count()
|
|
== 0
|
|
):
|
|
try:
|
|
page.wait_for_function(token_wait_js, timeout=2000)
|
|
return
|
|
except Exception:
|
|
pass
|
|
except Exception:
|
|
pass
|
|
|
|
page.goto(login_url, wait_until="domcontentloaded")
|
|
|
|
form, _ = active_auth_context()
|
|
email_input = form.locator(
|
|
"input[data-testid='auth-email'], [data-testid='auth-email'] input"
|
|
)
|
|
password_input = form.locator(
|
|
"input[data-testid='auth-password'], [data-testid='auth-password'] input"
|
|
)
|
|
expect(email_input).to_have_count(1)
|
|
expect(password_input).to_have_count(1)
|
|
email_input.fill(email)
|
|
password_input.fill(password)
|
|
password_input.blur()
|
|
|
|
submit_button = form.locator(
|
|
"button[data-testid='auth-submit'], [data-testid='auth-submit'] button, [data-testid='auth-submit']"
|
|
)
|
|
expect(submit_button).to_have_count(1)
|
|
auth_click(submit_button, "submit_login")
|
|
|
|
_wait_for_login_complete(page, timeout_ms=timeout_ms)
|
|
expect(page.locator("form[data-testid='auth-form'][data-active='true']")).to_have_count(
|
|
0, timeout=timeout_ms
|
|
)
|