67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
import requests
|
|
import json
|
|
|
|
# 创建会话以保持登录状态
|
|
session = requests.Session()
|
|
|
|
def login():
|
|
"""登录系统"""
|
|
url = "http://localhost:5000/api/v1/auth/login"
|
|
headers = {"Content-Type": "application/json"}
|
|
data = {
|
|
"username": "admin",
|
|
"password": "admin123"
|
|
}
|
|
|
|
try:
|
|
response = session.post(url, headers=headers, data=json.dumps(data))
|
|
print(f"Login Status Code: {response.status_code}")
|
|
print(f"Login Response: {response.text}")
|
|
return response.status_code == 200
|
|
except Exception as e:
|
|
print(f"Login Error: {e}")
|
|
return False
|
|
|
|
def test_create_product():
|
|
"""测试创建产品API"""
|
|
url = "http://localhost:5000/api/v1/products"
|
|
headers = {"Content-Type": "application/json"}
|
|
data = {
|
|
"product_name": "测试产品",
|
|
"description": "这是一个测试产品"
|
|
}
|
|
|
|
try:
|
|
response = session.post(url, headers=headers, data=json.dumps(data))
|
|
print(f"Create Product Status Code: {response.status_code}")
|
|
print(f"Create Product Response: {response.text}")
|
|
return response.status_code == 200
|
|
except Exception as e:
|
|
print(f"Create Product Error: {e}")
|
|
return False
|
|
|
|
def test_get_logs():
|
|
"""测试获取日志API"""
|
|
url = "http://localhost:5000/api/v1/logs"
|
|
try:
|
|
response = session.get(url)
|
|
print(f"Get Logs Status Code: {response.status_code}")
|
|
print(f"Get Logs Response: {response.text}")
|
|
return response.status_code == 200
|
|
except Exception as e:
|
|
print(f"Get Logs Error: {e}")
|
|
return False
|
|
|
|
if __name__ == "__main__":
|
|
# 登录
|
|
if login():
|
|
print("登录成功")
|
|
# 测试创建产品
|
|
if test_create_product():
|
|
print("创建产品成功")
|
|
# 测试获取日志
|
|
test_get_logs()
|
|
else:
|
|
print("创建产品失败")
|
|
else:
|
|
print("登录失败") |