278 lines
7.7 KiB
Python
278 lines
7.7 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
Sentinel 集成测试脚本
|
||
|
||
验证 Sentinel 解决方案是否正确集成
|
||
"""
|
||
|
||
import sys
|
||
import asyncio
|
||
from pathlib import Path
|
||
|
||
|
||
def test_imports():
|
||
"""测试所有必要的模块导入"""
|
||
print("=" * 60)
|
||
print("测试 1: 模块导入")
|
||
print("=" * 60)
|
||
|
||
try:
|
||
print("✓ 导入 utils.logger...")
|
||
from utils.logger import logger
|
||
|
||
print("✓ 导入 utils.crypto...")
|
||
from utils.crypto import generate_oai_did, generate_random_password
|
||
|
||
print("✓ 导入 utils.fingerprint...")
|
||
from utils.fingerprint import BrowserFingerprint
|
||
|
||
print("✓ 导入 core.session...")
|
||
from core.session import OAISession
|
||
|
||
print("✓ 导入 core.sentinel...")
|
||
from core.sentinel import SentinelHandler
|
||
|
||
print("✓ 导入 reference.sentinel_solver...")
|
||
from reference.sentinel_solver import SentinelSolver
|
||
|
||
print("✓ 导入 reference.js_executor...")
|
||
from reference.js_executor import JSExecutor
|
||
|
||
print("\n✅ 所有模块导入成功！\n")
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"\n❌ 导入失败: {e}\n")
|
||
import traceback
|
||
traceback.print_exc()
|
||
return False
|
||
|
||
|
||
def test_node_availability():
|
||
"""测试 Node.js 是否可用"""
|
||
print("=" * 60)
|
||
print("测试 2: Node.js 环境检查")
|
||
print("=" * 60)
|
||
|
||
import subprocess
|
||
|
||
try:
|
||
result = subprocess.run(
|
||
["node", "--version"],
|
||
capture_output=True,
|
||
text=True,
|
||
timeout=5
|
||
)
|
||
|
||
if result.returncode == 0:
|
||
version = result.stdout.strip()
|
||
print(f"✓ Node.js 已安装: {version}")
|
||
return True
|
||
else:
|
||
print(f"❌ Node.js 执行失败: {result.stderr}")
|
||
return False
|
||
|
||
except FileNotFoundError:
|
||
print("❌ Node.js 未安装或不在 PATH 中")
|
||
print(" 请安装 Node.js: https://nodejs.org/")
|
||
return False
|
||
except Exception as e:
|
||
print(f"❌ Node.js 检查失败: {e}")
|
||
return False
|
||
|
||
|
||
def test_sdk_file():
|
||
"""测试 SDK 文件是否存在"""
|
||
print("\n" + "=" * 60)
|
||
print("测试 3: SDK 文件检查")
|
||
print("=" * 60)
|
||
|
||
sdk_path = Path("/home/carry/myprj/gptAutoPlus/sdk/sdk.js")
|
||
|
||
if sdk_path.exists():
|
||
size = sdk_path.stat().st_size
|
||
print(f"✓ SDK 文件存在: {sdk_path}")
|
||
print(f" 文件大小: {size:,} bytes ({size/1024:.1f} KB)")
|
||
return True
|
||
else:
|
||
print(f"❌ SDK 文件不存在: {sdk_path}")
|
||
print(" 请确保 sdk/sdk.js 文件存在")
|
||
return False
|
||
|
||
|
||
def test_fingerprint():
|
||
"""测试浏览器指纹生成"""
|
||
print("\n" + "=" * 60)
|
||
print("测试 4: 浏览器指纹生成")
|
||
print("=" * 60)
|
||
|
||
try:
|
||
from utils.fingerprint import BrowserFingerprint
|
||
|
||
fp = BrowserFingerprint()
|
||
config_array = fp.get_config_array()
|
||
|
||
print(f"✓ 指纹生成成功")
|
||
print(f" Session ID: {fp.session_id}")
|
||
print(f" 配置数组长度: {len(config_array)}")
|
||
print(f" 配置数组前 3 项: {config_array[:3]}")
|
||
|
||
if len(config_array) == 18:
|
||
print("✓ 配置数组长度正确 (18 个元素)")
|
||
return True
|
||
else:
|
||
print(f"❌ 配置数组长度错误: {len(config_array)} (期望 18)")
|
||
return False
|
||
|
||
except Exception as e:
|
||
print(f"❌ 指纹生成失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
return False
|
||
|
||
|
||
async def test_sentinel_token():
|
||
"""测试 Sentinel Token 生成"""
|
||
print("\n" + "=" * 60)
|
||
print("测试 5: Sentinel Token 生成")
|
||
print("=" * 60)
|
||
|
||
try:
|
||
from core.session import OAISession
|
||
from core.sentinel import SentinelHandler
|
||
|
||
print("✓ 创建测试会话...")
|
||
session = OAISession()
|
||
|
||
print(f"✓ Session 创建成功，oai-did: {session.oai_did}")
|
||
|
||
print("✓ 初始化 SentinelHandler...")
|
||
sentinel = SentinelHandler(session)
|
||
|
||
print("✓ 生成 Sentinel Token...")
|
||
print(" （这可能需要几秒钟，正在执行 PoW 计算...）")
|
||
|
||
token = await sentinel.get_token()
|
||
|
||
print(f"\n✅ Sentinel Token 生成成功！")
|
||
print(f" Token 前缀: {token[:30]}...")
|
||
print(f" Token 长度: {len(token)}")
|
||
|
||
# 验证 token 格式
|
||
if token.startswith("gAAAAA"):
|
||
print("✓ Token 格式正确")
|
||
return True
|
||
else:
|
||
print(f"⚠️ Token 格式异常: {token[:20]}...")
|
||
return True # 仍然算成功，因为可能是格式变化
|
||
|
||
except Exception as e:
|
||
print(f"\n❌ Sentinel Token 生成失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
return False
|
||
|
||
|
||
def test_crypto_utils():
|
||
"""测试加密工具"""
|
||
print("\n" + "=" * 60)
|
||
print("测试 6: 加密工具")
|
||
print("=" * 60)
|
||
|
||
try:
|
||
from utils.crypto import (
|
||
generate_oai_did,
|
||
generate_random_password,
|
||
validate_oai_did,
|
||
validate_password
|
||
)
|
||
|
||
# 测试 oai-did 生成
|
||
oai_did = generate_oai_did()
|
||
print(f"✓ OAI-DID 生成: {oai_did}")
|
||
|
||
is_valid = validate_oai_did(oai_did)
|
||
print(f"✓ OAI-DID 验证: {is_valid}")
|
||
|
||
# 测试密码生成
|
||
password = generate_random_password()
|
||
print(f"✓ 密码生成: {password}")
|
||
|
||
is_valid, error = validate_password(password)
|
||
print(f"✓ 密码验证: {is_valid} {f'({error})' if error else ''}")
|
||
|
||
if is_valid:
|
||
print("\n✅ 加密工具测试通过！")
|
||
return True
|
||
else:
|
||
print(f"\n❌ 密码验证失败: {error}")
|
||
return False
|
||
|
||
except Exception as e:
|
||
print(f"\n❌ 加密工具测试失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
return False
|
||
|
||
|
||
async def main():
|
||
"""运行所有测试"""
|
||
print("\n" + "=" * 60)
|
||
print(" OpenAI 注册系统 - Sentinel 集成测试")
|
||
print("=" * 60)
|
||
print()
|
||
|
||
results = []
|
||
|
||
# 运行所有测试
|
||
results.append(("模块导入", test_imports()))
|
||
results.append(("Node.js 环境", test_node_availability()))
|
||
results.append(("SDK 文件", test_sdk_file()))
|
||
results.append(("浏览器指纹", test_fingerprint()))
|
||
results.append(("加密工具", test_crypto_utils()))
|
||
results.append(("Sentinel Token", await test_sentinel_token()))
|
||
|
||
# 打印总结
|
||
print("\n" + "=" * 60)
|
||
print(" 测试总结")
|
||
print("=" * 60)
|
||
|
||
for name, passed in results:
|
||
status = "✅ 通过" if passed else "❌ 失败"
|
||
print(f" {name:20s} {status}")
|
||
|
||
print("=" * 60)
|
||
|
||
total = len(results)
|
||
passed = sum(1 for _, p in results if p)
|
||
|
||
print(f"\n总计: {passed}/{total} 个测试通过")
|
||
|
||
if passed == total:
|
||
print("\n🎉 所有测试通过！系统已准备就绪。")
|
||
print("\n下一步:")
|
||
print(" 1. 配置邮箱（修改 .env 文件）")
|
||
print(" 2. 运行主程序: python main.py")
|
||
return 0
|
||
else:
|
||
print("\n⚠️ 部分测试失败，请检查上述错误信息。")
|
||
print("\n常见问题:")
|
||
print(" - Node.js 未安装: 请安装 Node.js v16+")
|
||
print(" - SDK 文件缺失: 确保 sdk/sdk.js 存在")
|
||
print(" - 依赖未安装: 运行 pip install -e .")
|
||
return 1
|
||
|
||
|
||
if __name__ == "__main__":
|
||
try:
|
||
exit_code = asyncio.run(main())
|
||
sys.exit(exit_code)
|
||
except KeyboardInterrupt:
|
||
print("\n\n⚠️ 测试被用户中断")
|
||
sys.exit(1)
|
||
except Exception as e:
|
||
print(f"\n\n❌ 测试程序异常: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
sys.exit(1)
|