107 lines
3.4 KiB
Python
107 lines
3.4 KiB
Python
|
"""
|
|||
|
|
测试登录页面配置功能
|
|||
|
|
验证通过配置文件控制登录页面类型（creator vs home）
|
|||
|
|
"""
|
|||
|
|
import sys
|
|||
|
|
from config import load_config
|
|||
|
|
|
|||
|
|
def test_config_reading():
|
|||
|
|
"""测试配置读取"""
|
|||
|
|
print("="*60)
|
|||
|
|
print("测试配置文件读取")
|
|||
|
|
print("="*60)
|
|||
|
|
|
|||
|
|
# 测试dev配置
|
|||
|
|
print("\n1. 测试开发环境配置 (config.dev.yaml)")
|
|||
|
|
config_dev = load_config('dev')
|
|||
|
|
login_page = config_dev.get_str('login.page', 'creator')
|
|||
|
|
login_headless = config_dev.get_bool('login.headless', False)
|
|||
|
|
|
|||
|
|
print(f" login.page = {login_page}")
|
|||
|
|
print(f" login.headless = {login_headless}")
|
|||
|
|
|
|||
|
|
# 根据配置决定预热URL
|
|||
|
|
if login_page == "home":
|
|||
|
|
preheat_url = "https://www.xiaohongshu.com"
|
|||
|
|
else:
|
|||
|
|
preheat_url = "https://creator.xiaohongshu.com/login"
|
|||
|
|
|
|||
|
|
print(f" 预热URL = {preheat_url}")
|
|||
|
|
|
|||
|
|
# 测试prod配置
|
|||
|
|
print("\n2. 测试生产环境配置 (config.prod.yaml)")
|
|||
|
|
config_prod = load_config('prod')
|
|||
|
|
login_page_prod = config_prod.get_str('login.page', 'creator')
|
|||
|
|
login_headless_prod = config_prod.get_bool('login.headless', False)
|
|||
|
|
|
|||
|
|
print(f" login.page = {login_page_prod}")
|
|||
|
|
print(f" login.headless = {login_headless_prod}")
|
|||
|
|
|
|||
|
|
if login_page_prod == "home":
|
|||
|
|
preheat_url_prod = "https://www.xiaohongshu.com"
|
|||
|
|
else:
|
|||
|
|
preheat_url_prod = "https://creator.xiaohongshu.com/login"
|
|||
|
|
|
|||
|
|
print(f" 预热URL = {preheat_url_prod}")
|
|||
|
|
|
|||
|
|
print("\n" + "="*60)
|
|||
|
|
print("✅ 配置读取测试完成")
|
|||
|
|
print("="*60)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_api_parameter_override():
|
|||
|
|
"""测试API参数覆盖配置"""
|
|||
|
|
print("\n" + "="*60)
|
|||
|
|
print("测试API参数覆盖配置")
|
|||
|
|
print("="*60)
|
|||
|
|
|
|||
|
|
config = load_config('dev')
|
|||
|
|
default_login_page = config.get_str('login.page', 'creator')
|
|||
|
|
|
|||
|
|
# 模拟不同的API参数情况
|
|||
|
|
test_cases = [
|
|||
|
|
(None, "应使用配置默认值"),
|
|||
|
|
("creator", "API指定creator"),
|
|||
|
|
("home", "API指定home"),
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
for api_param, description in test_cases:
|
|||
|
|
login_page = api_param if api_param else default_login_page
|
|||
|
|
print(f"\n场景: {description}")
|
|||
|
|
print(f" 配置默认值 = {default_login_page}")
|
|||
|
|
print(f" API参数 = {api_param}")
|
|||
|
|
print(f" 最终使用 = {login_page}")
|
|||
|
|
|
|||
|
|
# 决定URL
|
|||
|
|
if login_page == "home":
|
|||
|
|
url = "https://www.xiaohongshu.com"
|
|||
|
|
page_name = "小红书首页"
|
|||
|
|
else:
|
|||
|
|
url = "https://creator.xiaohongshu.com/login"
|
|||
|
|
page_name = "创作者中心"
|
|||
|
|
|
|||
|
|
print(f" → 将访问: {page_name} ({url})")
|
|||
|
|
|
|||
|
|
print("\n" + "="*60)
|
|||
|
|
print("✅ API参数覆盖测试完成")
|
|||
|
|
print("="*60)
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
try:
|
|||
|
|
test_config_reading()
|
|||
|
|
test_api_parameter_override()
|
|||
|
|
|
|||
|
|
print("\n🎉 所有测试通过！")
|
|||
|
|
print("\n使用说明：")
|
|||
|
|
print("1. 在 config.dev.yaml 或 config.prod.yaml 中修改 login.page 配置")
|
|||
|
|
print("2. 可选值: creator (创作者中心) 或 home (小红书首页)")
|
|||
|
|
print("3. API请求中的 login_page 参数可以覆盖配置文件的默认值")
|
|||
|
|
print("4. 如果API请求不传 login_page 参数，将使用配置文件中的默认值")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"\n❌ 测试失败: {str(e)}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|
|||
|
|
sys.exit(1)
|