mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-03-16 12:27:49 +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>
51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
import re
|
|
|
|
import pytest
|
|
|
|
from test.playwright.helpers.flow_steps import flow_params, require
|
|
|
|
|
|
def step_01_open_login(flow_page, flow_state, login_url, active_auth_context, step, snap):
|
|
with step("open login page"):
|
|
flow_page.goto(login_url, wait_until="domcontentloaded")
|
|
flow_state["login_opened"] = True
|
|
snap("open")
|
|
|
|
|
|
def step_02_initiate_sso(flow_page, flow_state, login_url, active_auth_context, step, snap):
|
|
require(flow_state, "login_opened")
|
|
page = flow_page
|
|
form, _ = active_auth_context()
|
|
sso_buttons = form.locator("button:has-text('Sign in with')")
|
|
if sso_buttons.count() == 0:
|
|
pytest.skip("No SSO providers rendered on the login page")
|
|
|
|
with step("initiate SSO navigation"):
|
|
clicked = False
|
|
for handle in sso_buttons.element_handles():
|
|
if handle.is_visible() and handle.is_enabled():
|
|
handle.click()
|
|
clicked = True
|
|
break
|
|
if not clicked:
|
|
pytest.skip("SSO buttons were present but not interactable")
|
|
|
|
page.wait_for_url(re.compile(r".*/v1/user/login/"), timeout=5000)
|
|
flow_state["sso_clicked"] = True
|
|
snap("sso_clicked")
|
|
|
|
|
|
STEPS = [
|
|
("01_open_login", step_01_open_login),
|
|
("02_initiate_sso", step_02_initiate_sso),
|
|
]
|
|
|
|
|
|
@pytest.mark.p1
|
|
@pytest.mark.auth
|
|
@pytest.mark.parametrize("step_fn", flow_params(STEPS))
|
|
def test_sso_optional_flow(
|
|
step_fn, flow_page, flow_state, login_url, active_auth_context, step, snap
|
|
):
|
|
step_fn(flow_page, flow_state, login_url, active_auth_context, step, snap)
|