note/work/AI/course/test_token_config.py
2025-11-19 10:16:05 +08:00

47 lines
1.5 KiB
Python

#!/usr/bin/env python3
"""测试 token 配置是否正确设置"""
from scraper import DEFAULT_TOKEN
def test_token_config():
"""测试 token 配置"""
print("测试 token 配置...")
# 检查 token 是否已设置
expected_token = 'MAC id="7F938B205F876FC398BCDC5BCE419D078A9A9DC46BC1C5EB5D458752DA28A954776C4459233C9F6209FA0EC2EC21AE85202FAE132D402538",nonce="1758355290351:STU4ZCMA",mac="cmPIHUYMwn6OiCanuD/OLV75xyyhxyGZzzEwFwMaKbc="'
if DEFAULT_TOKEN == expected_token:
print("✅ Token 配置正确")
print(f"Token: {DEFAULT_TOKEN}")
return True
else:
print("❌ Token 配置不正确")
print(f"期望: {expected_token}")
print(f"实际: {DEFAULT_TOKEN}")
return False
def test_argparse_default():
"""测试 argparse 默认值"""
import argparse
from scraper import DEFAULT_TOKEN
parser = argparse.ArgumentParser()
parser.add_argument("--token", default=DEFAULT_TOKEN)
args = parser.parse_args([]) # 不传递参数
if args.token == DEFAULT_TOKEN:
print("✅ argparse 默认值设置正确")
return True
else:
print("❌ argparse 默认值设置不正确")
return False
if __name__ == "__main__":
token_ok = test_token_config()
argparse_ok = test_argparse_default()
if token_ok and argparse_ok:
print("\n🎉 所有 token 配置测试通过!")
else:
print("\n❌ 部分测试失败,请检查配置")