208 lines
6.6 KiB
Python
208 lines
6.6 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
功能使用API测试
|
||
测试核心扣费逻辑、VIP配额管理
|
||
"""
|
||
|
||
import pytest
|
||
import requests
|
||
from conftest import API_BASE_URL
|
||
|
||
|
||
class TestFeatureUsage:
|
||
"""功能使用测试"""
|
||
|
||
def test_use_feature_normal_user(self, test_user, test_feature):
|
||
"""测试普通用户使用功能（扣积分）"""
|
||
request_data = {
|
||
'username': test_user['username'],
|
||
'feature_key': test_feature['feature_key'],
|
||
'device_id': 'test_device'
|
||
}
|
||
|
||
response = requests.post(
|
||
f"{API_BASE_URL}/feature/use",
|
||
json=request_data
|
||
)
|
||
|
||
assert response.status_code == 200
|
||
result = response.json()
|
||
assert result['code'] == 200
|
||
|
||
data = result['data']
|
||
assert data['success'] is True
|
||
assert data['cost_type'] == 'points'
|
||
assert data['points_cost'] == test_feature['points_cost']
|
||
assert data['points_remaining'] == test_user['points'] - test_feature['points_cost']
|
||
|
||
def test_use_feature_insufficient_points(self, test_user, test_feature, db_cursor):
|
||
"""测试积分不足"""
|
||
# 设置用户积分为0
|
||
db_cursor.execute(
|
||
"UPDATE users SET points = 0 WHERE username = %s",
|
||
(test_user['username'],)
|
||
)
|
||
db_cursor.connection.commit()
|
||
|
||
request_data = {
|
||
'username': test_user['username'],
|
||
'feature_key': test_feature['feature_key'],
|
||
'device_id': 'test_device'
|
||
}
|
||
|
||
response = requests.post(
|
||
f"{API_BASE_URL}/feature/use",
|
||
json=request_data
|
||
)
|
||
|
||
assert response.status_code == 402
|
||
assert '积分不足' in response.json()['detail']
|
||
|
||
def test_use_feature_vip_user_with_quota(self, vip_user, test_feature):
|
||
"""测试VIP用户使用配额"""
|
||
request_data = {
|
||
'username': vip_user['username'],
|
||
'feature_key': test_feature['feature_key'],
|
||
'device_id': 'test_device'
|
||
}
|
||
|
||
response = requests.post(
|
||
f"{API_BASE_URL}/feature/use",
|
||
json=request_data
|
||
)
|
||
|
||
assert response.status_code == 200
|
||
result = response.json()
|
||
|
||
data = result['data']
|
||
# VIP用户应该使用配额（如果vip_points_cost=0）
|
||
assert data['cost_type'] in ['vip_quota', 'points']
|
||
|
||
if data['cost_type'] == 'vip_quota':
|
||
assert data['vip_remaining_quota'] == vip_user['vip_daily_quota'] - 1
|
||
|
||
def test_use_disabled_feature(self, test_user, test_feature, db_cursor):
|
||
"""测试使用已禁用的功能"""
|
||
# 禁用功能
|
||
db_cursor.execute(
|
||
"UPDATE features_config SET enabled = 0 WHERE feature_key = %s",
|
||
(test_feature['feature_key'],)
|
||
)
|
||
db_cursor.connection.commit()
|
||
|
||
request_data = {
|
||
'username': test_user['username'],
|
||
'feature_key': test_feature['feature_key'],
|
||
'device_id': 'test_device'
|
||
}
|
||
|
||
response = requests.post(
|
||
f"{API_BASE_URL}/feature/use",
|
||
json=request_data
|
||
)
|
||
|
||
assert response.status_code == 400
|
||
assert '禁用' in response.json()['detail']
|
||
|
||
# 恢复
|
||
db_cursor.execute(
|
||
"UPDATE features_config SET enabled = 1 WHERE feature_key = %s",
|
||
(test_feature['feature_key'],)
|
||
)
|
||
db_cursor.connection.commit()
|
||
|
||
def test_use_nonexistent_feature(self, test_user):
|
||
"""测试使用不存在的功能"""
|
||
request_data = {
|
||
'username': test_user['username'],
|
||
'feature_key': 'nonexistent_feature',
|
||
'device_id': 'test_device'
|
||
}
|
||
|
||
response = requests.post(
|
||
f"{API_BASE_URL}/feature/use",
|
||
json=request_data
|
||
)
|
||
|
||
assert response.status_code == 400
|
||
|
||
def test_use_feature_nonexistent_user(self, test_feature):
|
||
"""测试不存在的用户使用功能"""
|
||
request_data = {
|
||
'username': 'nonexistent_user',
|
||
'feature_key': test_feature['feature_key'],
|
||
'device_id': 'test_device'
|
||
}
|
||
|
||
response = requests.post(
|
||
f"{API_BASE_URL}/feature/use",
|
||
json=request_data
|
||
)
|
||
|
||
assert response.status_code == 404
|
||
|
||
|
||
class TestFeatureUsageLogging:
|
||
"""功能使用日志测试"""
|
||
|
||
def test_usage_log_created(self, test_user, test_feature, db_cursor):
|
||
"""测试使用后创建日志"""
|
||
# 使用前记录数
|
||
db_cursor.execute(
|
||
"SELECT COUNT(*) as count FROM feature_usage_logs WHERE username = %s",
|
||
(test_user['username'],)
|
||
)
|
||
count_before = db_cursor.fetchone()['count']
|
||
|
||
# 使用功能
|
||
request_data = {
|
||
'username': test_user['username'],
|
||
'feature_key': test_feature['feature_key'],
|
||
'device_id': 'test_device'
|
||
}
|
||
|
||
requests.post(f"{API_BASE_URL}/feature/use", json=request_data)
|
||
|
||
# 使用后记录数
|
||
db_cursor.execute(
|
||
"SELECT COUNT(*) as count FROM feature_usage_logs WHERE username = %s",
|
||
(test_user['username'],)
|
||
)
|
||
count_after = db_cursor.fetchone()['count']
|
||
|
||
assert count_after == count_before + 1
|
||
|
||
def test_points_history_created(self, test_user, test_feature, db_cursor):
|
||
"""测试积分历史记录"""
|
||
# 使用前记录数
|
||
db_cursor.execute(
|
||
"SELECT COUNT(*) as count FROM points_history WHERE username = %s",
|
||
(test_user['username'],)
|
||
)
|
||
count_before = db_cursor.fetchone()['count']
|
||
|
||
# 使用功能
|
||
request_data = {
|
||
'username': test_user['username'],
|
||
'feature_key': test_feature['feature_key'],
|
||
'device_id': 'test_device'
|
||
}
|
||
|
||
response = requests.post(f"{API_BASE_URL}/feature/use", json=request_data)
|
||
|
||
# 只有扣积分时才记录积分历史
|
||
if response.status_code == 200:
|
||
data = response.json()['data']
|
||
if data['cost_type'] == 'points':
|
||
db_cursor.execute(
|
||
"SELECT COUNT(*) as count FROM points_history WHERE username = %s",
|
||
(test_user['username'],)
|
||
)
|
||
count_after = db_cursor.fetchone()['count']
|
||
assert count_after == count_before + 1
|
||
|
||
|
||
if __name__ == '__main__':
|
||
pytest.main([__file__, '-v'])
|
||
|