主要改进: - 新增统一测试器 (universal_tester.py) 支持多种测试模式 - 优化测试报告生成器，支持汇总报告和操作截图 - 增强探索器 DFS 算法和状态指纹识别 - 新增智能测试配置 (smart_test.yaml) - 改进 AI 模型集成 (GLM/Gemini 支持) - 添加开发调试工具和文档
This commit is contained in:
empty
205
tests/smart_test.py
Normal file
205
tests/smart_test.py
Normal file
@@ -0,0 +1,205 @@
|
||||
#!/usr/bin/env python3
|
||||
"""
|
||||
增强的零配置测试 - 支持无AI模式
|
||||
"""
|
||||
|
||||
import sys
|
||||
sys.path.insert(0, ".")
|
||||
|
||||
from src import WebTester
|
||||
import time
|
||||
|
||||
def smart_test(url: str, model: str = "glm", headless: bool = False, auto_login: bool = True):
|
||||
"""
|
||||
智能测试，即使AI失败也能继续
|
||||
"""
|
||||
|
||||
print("=" * 60)
|
||||
print("🤖 增强智能测试")
|
||||
print("=" * 60)
|
||||
print(f"🌐 目标URL: {url}")
|
||||
print(f"🤖 AI模型: {model}")
|
||||
print(f"🖥️ 无头模式: {'是' if headless else '否'}")
|
||||
print(f"🔐 自动登录: {'是' if auto_login else '否'}")
|
||||
print("-" * 60)
|
||||
|
||||
tester = WebTester(model=model, headless=headless)
|
||||
|
||||
try:
|
||||
# 启动并导航
|
||||
tester.start()
|
||||
tester.goto(url)
|
||||
time.sleep(2) # 等待页面加载
|
||||
|
||||
# 步骤1: 处理登录
|
||||
current_url = tester.browser.page.url
|
||||
print(f"\n📍 当前URL: {current_url}")
|
||||
|
||||
# 检查是否需要登录
|
||||
need_login = ("login" in current_url.lower() or
|
||||
tester.browser.page.locator("input[type='password']").count() > 0)
|
||||
|
||||
if need_login and auto_login:
|
||||
print("\n🔐 检测到登录页面，尝试登录...")
|
||||
|
||||
# 方法1: 尝试AI登录（如果API可用）
|
||||
login_success = False
|
||||
try:
|
||||
result = tester.test("输入用户名admin和密码password，点击登录按钮")
|
||||
time.sleep(3)
|
||||
new_url = tester.browser.page.url
|
||||
if new_url != current_url and "login" not in new_url.lower():
|
||||
print(" ✅ AI登录成功")
|
||||
login_success = True
|
||||
except Exception as e:
|
||||
print(f" ⚠️ AI登录失败: {str(e)[:50]}...")
|
||||
|
||||
# 方法2: 如果AI失败，使用DOM直接登录
|
||||
if not login_success:
|
||||
print(" 🔄 尝试直接登录...")
|
||||
try:
|
||||
# 查找用户名和密码输入框
|
||||
page = tester.browser.page
|
||||
|
||||
# 尝试多种可能的用户名输入框
|
||||
username_selectors = [
|
||||
"input[placeholder*='用户名']",
|
||||
"input[placeholder*='账号']",
|
||||
"input[name='username']",
|
||||
"input[name='account']",
|
||||
"input[type='text']"
|
||||
]
|
||||
|
||||
username_input = None
|
||||
for selector in username_selectors:
|
||||
if page.locator(selector).count() > 0:
|
||||
username_input = page.locator(selector).first
|
||||
break
|
||||
|
||||
# 尝试多种可能的密码输入框
|
||||
password_selectors = [
|
||||
"input[type='password']",
|
||||
"input[placeholder*='密码']",
|
||||
"input[name='password']"
|
||||
]
|
||||
|
||||
password_input = None
|
||||
for selector in password_selectors:
|
||||
if page.locator(selector).count() > 0:
|
||||
password_input = page.locator(selector).first
|
||||
break
|
||||
|
||||
# 填写并提交
|
||||
if username_input and password_input:
|
||||
username_input.fill("admin")
|
||||
password_input.fill("password")
|
||||
|
||||
# 查找登录按钮
|
||||
login_selectors = [
|
||||
"button:has-text('登录')",
|
||||
"button:has-text('登陆')",
|
||||
"button:has-text('确定')",
|
||||
"input[type='submit']",
|
||||
".login-btn"
|
||||
]
|
||||
|
||||
for selector in login_selectors:
|
||||
if page.locator(selector).count() > 0:
|
||||
page.locator(selector).first.click()
|
||||
break
|
||||
|
||||
time.sleep(3)
|
||||
new_url = tester.browser.page.url
|
||||
if new_url != current_url:
|
||||
print(" ✅ 直接登录成功")
|
||||
login_success = True
|
||||
except Exception as e:
|
||||
print(f" ❌ 直接登录失败: {e}")
|
||||
|
||||
if not login_success:
|
||||
print(" ⚠️ 无法自动登录，将测试登录页面功能")
|
||||
|
||||
# 步骤2: 开始探索
|
||||
print("\n🔍 开始智能探索...")
|
||||
|
||||
# 根据是否登录成功调整探索策略
|
||||
if login_success:
|
||||
explore_config = {
|
||||
"max_depth": 5,
|
||||
"max_clicks": 100,
|
||||
"focus_patterns": ["管理", "查询", "新增", "详情"],
|
||||
"dangerous_patterns": ["删除", "退出", "注销"]
|
||||
}
|
||||
else:
|
||||
explore_config = {
|
||||
"max_depth": 2,
|
||||
"max_clicks": 20,
|
||||
"focus_patterns": ["登录", "用户", "密码"],
|
||||
"dangerous_patterns": []
|
||||
}
|
||||
|
||||
# 尝试AI探索，如果失败则使用DOM探索
|
||||
try:
|
||||
explore_result = tester.explore(explore_config)
|
||||
print(" ✅ AI探索完成")
|
||||
except Exception as e:
|
||||
print(f" ⚠️ AI探索失败: {str(e)[:50]}...")
|
||||
print(" 🔄 使用基础探索...")
|
||||
|
||||
# 基础探索：点击所有可见的按钮和链接
|
||||
page = tester.browser.page
|
||||
clickable_elements = page.locator("button, a, [role='button'], input[type='button']")
|
||||
|
||||
clicked_count = 0
|
||||
for i in range(min(clickable_elements.count(), 10)):
|
||||
try:
|
||||
element = clickable_elements.nth(i)
|
||||
if element.is_visible():
|
||||
text = element.inner_text()[:20]
|
||||
print(f" 点击: {text or '无文本'}")
|
||||
element.click()
|
||||
time.sleep(1)
|
||||
page.go_back()
|
||||
time.sleep(1)
|
||||
clicked_count += 1
|
||||
except:
|
||||
continue
|
||||
|
||||
print(f" 📊 基础探索完成，点击了 {clicked_count} 个元素")
|
||||
|
||||
# 步骤3: 生成测试报告
|
||||
print("\n📊 测试总结:")
|
||||
print(f" ✅ 测试完成")
|
||||
print(f" 📄 已访问页面: {tester.browser.page.url}")
|
||||
|
||||
return True
|
||||
|
||||
except Exception as e:
|
||||
print(f"\n❌ 测试失败: {e}")
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
return False
|
||||
finally:
|
||||
tester.stop()
|
||||
|
||||
def main():
|
||||
import argparse
|
||||
|
||||
parser = argparse.ArgumentParser(description="增强智能测试工具")
|
||||
parser.add_argument("url", help="要测试的网站URL")
|
||||
parser.add_argument("--model", default="glm",
|
||||
choices=["claude", "openai", "glm", "mimo"],
|
||||
help="AI模型")
|
||||
parser.add_argument("--headless", action="store_true",
|
||||
help="无头模式")
|
||||
parser.add_argument("--no-login", action="store_true",
|
||||
help="跳过自动登录")
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
# 运行测试
|
||||
success = smart_test(args.url, args.model, args.headless, not args.no_login)
|
||||
sys.exit(0 if success else 1)
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user