kevin/life-echo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore/ 删除无用文件

Browse Source
This commit is contained in:
Kevin
2026-03-19 14:36:14 +08:00
parent 2f60858c9c
commit c6e07ce5ca
135 changed files with 2111 additions and 4510 deletions
Download Patch File Download Diff File Expand all files Collapse all files

224
api/tests/test_sms_verification.py
View File

@@ -37,22 +37,23 @@ refresh_token: Optional[str] = None
class Colors:
"""终端颜色"""
HEADER = '\033[95m'
OKBLUE = '\033[94m'
OKCYAN = '\033[96m'
OKGREEN = '\033[92m'
WARNING = '\033[93m'
FAIL = '\033[91m'
ENDC = '\033[0m'
BOLD = '\033[1m'
UNDERLINE = '\033[4m'
HEADER = "\033[95m"
OKBLUE = "\033[94m"
OKCYAN = "\033[96m"
OKGREEN = "\033[92m"
WARNING = "\033[93m"
FAIL = "\033[91m"
ENDC = "\033[0m"
BOLD = "\033[1m"
UNDERLINE = "\033[4m"
def print_header(text: str):
"""打印测试标题"""
print(f"\n{Colors.HEADER}{Colors.BOLD}{'='*60}{Colors.ENDC}")
print(f"\n{Colors.HEADER}{Colors.BOLD}{'=' * 60}{Colors.ENDC}")
print(f"{Colors.HEADER}{Colors.BOLD}{text}{Colors.ENDC}")
print(f"{Colors.HEADER}{Colors.BOLD}{'='*60}{Colors.ENDC}\n")
print(f"{Colors.HEADER}{Colors.BOLD}{'=' * 60}{Colors.ENDC}\n")
def print_success(text: str):
@@ -80,11 +81,11 @@ def make_request(
endpoint: str,
data: Optional[Dict[str, Any]] = None,
headers: Optional[Dict[str, str]] = None,
expected_status: int = 200
expected_status: int = 200,
) -> Optional[Dict[str, Any]]:
"""发送HTTP请求"""
url = f"{BASE_URL}{API_PREFIX}{endpoint}"
try:
if method.upper() == "GET":
response = requests.get(url, headers=headers)
@@ -97,9 +98,9 @@ def make_request(
else:
print_error(f"不支持的HTTP方法: {method}")
return None
print_info(f"{method.upper()} {endpoint} - Status: {response.status_code}")
if response.status_code == expected_status:
print_success(f"请求成功 (状态码: {response.status_code})")
try:
@@ -107,14 +108,18 @@ def make_request(
except:
return {"status": "success"}
else:
print_error(f"请求失败 (期望: {expected_status}, 实际: {response.status_code})")
print_error(
f"请求失败 (期望: {expected_status}, 实际: {response.status_code})"
)
try:
error_data = response.json()
print_error(f"错误信息: {json.dumps(error_data, ensure_ascii=False, indent=2)}")
print_error(
f"错误信息: {json.dumps(error_data, ensure_ascii=False, indent=2)}"
)
except:
print_error(f"响应内容: {response.text}")
return None
except requests.exceptions.ConnectionError:
print_error(f"连接失败: 无法连接到 {BASE_URL}")
print_warning("请确保后端服务正在运行")
@@ -127,41 +132,35 @@ def make_request(
def test_send_verification_code(phone: str, purpose: str) -> bool:
"""测试发送验证码"""
print_header(f"测试发送验证码 - {purpose}")
data = {
"phone": phone,
"purpose": purpose
}
data = {"phone": phone, "purpose": purpose}
result = make_request("POST", "/auth/sms/send", data=data)
if result:
print_success(f"验证码已发送: {result.get('message', '')}")
print_info(f"有效期: {result.get('expires_in', 0)} 秒")
return True
return False
def test_rate_limit(phone: str) -> bool:
"""测试频率限制"""
print_header("测试频率限制")
# 第一次发送应该成功
if not test_send_verification_code(phone, "register"):
return False
print_info("等待1秒后再次发送...")
time.sleep(1)
# 第二次发送应该被限制
data = {
"phone": phone,
"purpose": "register"
}
data = {"phone": phone, "purpose": "register"}
result = make_request("POST", "/auth/sms/send", data=data, expected_status=429)
if result is None:
print_success("频率限制生效")
return True
@@ -173,165 +172,144 @@ def test_rate_limit(phone: str) -> bool:
def test_register_with_sms(phone: str, code: str) -> bool:
"""测试验证码注册"""
print_header("测试验证码注册")
data = {
"phone": phone,
"code": code,
"password": TEST_PASSWORD,
"nickname": TEST_NICKNAME,
"email": TEST_EMAIL
"email": TEST_EMAIL,
}
result = make_request("POST", "/auth/register/sms", data=data, expected_status=201)
if result:
global access_token, refresh_token
access_token = result.get("access_token")
refresh_token = result.get("refresh_token")
print_success("注册成功")
print_info(f"Access Token: {access_token[:20]}...")
print_info(f"Refresh Token: {refresh_token[:20]}...")
return True
return False
def test_login_with_sms(phone: str, code: str) -> bool:
"""测试验证码登录"""
print_header("测试验证码登录")
data = {
"phone": phone,
"code": code
}
data = {"phone": phone, "code": code}
result = make_request("POST", "/auth/login/sms", data=data)
if result:
global access_token, refresh_token
access_token = result.get("access_token")
refresh_token = result.get("refresh_token")
print_success("登录成功")
print_info(f"Access Token: {access_token[:20]}...")
return True
return False
def test_reset_password(phone: str, code: str, new_password: str) -> bool:
"""测试重置密码"""
print_header("测试重置密码")
data = {
"phone": phone,
"code": code,
"new_password": new_password
}
data = {"phone": phone, "code": code, "new_password": new_password}
result = make_request("POST", "/auth/password/reset", data=data)
if result:
print_success(f"密码重置成功: {result.get('message', '')}")
return True
return False
def test_change_password(old_password: str, new_password: str) -> bool:
"""测试修改密码（已登录）"""
print_header("测试修改密码")
if not access_token:
print_error("未登录，无法测试")
return False
data = {
"old_password": old_password,
"new_password": new_password
}
headers = {
"Authorization": f"Bearer {access_token}"
}
data = {"old_password": old_password, "new_password": new_password}
headers = {"Authorization": f"Bearer {access_token}"}
result = make_request("POST", "/auth/password/change", data=data, headers=headers)
if result:
print_success(f"密码修改成功: {result.get('message', '')}")
return True
return False
def test_change_phone(new_phone: str, code: str) -> bool:
"""测试修改手机号"""
print_header("测试修改手机号")
if not access_token:
print_error("未登录，无法测试")
return False
data = {
"new_phone": new_phone,
"code": code
}
headers = {
"Authorization": f"Bearer {access_token}"
}
data = {"new_phone": new_phone, "code": code}
headers = {"Authorization": f"Bearer {access_token}"}
result = make_request("POST", "/auth/phone/change", data=data, headers=headers)
if result:
print_success(f"手机号修改成功")
print_info(f"新手机号: {result.get('phone', '')}")
return True
return False
def test_logout_all() -> bool:
"""测试登出所有设备"""
print_header("测试登出所有设备")
if not access_token:
print_error("未登录，无法测试")
return False
headers = {
"Authorization": f"Bearer {access_token}"
}
headers = {"Authorization": f"Bearer {access_token}"}
result = make_request("POST", "/auth/logout/all", headers=headers)
if result:
print_success(f"登出成功: {result.get('message', '')}")
return True
return False
def test_get_current_user() -> bool:
"""测试获取当前用户信息"""
print_header("测试获取当前用户信息")
if not access_token:
print_error("未登录，无法测试")
return False
headers = {
"Authorization": f"Bearer {access_token}"
}
headers = {"Authorization": f"Bearer {access_token}"}
result = make_request("GET", "/auth/me", headers=headers)
if result:
print_success("获取用户信息成功")
print_info(f"用户信息: {json.dumps(result, ensure_ascii=False, indent=2)}")
return True
return False
@@ -340,41 +318,43 @@ def interactive_test():
print_header("短信验证码功能交互式测试")
print_info("此模式需要您手动输入收到的验证码")
print_warning("请确保已配置腾讯云短信服务")
phone = input(f"\n请输入测试手机号 (默认: {TEST_PHONE}): ").strip() or TEST_PHONE
# 1. 测试发送验证码
if not test_send_verification_code(phone, "register"):
print_error("发送验证码失败，测试终止")
return
code = input("\n请输入收到的验证码: ").strip()
if not code or len(code) != 6:
print_error("验证码格式错误")
return
# 2. 测试注册
if test_register_with_sms(phone, code):
print_success("注册测试通过")
# 3. 测试获取用户信息
test_get_current_user()
# 4. 测试修改密码
if input("\n是否测试修改密码? (y/n): ").lower() == 'y':
if input("\n是否测试修改密码? (y/n): ").lower() == "y":
test_change_password(TEST_PASSWORD, NEW_PASSWORD)
# 5. 测试修改手机号
if input("\n是否测试修改手机号? (y/n): ").lower() == 'y':
new_phone = input(f"请输入新手机号 (默认: {NEW_PHONE}): ").strip() or NEW_PHONE
if input("\n是否测试修改手机号? (y/n): ").lower() == "y":
new_phone = (
input(f"请输入新手机号 (默认: {NEW_PHONE}): ").strip() or NEW_PHONE
)
if test_send_verification_code(new_phone, "change_phone"):
code = input("请输入收到的验证码: ").strip()
test_change_phone(new_phone, code)
# 6. 测试登出所有设备
if input("\n是否测试登出所有设备? (y/n): ").lower() == 'y':
if input("\n是否测试登出所有设备? (y/n): ").lower() == "y":
test_logout_all()
@@ -382,14 +362,14 @@ def automated_test():
"""自动化测试（需要mock验证码）"""
print_header("短信验证码功能自动化测试")
print_warning("此模式需要后端支持测试验证码（如：123456）")
# 测试发送验证码
test_send_verification_code(TEST_PHONE, "register")
# 等待一段时间
print_info("等待60秒以测试频率限制...")
time.sleep(60)
# 测试频率限制
test_rate_limit(TEST_PHONE)
@@ -400,14 +380,14 @@ if __name__ == "__main__":
print("短信验证码功能测试脚本")
print("=" * 60)
print(f"{Colors.ENDC}")
print("\n请选择测试模式:")
print("1. 交互式测试（需要真实短信验证码）")
print("2. 自动化测试（需要测试验证码支持）")
print("3. 仅测试API连接")
choice = input("\n请输入选项 (1/2/3): ").strip()
if choice == "1":
interactive_test()
elif choice == "2":
@@ -421,5 +401,5 @@ if __name__ == "__main__":
print_error("API连接失败")
else:
print_error("无效的选项")
print(f"\n{Colors.BOLD}{Colors.OKBLUE}测试完成{Colors.ENDC}\n")