48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import requests
|
||
|
|
import json
|
||
|
|
|
||
|
|
# 测试创建管理员功能
|
||
|
|
def test_create_admin():
|
||
|
|
# API基础URL
|
||
|
|
base_url = "http://127.0.0.1:5000"
|
||
|
|
|
||
|
|
# 先登录获取session
|
||
|
|
login_data = {
|
||
|
|
"username": "admin",
|
||
|
|
"password": "admin123"
|
||
|
|
}
|
||
|
|
|
||
|
|
session = requests.Session()
|
||
|
|
login_response = session.post(f"{base_url}/login", data=login_data)
|
||
|
|
|
||
|
|
if login_response.status_code != 200 and login_response.status_code != 302:
|
||
|
|
print(f"登录失败: {login_response.status_code}")
|
||
|
|
print(login_response.text)
|
||
|
|
return
|
||
|
|
|
||
|
|
print("登录成功")
|
||
|
|
|
||
|
|
# 测试创建管理员
|
||
|
|
admin_data = {
|
||
|
|
"username": "test_admin",
|
||
|
|
"email": "test@example.com",
|
||
|
|
"password": "test123",
|
||
|
|
"role": 0,
|
||
|
|
"status": 1
|
||
|
|
}
|
||
|
|
|
||
|
|
response = session.post(f"{base_url}/api/v1/admins", json=admin_data)
|
||
|
|
|
||
|
|
print(f"创建管理员状态码: {response.status_code}")
|
||
|
|
print(f"响应内容: {response.text}")
|
||
|
|
|
||
|
|
if response.status_code == 200:
|
||
|
|
print("管理员创建成功!")
|
||
|
|
else:
|
||
|
|
print("管理员创建失败!")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
test_create_admin()
|