203 lines
7.2 KiB
Python
203 lines
7.2 KiB
Python
import pytest
|
|
from playwright.async_api import async_playwright
|
|
|
|
from common.config import Config
|
|
from common.login_helper import LoginHelper, get_test_password
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_login_page_loads() -> None:
|
|
"""测试登录页面是否能正常加载"""
|
|
async with async_playwright() as p:
|
|
browser = await p.chromium.launch(headless=True)
|
|
page = await browser.new_page()
|
|
|
|
# 访问登录页面
|
|
await page.goto(Config.frontend_url + '/login')
|
|
|
|
# 检查页面标题或特定元素是否存在
|
|
title = await page.title()
|
|
assert 'vfadmin' in title
|
|
|
|
# 检查登录表单元素是否存在
|
|
username_input = await page.query_selector('input[placeholder="账号"]')
|
|
password_input = await page.query_selector('input[placeholder="密码"]')
|
|
|
|
assert username_input is not None
|
|
assert password_input is not None
|
|
|
|
await browser.close()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_captcha_generation() -> None:
|
|
"""测试验证码是否正常生成（在测试环境中，验证码已禁用，但仍应能访问登录页面）"""
|
|
async with async_playwright() as p:
|
|
browser = await p.chromium.launch(headless=True)
|
|
page = await browser.new_page()
|
|
|
|
# 访问登录页面
|
|
await page.goto(Config.frontend_url + '/login')
|
|
|
|
# 在测试环境中验证码已禁用，但仍应能访问登录页面
|
|
title = await page.title()
|
|
assert 'vfadmin' in title
|
|
|
|
await browser.close()
|
|
|
|
|
|
def test_login_without_captcha() -> None:
|
|
"""测试在禁用验证码的情况下登录流程"""
|
|
helper = LoginHelper()
|
|
|
|
# 尝试登录（测试环境中验证码已禁用）
|
|
token = helper.login(username='admin')
|
|
|
|
# 验证登录结果
|
|
assert token is not None, '登录应该成功'
|
|
assert len(token) > 0, '应该返回有效的token'
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_login_flow_with_playwright() -> None:
|
|
"""使用Playwright测试完整的登录流程（测试环境中验证码已禁用）"""
|
|
async with async_playwright() as p:
|
|
browser = await p.chromium.launch(headless=True)
|
|
page = await browser.new_page()
|
|
|
|
# 访问登录页面
|
|
await page.goto(Config.frontend_url + '/login')
|
|
|
|
# 等待页面加载
|
|
await page.wait_for_selector('input[placeholder="账号"]')
|
|
|
|
# 输入用户名和密码
|
|
await page.fill('input[placeholder="账号"]', 'admin')
|
|
await page.fill('input[placeholder="密码"]', get_test_password('admin'))
|
|
|
|
# 在测试环境中验证码已禁用，所以不需要处理验证码
|
|
# 直接点击登录按钮
|
|
# 尝试多种可能的选择器
|
|
try:
|
|
await page.click('button:has-text("登录")')
|
|
except Exception:
|
|
try:
|
|
await page.click('button[type="submit"]')
|
|
except Exception:
|
|
await page.click('button.el-button')
|
|
|
|
# 等待页面跳转或检查登录结果
|
|
try:
|
|
# 等待跳转到主页或检查登录成功消息
|
|
await page.wait_for_url('**/index**', timeout=10000)
|
|
success = True
|
|
except Exception:
|
|
# 检查是否有错误消息
|
|
error_message = await page.query_selector('.el-message')
|
|
if error_message:
|
|
error_text = await error_message.text_content()
|
|
print(f'登录失败: {error_text}')
|
|
success = False
|
|
else:
|
|
success = False
|
|
|
|
assert success, '登录应该成功'
|
|
|
|
await browser.close()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_protected_routes_require_auth() -> None:
|
|
"""测试受保护的路由需要认证"""
|
|
async with async_playwright() as p:
|
|
browser = await p.chromium.launch(headless=True)
|
|
page = await browser.new_page()
|
|
|
|
# 直接访问受保护的页面（如用户管理）
|
|
await page.goto(Config.frontend_url + '/system/user')
|
|
|
|
# 检查是否被重定向到登录页面
|
|
current_url = page.url
|
|
|
|
# 检查页面是否包含登录相关的元素（如账号密码输入框）
|
|
try:
|
|
has_username_input = await page.is_visible('input[placeholder*="账号" i], input[placeholder*="username" i]')
|
|
except Exception:
|
|
has_username_input = False
|
|
|
|
try:
|
|
has_password_input = await page.is_visible('input[placeholder*="密码" i], input[placeholder*="password" i]')
|
|
except Exception:
|
|
has_password_input = False
|
|
|
|
try:
|
|
has_login_button = await page.is_visible('button:has-text("登录"), button:has-text("Login"), .login-btn')
|
|
except Exception:
|
|
has_login_button = False
|
|
|
|
# 检查URL是否包含login路径
|
|
is_redirected_to_login = '/login' in current_url.lower() or '/#/login' in current_url
|
|
|
|
# 断言：应该重定向到登录页面或页面包含登录相关元素
|
|
assert is_redirected_to_login or has_username_input or has_password_input or has_login_button, (
|
|
f'未登录用户访问受保护页面时，没有重定向到登录页。当前URL: {current_url}, 页面内容: {await page.content()}'
|
|
)
|
|
|
|
await browser.close()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_authenticated_access() -> None:
|
|
"""测试认证后的访问"""
|
|
# 首先通过API登录获取token
|
|
helper = LoginHelper()
|
|
token = helper.login(username='admin')
|
|
|
|
assert token is not None, '登录应该成功'
|
|
|
|
# 使用Playwright测试带认证的访问
|
|
async with async_playwright() as p:
|
|
browser = await p.chromium.launch(headless=True)
|
|
context = await browser.new_context(
|
|
storage_state={ # 使用storage state来保持登录状态
|
|
'cookies': [
|
|
{
|
|
'name': 'Admin-Token',
|
|
'value': token,
|
|
'domain': 'localhost',
|
|
'path': '/',
|
|
'httpOnly': False,
|
|
'secure': False,
|
|
'sameSite': 'Lax',
|
|
}
|
|
],
|
|
'origins': [],
|
|
}
|
|
)
|
|
page = await context.new_page()
|
|
|
|
# 访问受保护的页面
|
|
await page.goto(Config.frontend_url + '/system/user')
|
|
|
|
# 检查是否能够访问受保护的页面（而不是被重定向到登录页）
|
|
current_url = page.url
|
|
assert '/login' not in current_url, '已登录用户应该能够访问受保护页面'
|
|
|
|
# 检查页面是否包含用户管理相关的元素
|
|
try:
|
|
# 等待页面加载
|
|
await page.wait_for_selector('div:has-text("用户管理")', timeout=10000)
|
|
has_user_management = True
|
|
except Exception:
|
|
has_user_management = False
|
|
|
|
# 或者检查是否有表格或其他用户管理组件
|
|
table_element = await page.query_selector('el-table')
|
|
if not has_user_management and table_element:
|
|
has_user_management = True
|
|
|
|
assert has_user_management, '应该能够访问用户管理页面'
|
|
|
|
await context.close()
|
|
await browser.close()
|