307 lines
10 KiB
Python
307 lines
10 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
配置服务测试脚本
|
||
用于验证配置服务的各项功能
|
||
"""
|
||
|
||
import requests
|
||
import json
|
||
import time
|
||
|
||
# 配置
|
||
SERVICE_HOST = "127.0.0.1"
|
||
SERVICE_PORT = 5000
|
||
BASE_URL = f"http://{SERVICE_HOST}:{SERVICE_PORT}"
|
||
|
||
def print_section(title):
|
||
"""打印分隔线"""
|
||
print("\n" + "="*60)
|
||
print(f" {title}")
|
||
print("="*60)
|
||
|
||
def test_health_check():
|
||
"""测试健康检查接口"""
|
||
print_section("测试1: 健康检查")
|
||
|
||
try:
|
||
response = requests.get(f"{BASE_URL}/api/health", timeout=3)
|
||
print(f"状态码: {response.status_code}")
|
||
|
||
if response.status_code == 200:
|
||
data = response.json()
|
||
print(f"服务状态: {data.get('status')}")
|
||
print(f"服务名称: {data.get('service')}")
|
||
print(f"服务版本: {data.get('version')}")
|
||
print("✓ 健康检查通过")
|
||
return True
|
||
else:
|
||
print("✗ 健康检查失败")
|
||
return False
|
||
except requests.exceptions.ConnectionError:
|
||
print("✗ 无法连接到配置服务")
|
||
print(" 请确保配置服务已启动: python config_service.py")
|
||
return False
|
||
except Exception as e:
|
||
print(f"✗ 错误: {e}")
|
||
return False
|
||
|
||
def test_get_config_text():
|
||
"""测试获取配置文件（文本格式）"""
|
||
print_section("测试2: 获取配置文件（文本格式）")
|
||
|
||
try:
|
||
response = requests.get(f"{BASE_URL}/api/config/text?path=config.json", timeout=5)
|
||
print(f"状态码: {response.status_code}")
|
||
|
||
if response.status_code == 200:
|
||
data = response.json()
|
||
if data.get("success"):
|
||
config_text = data.get("text", "")
|
||
print(f"配置文件路径: {data.get('path')}")
|
||
print(f"配置文件大小: {len(config_text)} 字节")
|
||
print(f"前100个字符: {config_text[:100]}...")
|
||
|
||
# 验证JSON格式
|
||
try:
|
||
json.loads(config_text)
|
||
print("✓ JSON格式正确")
|
||
except json.JSONDecodeError:
|
||
print("✗ JSON格式错误")
|
||
return False
|
||
|
||
print("✓ 获取配置文件成功")
|
||
return True
|
||
else:
|
||
print(f"✗ 获取失败: {data.get('error')}")
|
||
return False
|
||
else:
|
||
print("✗ 获取配置文件失败")
|
||
return False
|
||
except Exception as e:
|
||
print(f"✗ 错误: {e}")
|
||
return False
|
||
|
||
def test_get_config_json():
|
||
"""测试获取配置文件（JSON格式）"""
|
||
print_section("测试3: 获取配置文件（JSON格式）")
|
||
|
||
try:
|
||
response = requests.get(f"{BASE_URL}/api/config?path=config.json", timeout=5)
|
||
print(f"状态码: {response.status_code}")
|
||
|
||
if response.status_code == 200:
|
||
data = response.json()
|
||
if data.get("success"):
|
||
config = data.get("config", {})
|
||
print(f"配置文件路径: {data.get('path')}")
|
||
print(f"配置项数量: {len(config)}")
|
||
|
||
# 显示部分配置内容
|
||
if "influx" in config:
|
||
print(" - 包含 InfluxDB 配置")
|
||
if "placeholders" in config:
|
||
print(f" - 包含 {len(config.get('placeholders', {}))} 个占位符")
|
||
|
||
print("✓ 获取配置文件成功")
|
||
return config
|
||
else:
|
||
print(f"✗ 获取失败: {data.get('error')}")
|
||
return None
|
||
else:
|
||
print("✗ 获取配置文件失败")
|
||
return None
|
||
except Exception as e:
|
||
print(f"✗ 错误: {e}")
|
||
return None
|
||
|
||
def test_update_config_validation():
|
||
"""测试配置更新的JSON验证"""
|
||
print_section("测试4: JSON格式验证")
|
||
|
||
# 测试无效的JSON
|
||
invalid_json = '{"invalid": json content}'
|
||
|
||
try:
|
||
payload = {
|
||
"text": invalid_json,
|
||
"path": "test_invalid.json"
|
||
}
|
||
response = requests.post(f"{BASE_URL}/api/config/text", json=payload, timeout=5)
|
||
print(f"状态码: {response.status_code}")
|
||
|
||
if response.status_code == 400:
|
||
data = response.json()
|
||
print(f"预期的错误: {data.get('error')}")
|
||
print("✓ JSON验证功能正常")
|
||
return True
|
||
else:
|
||
print("✗ JSON验证没有正确工作")
|
||
return False
|
||
except Exception as e:
|
||
print(f"✗ 错误: {e}")
|
||
return False
|
||
|
||
def test_full_workflow():
|
||
"""测试完整的读取-修改-保存流程"""
|
||
print_section("测试5: 完整工作流程测试")
|
||
|
||
test_file = "test_config_temp.json"
|
||
|
||
# 1. 创建测试配置
|
||
print("\n步骤1: 创建测试配置文件")
|
||
test_config = {
|
||
"test": True,
|
||
"timestamp": time.time(),
|
||
"message": "这是一个测试配置"
|
||
}
|
||
|
||
try:
|
||
payload = {
|
||
"text": json.dumps(test_config, ensure_ascii=False, indent=2),
|
||
"path": test_file
|
||
}
|
||
response = requests.post(f"{BASE_URL}/api/config/text", json=payload, timeout=5)
|
||
|
||
if response.status_code == 200 and response.json().get("success"):
|
||
print("✓ 创建测试配置成功")
|
||
else:
|
||
print("✗ 创建测试配置失败")
|
||
return False
|
||
except Exception as e:
|
||
print(f"✗ 错误: {e}")
|
||
return False
|
||
|
||
# 2. 读取测试配置
|
||
print("\n步骤2: 读取测试配置文件")
|
||
try:
|
||
response = requests.get(f"{BASE_URL}/api/config?path={test_file}", timeout=5)
|
||
|
||
if response.status_code == 200:
|
||
data = response.json()
|
||
if data.get("success"):
|
||
config = data.get("config", {})
|
||
if config.get("test") == True and "message" in config:
|
||
print("✓ 读取测试配置成功")
|
||
print(f" 消息: {config.get('message')}")
|
||
else:
|
||
print("✗ 配置内容不匹配")
|
||
return False
|
||
else:
|
||
print("✗ 读取失败")
|
||
return False
|
||
else:
|
||
print("✗ 读取测试配置失败")
|
||
return False
|
||
except Exception as e:
|
||
print(f"✗ 错误: {e}")
|
||
return False
|
||
|
||
# 3. 修改测试配置
|
||
print("\n步骤3: 修改测试配置文件")
|
||
test_config["message"] = "配置已被修改"
|
||
test_config["modified"] = True
|
||
|
||
try:
|
||
payload = {
|
||
"text": json.dumps(test_config, ensure_ascii=False, indent=2),
|
||
"path": test_file
|
||
}
|
||
response = requests.post(f"{BASE_URL}/api/config/text", json=payload, timeout=5)
|
||
|
||
if response.status_code == 200 and response.json().get("success"):
|
||
print("✓ 修改测试配置成功")
|
||
else:
|
||
print("✗ 修改测试配置失败")
|
||
return False
|
||
except Exception as e:
|
||
print(f"✗ 错误: {e}")
|
||
return False
|
||
|
||
# 4. 验证修改结果
|
||
print("\n步骤4: 验证修改结果")
|
||
try:
|
||
response = requests.get(f"{BASE_URL}/api/config?path={test_file}", timeout=5)
|
||
|
||
if response.status_code == 200:
|
||
data = response.json()
|
||
if data.get("success"):
|
||
config = data.get("config", {})
|
||
if config.get("modified") == True and config.get("message") == "配置已被修改":
|
||
print("✓ 修改验证成功")
|
||
print(f" 新消息: {config.get('message')}")
|
||
else:
|
||
print("✗ 修改未生效")
|
||
return False
|
||
else:
|
||
print("✗ 验证失败")
|
||
return False
|
||
else:
|
||
print("✗ 验证请求失败")
|
||
return False
|
||
except Exception as e:
|
||
print(f"✗ 错误: {e}")
|
||
return False
|
||
|
||
print("\n✓ 完整工作流程测试通过")
|
||
print(f" 注意: 测试文件 {test_file} 已创建，您可以手动删除")
|
||
return True
|
||
|
||
def run_all_tests():
|
||
"""运行所有测试"""
|
||
print("\n" + "="*60)
|
||
print(" 配置服务功能测试")
|
||
print("="*60)
|
||
print(f"\n服务地址: {BASE_URL}")
|
||
print("开始测试...\n")
|
||
|
||
results = []
|
||
|
||
# 测试1: 健康检查
|
||
results.append(("健康检查", test_health_check()))
|
||
|
||
# 如果健康检查失败，不继续后续测试
|
||
if not results[0][1]:
|
||
print("\n" + "="*60)
|
||
print(" 配置服务未运行，无法继续测试")
|
||
print(" 请先启动配置服务: python config_service.py")
|
||
print("="*60)
|
||
return
|
||
|
||
# 测试2: 获取配置（文本）
|
||
results.append(("获取配置（文本）", test_get_config_text()))
|
||
|
||
# 测试3: 获取配置（JSON）
|
||
config = test_get_config_json()
|
||
results.append(("获取配置（JSON）", config is not None))
|
||
|
||
# 测试4: JSON验证
|
||
results.append(("JSON格式验证", test_update_config_validation()))
|
||
|
||
# 测试5: 完整工作流程
|
||
results.append(("完整工作流程", test_full_workflow()))
|
||
|
||
# 打印测试结果摘要
|
||
print_section("测试结果摘要")
|
||
|
||
total_tests = len(results)
|
||
passed_tests = sum(1 for _, result in results if result)
|
||
failed_tests = total_tests - passed_tests
|
||
|
||
for test_name, result in results:
|
||
status = "✓ 通过" if result else "✗ 失败"
|
||
print(f"{test_name:20s} - {status}")
|
||
|
||
print(f"\n总测试数: {total_tests}")
|
||
print(f"通过: {passed_tests}")
|
||
print(f"失败: {failed_tests}")
|
||
|
||
if failed_tests == 0:
|
||
print("\n🎉 所有测试通过！配置服务运行正常。")
|
||
else:
|
||
print(f"\n⚠️ 有 {failed_tests} 个测试失败，请检查配置服务。")
|
||
|
||
if __name__ == "__main__":
|
||
run_all_tests()
|
||
|