126 lines
3.9 KiB
Python
Executable File
126 lines
3.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""测试通知发送脚本."""
|
|
|
|
import json
|
|
import sys
|
|
from datetime import datetime
|
|
|
|
import requests
|
|
|
|
|
|
def test_notification(server_url: str, api_key: str):
|
|
"""测试发送通知."""
|
|
|
|
# 测试简化版 API
|
|
simple_test_data = {
|
|
"api_key": api_key,
|
|
"content": "这是一条测试短信，用于验证通知转发功能是否正常工作。",
|
|
"sender": "测试发送者"
|
|
}
|
|
|
|
try:
|
|
print(f"🔄 正在向 {server_url} 发送测试通知...")
|
|
|
|
# 发送请求到简化版 API
|
|
response = requests.post(
|
|
f"{server_url}/notify/simple",
|
|
json=simple_test_data,
|
|
headers={"Content-Type": "application/json"},
|
|
timeout=10
|
|
)
|
|
|
|
# 检查响应
|
|
if response.status_code == 200:
|
|
result = response.json()
|
|
if result.get("success"):
|
|
print("✅ 测试通知发送成功!")
|
|
print(f"📱 请检查你的 Android 设备是否收到通知")
|
|
else:
|
|
print(f"❌ 通知发送失败: {result.get('message')}")
|
|
return False
|
|
else:
|
|
print(f"❌ HTTP 请求失败: {response.status_code}")
|
|
print(f"响应内容: {response.text}")
|
|
return False
|
|
|
|
except requests.exceptions.ConnectionError:
|
|
print(f"❌ 无法连接到服务器 {server_url}")
|
|
print("请确认服务器正在运行且地址正确")
|
|
return False
|
|
except requests.exceptions.Timeout:
|
|
print("❌ 请求超时")
|
|
return False
|
|
except Exception as e:
|
|
print(f"❌ 发生错误: {e}")
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
def test_server_status(server_url: str):
|
|
"""测试服务器状态."""
|
|
try:
|
|
print(f"🔄 检查服务器状态...")
|
|
|
|
# 健康检查
|
|
health_response = requests.get(f"{server_url}/health", timeout=5)
|
|
if health_response.status_code == 200:
|
|
print("✅ 服务器健康检查通过")
|
|
else:
|
|
print(f"⚠️ 健康检查失败: {health_response.status_code}")
|
|
|
|
# 状态信息
|
|
status_response = requests.get(f"{server_url}/status", timeout=5)
|
|
if status_response.status_code == 200:
|
|
status = status_response.json()
|
|
print(f"📊 服务器状态:")
|
|
print(f" 版本: {status.get('version')}")
|
|
print(f" 运行时间: {status.get('uptime', 0):.2f} 秒")
|
|
print(f" 已发送通知: {status.get('notifications_sent', 0)}")
|
|
print(f" 失败通知: {status.get('notifications_failed', 0)}")
|
|
else:
|
|
print(f"⚠️ 获取状态信息失败: {status_response.status_code}")
|
|
|
|
except Exception as e:
|
|
print(f"❌ 检查服务器状态时出错: {e}")
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
def main():
|
|
"""主函数."""
|
|
if len(sys.argv) < 3:
|
|
print("使用方法: python test_notification.py <server_url> <api_key>")
|
|
print("示例: python test_notification.py http://localhost:8000 your-api-key")
|
|
sys.exit(1)
|
|
|
|
server_url = sys.argv[1].rstrip('/')
|
|
api_key = sys.argv[2]
|
|
|
|
print("SMS Forwarder 通知测试")
|
|
print("=" * 30)
|
|
|
|
# 测试服务器状态
|
|
if not test_server_status(server_url):
|
|
print("❌ 服务器状态检查失败")
|
|
sys.exit(1)
|
|
|
|
print()
|
|
|
|
# 测试通知发送
|
|
if test_notification(server_url, api_key):
|
|
print("\n🎉 所有测试通过!")
|
|
print("如果你的 Android 设备收到了通知，说明配置正确。")
|
|
else:
|
|
print("\n❌ 测试失败")
|
|
print("请检查:")
|
|
print("1. 服务器配置是否正确")
|
|
print("2. API 密钥是否正确")
|
|
print("3. 推送服务是否正确配置")
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|