mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-06-01 05:17:51 +08:00
199fbceb72
### What problem does this PR solve? Refactor user REST API ### Type of change - [x] Refactoring
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".*/api/v1/auth/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)
|