import requests
|
import json
|
import time
|
|
# API的基础URL
|
BASE_URL = "http://localhost:9676/api/v1"
|
|
def test_health_check():
|
"""测试健康检查端点"""
|
print("测试健康检查端点...")
|
|
try:
|
response = requests.get("http://localhost:9676/health")
|
print(f"健康检查响应: {response.status_code}")
|
print(f"健康检查响应正文: {response.text}")
|
return response
|
except Exception as e:
|
print(f"健康检查时出错: {e}")
|
return None
|
|
|
def test_save_contenttype():
|
"""测试ContentType保存端点"""
|
print("\n测试ContentType保存端点...")
|
|
# 要保存的ContentType数据
|
data_to_save = {"code": "FAQ", "name": "常见问题"}
|
|
try:
|
response = requests.post(f"{BASE_URL}/type/save", json=data_to_save)
|
print(f"ContentType保存响应: {response.status_code}")
|
print(f"ContentType保存响应正文: {response.text}")
|
return response
|
except Exception as e:
|
print(f"ContentType保存测试时出错: {e}")
|
return None
|
|
|
def test_get_contenttype():
|
"""测试ContentType检索端点"""
|
print("\n测试ContentType检索端点...")
|
|
try:
|
# 获取所有ContentType
|
response = requests.get(f"{BASE_URL}/type/get")
|
print(f"ContentType获取响应: {response.status_code}")
|
print(f"ContentType获取响应正文: {response.text}")
|
return response
|
except Exception as e:
|
print(f"ContentType获取测试时出错: {e}")
|
return None
|
|
|
def test_save_content():
|
"""测试Content保存端点"""
|
print("\n测试Content保存端点...")
|
|
# 先获取FAQ类型的ID
|
try:
|
response = requests.get(f"{BASE_URL}/type/get?code=FAQ")
|
if response.status_code == 200:
|
data = response.json()
|
if data["count"] > 0:
|
content_type_id = data["data"][0]["id"]
|
|
# 要保存的Content数据
|
data_to_save = {
|
"type": content_type_id,
|
"question": "这是什么系统?",
|
"answer": "这是一个AI管理系统。",
|
}
|
|
response = requests.post(f"{BASE_URL}/content/save", json=data_to_save)
|
print(f"Content保存响应: {response.status_code}")
|
print(f"Content保存响应正文: {response.text}")
|
return response
|
else:
|
print("未找到FAQ ContentType，无法测试Content保存")
|
return None
|
else:
|
print("获取FAQ ContentType失败，无法测试Content保存")
|
return None
|
except Exception as e:
|
print(f"Content保存测试时出错: {e}")
|
return None
|
|
|
def test_get_content():
|
"""测试Content检索端点"""
|
print("\n测试Content检索端点...")
|
|
try:
|
# 获取所有Content
|
response = requests.get(f"{BASE_URL}/content/get")
|
print(f"Content获取响应: {response.status_code}")
|
print(f"Content获取响应正文: {response.text}")
|
return response
|
except Exception as e:
|
print(f"Content获取测试时出错: {e}")
|
return None
|
|
|
def test_get_content_by_type():
|
"""测试按类型检索Content"""
|
print("\n测试按类型检索Content...")
|
|
try:
|
# 获取FAQ类型的Content
|
response = requests.get(f"{BASE_URL}/content/get?type=1")
|
print(f"按类型Content获取响应: {response.status_code}")
|
print(f"按类型Content获取响应正文: {response.text}")
|
return response
|
except Exception as e:
|
print(f"按类型Content获取测试时出错: {e}")
|
return None
|
|
|
if __name__ == "__main__":
|
print("开始API测试...\n")
|
|
# 等待片刻以便服务器准备就绪（如果正在运行）
|
time.sleep(1)
|
|
# 首先测试健康检查
|
test_health_check()
|
|
# 测试ContentType保存
|
test_save_contenttype()
|
|
# 测试ContentType检索
|
test_get_contenttype()
|
|
# 测试Content保存
|
test_save_content()
|
|
# 测试Content检索
|
test_get_content()
|
|
# 测试按类型检索Content
|
test_get_content_by_type()
|
|
print("\n测试完成!")
|
