64 lines
2.2 KiB
Python
64 lines
2.2 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
测试系统设置修改功能
|
||
"""
|
||
|
||
import requests
|
||
import json
|
||
|
||
# 测试获取系统设置
|
||
def test_get_settings():
|
||
print("=== 测试获取系统设置 ===")
|
||
try:
|
||
response = requests.get('http://localhost:5000/api/admin/settings')
|
||
if response.status_code == 200:
|
||
settings = response.json()
|
||
print("获取设置成功")
|
||
for setting in settings:
|
||
if setting['key'] in ['daily_quota', 'max_file_size']:
|
||
print(f"{setting['key']}: {setting['value']} ({type(setting['value'])})")
|
||
else:
|
||
print(f"获取设置失败: {response.status_code}")
|
||
print(response.text)
|
||
except Exception as e:
|
||
print(f"获取设置出错: {e}")
|
||
|
||
# 测试更新每日配额
|
||
def test_update_daily_quota():
|
||
print("\n=== 测试更新每日配额 ===")
|
||
try:
|
||
# 注意：这里需要管理员权限，实际测试需要JWT token
|
||
data = {
|
||
'daily_quota': 10
|
||
}
|
||
response = requests.put('http://localhost:5000/api/admin/settings',
|
||
json=data,
|
||
headers={'Content-Type': 'application/json'})
|
||
print(f"更新每日配额响应: {response.status_code}")
|
||
print(response.text)
|
||
except Exception as e:
|
||
print(f"更新每日配额出错: {e}")
|
||
|
||
# 测试更新文件大小
|
||
def test_update_max_file_size():
|
||
print("\n=== 测试更新文件大小 ===")
|
||
try:
|
||
# 注意：这里需要管理员权限，实际测试需要JWT token
|
||
data = {
|
||
'max_file_size': 20 * 1024 * 1024 # 20MB in bytes
|
||
}
|
||
response = requests.put('http://localhost:5000/api/admin/settings',
|
||
json=data,
|
||
headers={'Content-Type': 'application/json'})
|
||
print(f"更新文件大小响应: {response.status_code}")
|
||
print(response.text)
|
||
except Exception as e:
|
||
print(f"更新文件大小出错: {e}")
|
||
|
||
if __name__ == '__main__':
|
||
print("开始测试系统设置功能...")
|
||
test_get_settings()
|
||
test_update_daily_quota()
|
||
test_update_max_file_size()
|
||
print("\n测试完成！") |