226 lines
7.2 KiB
Python
226 lines
7.2 KiB
Python
"""
|
|
测试工具函数和配置
|
|
"""
|
|
import pytest
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
import sys
|
|
import os
|
|
|
|
project_root = Path(__file__).parent.parent.absolute()
|
|
sys.path.insert(0, str(project_root))
|
|
os.chdir(project_root)
|
|
|
|
from app.utils import make_response, register_cors
|
|
from app.settings import config
|
|
|
|
|
|
class TestMakeResponse:
|
|
"""测试 make_response 工具函数"""
|
|
|
|
def test_make_response_default_success(self):
|
|
"""测试默认成功响应"""
|
|
response = make_response()
|
|
assert response['status'] == 'success'
|
|
assert response['message'] == '操作成功'
|
|
assert response['data'] == {}
|
|
assert response['errors'] == []
|
|
assert 'timestamp' in response
|
|
|
|
def test_make_response_custom_status(self):
|
|
"""测试自定义状态"""
|
|
response = make_response(status='error')
|
|
assert response['status'] == 'error'
|
|
assert response['message'] == '操作失败'
|
|
|
|
def test_make_response_custom_message(self):
|
|
"""测试自定义消息"""
|
|
response = make_response(message='自定义消息')
|
|
assert response['message'] == '自定义消息'
|
|
|
|
def test_make_response_with_data(self):
|
|
"""测试带数据的响应"""
|
|
test_data = {'key': 'value', 'number': 42}
|
|
response = make_response(data=test_data)
|
|
assert response['data'] == test_data
|
|
|
|
def test_make_response_with_errors(self):
|
|
"""测试带错误的响应"""
|
|
errors = ['错误 1', '错误 2']
|
|
response = make_response(errors=errors)
|
|
assert response['errors'] == errors
|
|
assert len(response['errors']) == 2
|
|
|
|
def test_make_response_with_extra(self):
|
|
"""测试带额外信息的响应"""
|
|
extra = {'extra_key': 'extra_value'}
|
|
response = make_response(extra=extra)
|
|
assert response['extra_key'] == 'extra_value'
|
|
|
|
def test_make_response_timestamp_format(self):
|
|
"""测试时间戳格式"""
|
|
response = make_response()
|
|
timestamp = response['timestamp']
|
|
# 验证 ISO 8601 格式
|
|
assert 'T' in timestamp
|
|
assert 'Z' in timestamp
|
|
# 验证可以解析
|
|
try:
|
|
datetime.strptime(timestamp, '%Y-%m-%dT%H:%M:%SZ')
|
|
except ValueError:
|
|
pytest.fail("时间戳格式不正确")
|
|
|
|
def test_make_response_combined(self):
|
|
"""测试组合响应"""
|
|
response = make_response(
|
|
status='success',
|
|
message='测试消息',
|
|
data={'result': 'ok'},
|
|
errors=[],
|
|
extra={'page': 1}
|
|
)
|
|
assert response['status'] == 'success'
|
|
assert response['message'] == '测试消息'
|
|
assert response['data']['result'] == 'ok'
|
|
assert response['errors'] == []
|
|
assert response['page'] == 1
|
|
|
|
|
|
class TestRegisterCORS:
|
|
"""测试 CORS 注册"""
|
|
|
|
def test_register_cors_sets_headers(self):
|
|
"""测试 CORS 设置正确的头"""
|
|
from flask import Flask
|
|
app = Flask(__name__)
|
|
register_cors(app)
|
|
|
|
# 验证注册了 after_request 处理器
|
|
assert len(app.after_request_funcs) > 0
|
|
|
|
def test_register_cors_options_route(self):
|
|
"""测试 OPTIONS 路由注册"""
|
|
from flask import Flask
|
|
app = Flask(__name__)
|
|
register_cors(app)
|
|
|
|
rules = [rule.rule for rule in app.url_map.iter_rules()]
|
|
assert '/config' in rules
|
|
|
|
|
|
class TestConfig:
|
|
"""测试配置模块"""
|
|
|
|
def test_config_is_dict(self):
|
|
"""测试配置是字典"""
|
|
assert isinstance(config, dict)
|
|
|
|
def test_config_max_content_length(self):
|
|
"""测试最大内容长度配置"""
|
|
assert 'MAX_CONTENT_LENGTH' in config
|
|
assert config['MAX_CONTENT_LENGTH'] == 500 * 1024 * 1024 # 500MB
|
|
|
|
def test_config_send_file_max_age(self):
|
|
"""测试文件发送最大年龄配置"""
|
|
assert 'SEND_FILE_MAX_AGE_DEFAULT' in config
|
|
assert config['SEND_FILE_MAX_AGE_DEFAULT'] == 300 # 300 秒
|
|
|
|
def test_config_input_dir(self):
|
|
"""测试输入目录配置"""
|
|
assert 'INPUT_DIR' in config
|
|
assert config['INPUT_DIR'] == 'input'
|
|
|
|
def test_config_output_dir(self):
|
|
"""测试输出目录配置"""
|
|
assert 'OUTPUT_DIR' in config
|
|
assert config['OUTPUT_DIR'] == 'output'
|
|
|
|
def test_config_task_timeout(self):
|
|
"""测试任务超时配置"""
|
|
assert 'TASK_TIMEOUT' in config
|
|
assert config['TASK_TIMEOUT'] == 600 # 600 秒
|
|
|
|
def test_config_api_port(self):
|
|
"""测试 API 端口配置"""
|
|
assert 'API_PORT' in config
|
|
assert config['API_PORT'] == 5000
|
|
assert isinstance(config['API_PORT'], int)
|
|
assert config['API_PORT'] > 0
|
|
|
|
def test_config_video_port(self):
|
|
"""测试视频端口配置"""
|
|
assert 'VIDEO_PORT' in config
|
|
assert config['VIDEO_PORT'] == 8086
|
|
assert isinstance(config['VIDEO_PORT'], int)
|
|
assert config['VIDEO_PORT'] > 0
|
|
|
|
def test_config_all_required_keys(self):
|
|
"""测试所有必需的键"""
|
|
required_keys = [
|
|
'MAX_CONTENT_LENGTH',
|
|
'SEND_FILE_MAX_AGE_DEFAULT',
|
|
'INPUT_DIR',
|
|
'OUTPUT_DIR',
|
|
'TASK_TIMEOUT',
|
|
'API_PORT',
|
|
'VIDEO_PORT'
|
|
]
|
|
for key in required_keys:
|
|
assert key in config, f"配置缺少必需的键：{key}"
|
|
|
|
def test_config_no_extra_keys(self):
|
|
"""测试没有多余的键（可选）"""
|
|
allowed_keys = [
|
|
'MAX_CONTENT_LENGTH',
|
|
'SEND_FILE_MAX_AGE_DEFAULT',
|
|
'INPUT_DIR',
|
|
'OUTPUT_DIR',
|
|
'TASK_TIMEOUT',
|
|
'API_PORT',
|
|
'VIDEO_PORT'
|
|
]
|
|
for key in config.keys():
|
|
assert key in allowed_keys, f"配置包含未预期的键：{key}"
|
|
|
|
|
|
class TestResponseFormat:
|
|
"""测试响应格式规范"""
|
|
|
|
def test_response_has_required_fields(self):
|
|
"""测试响应包含必需字段"""
|
|
response = make_response()
|
|
required_fields = ['status', 'data', 'errors', 'message', 'timestamp']
|
|
for field in required_fields:
|
|
assert field in response, f"响应缺少必需字段：{field}"
|
|
|
|
def test_response_status_values(self):
|
|
"""测试响应状态值"""
|
|
valid_statuses = ['success', 'error']
|
|
for status in valid_statuses:
|
|
response = make_response(status=status)
|
|
assert response['status'] == status
|
|
|
|
def test_response_data_is_dict(self):
|
|
"""测试数据字段是字典"""
|
|
response = make_response()
|
|
assert isinstance(response['data'], dict)
|
|
|
|
def test_response_errors_is_list(self):
|
|
"""测试错误字段是列表"""
|
|
response = make_response()
|
|
assert isinstance(response['errors'], list)
|
|
|
|
def test_response_message_is_string(self):
|
|
"""测试消息字段是字符串"""
|
|
response = make_response()
|
|
assert isinstance(response['message'], str)
|
|
|
|
def test_response_timestamp_is_string(self):
|
|
"""测试时间戳字段是字符串"""
|
|
response = make_response()
|
|
assert isinstance(response['timestamp'], str)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
pytest.main([__file__, '-v'])
|