You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
1.4 KiB
54 lines
1.4 KiB
2 months ago
|
"""
|
||
|
|
认证模块测试
|
||
|
|
"""
|
||
|
|
import pytest
|
||
|
|
from fastapi.testclient import TestClient
|
||
|
|
from app.main import app
|
||
|
|
|
||
|
|
client = TestClient(app)
|
||
|
|
|
||
|
|
|
||
|
|
class TestAuth:
|
||
|
|
"""认证测试类"""
|
||
|
|
|
||
|
|
def test_login_success(self):
|
||
|
|
"""测试登录成功"""
|
||
|
|
response = client.post(
|
||
|
|
"/api/v1/auth/login",
|
||
|
|
data={"username": "admin", "password": "admin123"}
|
||
|
|
)
|
||
|
|
# 注意：实际测试需要正确的密码
|
||
|
|
assert response.status_code in [200, 401]
|
||
|
|
|
||
|
|
def test_health_check(self):
|
||
|
|
"""测试健康检查"""
|
||
|
|
response = client.get("/health")
|
||
|
|
assert response.status_code == 200
|
||
|
|
assert response.json()["status"] == "healthy"
|
||
|
|
|
||
|
|
def test_root(self):
|
||
|
|
"""测试根路径"""
|
||
|
|
response = client.get("/")
|
||
|
|
assert response.status_code == 200
|
||
|
|
data = response.json()
|
||
|
|
assert "name" in data
|
||
|
|
assert "version" in data
|
||
|
|
|
||
|
|
|
||
|
|
class TestKline:
|
||
|
|
"""K 线数据测试类"""
|
||
|
|
|
||
|
|
def test_get_symbols(self):
|
||
|
|
"""测试获取品种列表"""
|
||
|
|
response = client.get("/api/v1/kline/symbols")
|
||
|
|
assert response.status_code == 200
|
||
|
|
|
||
|
|
def test_get_periods(self):
|
||
|
|
"""测试获取周期列表"""
|
||
|
|
response = client.get("/api/v1/kline/periods")
|
||
|
|
assert response.status_code == 200
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
pytest.main([__file__, "-v"])
|