mirror of
https://github.com/infiniflow/ragflow.git
synced 2026-07-05 19:08:38 +08:00
73 lines
2.5 KiB
Python
73 lines
2.5 KiB
Python
import pytest
|
|
|
|
from test.playwright.helpers.flow_context import FlowContext
|
|
from test.playwright.helpers.flow_steps import flow_params, require
|
|
|
|
|
|
def step_01_open_login(ctx: FlowContext, step, snap):
|
|
page = ctx.page
|
|
with step("navigate to login page"):
|
|
response = page.goto(ctx.smoke_login_url, wait_until="domcontentloaded")
|
|
ctx.state["smoke_opened"] = True
|
|
ctx.state["smoke_response"] = response
|
|
|
|
|
|
def step_02_validate_page(ctx: FlowContext, step, snap):
|
|
require(ctx.state, "smoke_opened")
|
|
page = ctx.page
|
|
response = ctx.state.get("smoke_response")
|
|
content = page.content()
|
|
content_type = ""
|
|
status = None
|
|
if response is not None:
|
|
status = response.status
|
|
content_type = response.headers.get("content-type", "")
|
|
|
|
content_head = content.lstrip()[:200]
|
|
looks_json = content_head.startswith("{") or content_head.startswith("[")
|
|
is_html = "text/html" in content_type.lower() or "<html" in content.lower()
|
|
|
|
if response is not None and status is not None and status >= 400:
|
|
raise AssertionError(_format_diag(page, response, "HTTP error status"))
|
|
|
|
if looks_json or not is_html:
|
|
raise AssertionError(_format_diag(page, response, "Non-HTML response"))
|
|
|
|
root_count = page.locator("#root").count()
|
|
input_count = page.locator("input").count()
|
|
logo_count = page.locator("img[alt='logo']").count()
|
|
if root_count + input_count + logo_count == 0:
|
|
raise AssertionError(_format_diag(page, response, "No SPA root, inputs, or logo found"))
|
|
|
|
|
|
STEPS = [
|
|
("01_open_login", step_01_open_login),
|
|
("02_validate_page", step_02_validate_page),
|
|
]
|
|
|
|
|
|
@pytest.mark.smoke
|
|
@pytest.mark.p0
|
|
@pytest.mark.auth
|
|
@pytest.mark.parametrize("step_fn", flow_params(STEPS))
|
|
def test_auth_page_smoke_flow(step_fn, flow_page, flow_state, base_url, smoke_login_url, step, snap):
|
|
ctx = FlowContext(
|
|
page=flow_page,
|
|
state=flow_state,
|
|
base_url=base_url,
|
|
login_url=smoke_login_url,
|
|
smoke_login_url=smoke_login_url,
|
|
)
|
|
step_fn(ctx, step, snap)
|
|
|
|
|
|
def _format_diag(page, response, reason: str) -> str:
|
|
status = response.status if response is not None else "<no response>"
|
|
content_type = ""
|
|
if response is not None:
|
|
content_type = response.headers.get("content-type", "")
|
|
url = page.url
|
|
title = page.title()
|
|
snippet = page.content().strip().replace("\n", " ")[:500]
|
|
return f"{reason}. url={url} title={title} status={status} content_type={content_type} snippet={snippet}"
|