|
|
"""测试服务 - 对应Go的internal/service/test.go"""
|
|
|
import asyncio
|
|
|
import json
|
|
|
from datetime import datetime, timedelta
|
|
|
from typing import List, Optional
|
|
|
from threading import RLock
|
|
|
|
|
|
import httpx
|
|
|
import websockets
|
|
|
|
|
|
from app.models import (
|
|
|
APITestListData, APITestCategory, APITestCase,
|
|
|
APITestRequest, APITestResult,
|
|
|
WSTestListData, WSTestCase, WSTestRequest, WSTestResult, WSMessage,
|
|
|
TestHistoryRequest, TestHistoryData
|
|
|
)
|
|
|
from app.core.logger import info, error
|
|
|
|
|
|
|
|
|
class TestService:
|
|
|
"""测试服务"""
|
|
|
|
|
|
def __init__(self):
|
|
|
self.lock = RLock()
|
|
|
self.api_history: List[APITestResult] = []
|
|
|
self.ws_history: List[WSTestResult] = []
|
|
|
self.history_size = 100
|
|
|
|
|
|
def get_api_test_list(self) -> APITestListData:
|
|
|
"""获取API测试列表"""
|
|
|
# 固定交易时间：2026年3月2日到2026年3月6日
|
|
|
test_start = datetime(2026, 3, 2)
|
|
|
test_end = datetime(2026, 3, 6)
|
|
|
|
|
|
categories = [
|
|
|
APITestCategory(
|
|
|
name="股票接口",
|
|
|
items=[
|
|
|
APITestCase(
|
|
|
id="stock_klines",
|
|
|
name="查询股票K线",
|
|
|
method="GET",
|
|
|
path="/v1/stock/klines/{symbol}",
|
|
|
description="查询指定股票的K线数据",
|
|
|
params={
|
|
|
"symbol": "000001.SZ",
|
|
|
"start": test_start.strftime("%Y%m%d"),
|
|
|
"end": test_end.strftime("%Y%m%d"),
|
|
|
"freq": "1d",
|
|
|
"adjust": "qfq"
|
|
|
}
|
|
|
),
|
|
|
APITestCase(
|
|
|
id="stock_symbols",
|
|
|
name="查询股票列表",
|
|
|
method="GET",
|
|
|
path="/v1/stock/symbols",
|
|
|
description="获取所有可用股票标的",
|
|
|
params={"page": "1", "size": "20"}
|
|
|
),
|
|
|
APITestCase(
|
|
|
id="stock_batch",
|
|
|
name="批量查询股票K线",
|
|
|
method="POST",
|
|
|
path="/v1/stock/klines/batch",
|
|
|
description="批量查询多只股票K线",
|
|
|
body={
|
|
|
"symbols": ["000001.SZ", "000002.SZ"],
|
|
|
"start": test_start.strftime("%Y%m%d"),
|
|
|
"end": test_end.strftime("%Y%m%d"),
|
|
|
"freq": "1d"
|
|
|
}
|
|
|
),
|
|
|
APITestCase(
|
|
|
id="stock_calendar",
|
|
|
name="查询交易日历",
|
|
|
method="GET",
|
|
|
path="/v1/stock/trading-dates",
|
|
|
description="查询股票交易日历",
|
|
|
params={
|
|
|
"start": test_start.strftime("%Y%m%d"),
|
|
|
"end": test_end.strftime("%Y%m%d")
|
|
|
}
|
|
|
),
|
|
|
]
|
|
|
),
|
|
|
APITestCategory(
|
|
|
name="期货接口",
|
|
|
items=[
|
|
|
APITestCase(
|
|
|
id="futures_klines",
|
|
|
name="查询期货K线",
|
|
|
method="GET",
|
|
|
path="/v1/futures/klines/{symbol}",
|
|
|
description="查询指定期货合约的K线数据",
|
|
|
params={
|
|
|
"symbol": "CU2504.SHFE",
|
|
|
"start": test_start.strftime("%Y%m%d"),
|
|
|
"end": test_end.strftime("%Y%m%d"),
|
|
|
"freq": "1d"
|
|
|
}
|
|
|
),
|
|
|
APITestCase(
|
|
|
id="futures_symbols",
|
|
|
name="查询期货列表",
|
|
|
method="GET",
|
|
|
path="/v1/futures/symbols",
|
|
|
description="获取所有可用期货标的",
|
|
|
params={"page": "1", "size": "20"}
|
|
|
),
|
|
|
APITestCase(
|
|
|
id="futures_batch",
|
|
|
name="批量查询期货K线",
|
|
|
method="POST",
|
|
|
path="/v1/futures/klines/batch",
|
|
|
description="批量查询多个期货合约K线",
|
|
|
body={
|
|
|
"symbols": ["CU2504.SHFE", "RB2505.SHFE"],
|
|
|
"start": test_start.strftime("%Y%m%d"),
|
|
|
"end": test_end.strftime("%Y%m%d"),
|
|
|
"freq": "1d"
|
|
|
}
|
|
|
),
|
|
|
APITestCase(
|
|
|
id="futures_contracts",
|
|
|
name="查询合约列表",
|
|
|
method="GET",
|
|
|
path="/v1/futures/contracts",
|
|
|
description="根据品种查询可交易合约",
|
|
|
params={"underlying": "CU", "exchange": "SHFE"}
|
|
|
),
|
|
|
APITestCase(
|
|
|
id="futures_calendar",
|
|
|
name="查询期货交易日历",
|
|
|
method="GET",
|
|
|
path="/v1/futures/trading-dates",
|
|
|
description="查询期货交易日历",
|
|
|
params={
|
|
|
"start": test_start.strftime("%Y%m%d"),
|
|
|
"end": test_end.strftime("%Y%m%d")
|
|
|
}
|
|
|
),
|
|
|
]
|
|
|
),
|
|
|
APITestCategory(
|
|
|
name="管理接口",
|
|
|
items=[
|
|
|
APITestCase(
|
|
|
id="admin_health",
|
|
|
name="健康检查",
|
|
|
method="GET",
|
|
|
path="/v1/admin/health",
|
|
|
description="检查服务健康状态",
|
|
|
params={}
|
|
|
),
|
|
|
APITestCase(
|
|
|
id="admin_source_status",
|
|
|
name="数据源状态",
|
|
|
method="GET",
|
|
|
path="/v1/admin/source/status",
|
|
|
description="获取当前数据源状态",
|
|
|
params={}
|
|
|
),
|
|
|
APITestCase(
|
|
|
id="admin_source_switch",
|
|
|
name="切换数据源",
|
|
|
method="POST",
|
|
|
path="/v1/admin/source/switch",
|
|
|
description="切换到指定数据源（amazingdata）",
|
|
|
body={
|
|
|
"asset_class": "all",
|
|
|
"source": "amazingdata",
|
|
|
"sync_backfill": False
|
|
|
}
|
|
|
),
|
|
|
APITestCase(
|
|
|
id="admin_system_status",
|
|
|
name="系统状态",
|
|
|
method="GET",
|
|
|
path="/v1/admin/system/status",
|
|
|
description="获取系统运行状态和资源使用情况",
|
|
|
params={}
|
|
|
),
|
|
|
APITestCase(
|
|
|
id="admin_config_list",
|
|
|
name="查询配置列表",
|
|
|
method="GET",
|
|
|
path="/v1/admin/config",
|
|
|
description="获取所有配置项列表",
|
|
|
params={}
|
|
|
),
|
|
|
APITestCase(
|
|
|
id="admin_config_update",
|
|
|
name="更新配置",
|
|
|
method="PUT",
|
|
|
path="/v1/admin/config",
|
|
|
description="更新系统配置",
|
|
|
body={
|
|
|
"key": "server.mode",
|
|
|
"value": "debug",
|
|
|
"description": "服务器运行模式"
|
|
|
}
|
|
|
),
|
|
|
APITestCase(
|
|
|
id="admin_reload_config",
|
|
|
name="热加载配置",
|
|
|
method="POST",
|
|
|
path="/v1/admin/system/reload",
|
|
|
description="重新加载配置文件",
|
|
|
body={}
|
|
|
),
|
|
|
]
|
|
|
),
|
|
|
APITestCategory(
|
|
|
name="适配器管理",
|
|
|
items=[
|
|
|
APITestCase(
|
|
|
id="admin_adapters_list",
|
|
|
name="适配器列表",
|
|
|
method="GET",
|
|
|
path="/v1/admin/adapters",
|
|
|
description="获取所有数据源适配器列表",
|
|
|
params={}
|
|
|
),
|
|
|
APITestCase(
|
|
|
id="admin_adapter_toggle",
|
|
|
name="切换适配器状态",
|
|
|
method="POST",
|
|
|
path="/v1/admin/adapters/toggle",
|
|
|
description="启用或禁用适配器",
|
|
|
body={
|
|
|
"name": "amazingdata",
|
|
|
"enable": True
|
|
|
}
|
|
|
),
|
|
|
APITestCase(
|
|
|
id="admin_adapter_config",
|
|
|
name="更新适配器配置",
|
|
|
method="PUT",
|
|
|
path="/v1/admin/adapters/config",
|
|
|
description="更新适配器配置参数",
|
|
|
body={
|
|
|
"name": "amazingdata",
|
|
|
"config": {
|
|
|
"timeout": "60"
|
|
|
}
|
|
|
}
|
|
|
),
|
|
|
]
|
|
|
),
|
|
|
APITestCategory(
|
|
|
name="测试管理",
|
|
|
items=[
|
|
|
APITestCase(
|
|
|
id="admin_test_history",
|
|
|
name="测试历史",
|
|
|
method="GET",
|
|
|
path="/v1/admin/tests/history",
|
|
|
description="获取测试执行历史记录",
|
|
|
params={"type": "api", "limit": "20"}
|
|
|
),
|
|
|
]
|
|
|
),
|
|
|
]
|
|
|
|
|
|
return APITestListData(categories=categories, base_url="")
|
|
|
|
|
|
async def run_api_test(self, base_url: str, req: APITestRequest) -> APITestResult:
|
|
|
"""执行API测试"""
|
|
|
# 获取测试用例
|
|
|
test_list = self.get_api_test_list()
|
|
|
|
|
|
test_case = None
|
|
|
for cat in test_list.categories:
|
|
|
for item in cat.items:
|
|
|
if item.id == req.id:
|
|
|
test_case = item
|
|
|
break
|
|
|
if test_case:
|
|
|
break
|
|
|
|
|
|
if not test_case:
|
|
|
raise ValueError(f"Test case not found: {req.id}")
|
|
|
|
|
|
# 合并参数
|
|
|
params = dict(test_case.params)
|
|
|
if req.params:
|
|
|
params.update(req.params)
|
|
|
|
|
|
# 构建URL
|
|
|
url = base_url + test_case.path
|
|
|
for k, v in params.items():
|
|
|
url = url.replace(f"{{{k}}}", str(v))
|
|
|
|
|
|
# 添加查询参数
|
|
|
if test_case.method == "GET" and params:
|
|
|
query_parts = []
|
|
|
for k, v in params.items():
|
|
|
if f"{{{k}}}" not in test_case.path:
|
|
|
query_parts.append(f"{k}={v}")
|
|
|
if query_parts:
|
|
|
url += "?" + "&".join(query_parts)
|
|
|
|
|
|
# 准备请求体
|
|
|
body = req.body if req.body is not None else test_case.body
|
|
|
|
|
|
# 执行请求
|
|
|
start_time = datetime.now()
|
|
|
|
|
|
async with httpx.AsyncClient() as client:
|
|
|
try:
|
|
|
headers = {"X-API-Key": "test-api-key"}
|
|
|
|
|
|
if test_case.method == "GET":
|
|
|
response = await client.get(url, headers=headers, timeout=30)
|
|
|
elif test_case.method == "POST":
|
|
|
response = await client.post(
|
|
|
url, json=body, headers=headers, timeout=30
|
|
|
)
|
|
|
else:
|
|
|
raise ValueError(f"Unsupported method: {test_case.method}")
|
|
|
|
|
|
latency = int((datetime.now() - start_time).total_seconds() * 1000)
|
|
|
|
|
|
result = APITestResult(
|
|
|
id=int(datetime.now().timestamp()),
|
|
|
case_id=req.id,
|
|
|
name=test_case.name,
|
|
|
success=200 <= response.status_code < 300,
|
|
|
status_code=response.status_code,
|
|
|
latency=latency,
|
|
|
request={
|
|
|
"method": test_case.method,
|
|
|
"url": url,
|
|
|
"body": body
|
|
|
},
|
|
|
response=response.json() if response.headers.get("content-type", "").startswith("application/json") else response.text,
|
|
|
timestamp=datetime.now()
|
|
|
)
|
|
|
|
|
|
self._add_api_history(result)
|
|
|
return result
|
|
|
|
|
|
except Exception as e:
|
|
|
latency = int((datetime.now() - start_time).total_seconds() * 1000)
|
|
|
result = APITestResult(
|
|
|
id=int(datetime.now().timestamp()),
|
|
|
case_id=req.id,
|
|
|
name=test_case.name,
|
|
|
success=False,
|
|
|
latency=latency,
|
|
|
request={
|
|
|
"method": test_case.method,
|
|
|
"url": url,
|
|
|
"body": body
|
|
|
},
|
|
|
error=str(e),
|
|
|
timestamp=datetime.now()
|
|
|
)
|
|
|
self._add_api_history(result)
|
|
|
return result
|
|
|
|
|
|
def get_ws_test_list(self) -> WSTestListData:
|
|
|
"""获取WebSocket测试列表"""
|
|
|
cases = [
|
|
|
WSTestCase(
|
|
|
id="ws_subscribe_stock",
|
|
|
name="订阅股票行情",
|
|
|
description="订阅单只股票实时行情",
|
|
|
action="subscribe",
|
|
|
symbols=["000001.SZ"]
|
|
|
),
|
|
|
WSTestCase(
|
|
|
id="ws_subscribe_futures",
|
|
|
name="订阅期货行情",
|
|
|
description="订阅单个期货合约实时行情",
|
|
|
action="subscribe",
|
|
|
symbols=["CU2504.SHFE"]
|
|
|
),
|
|
|
WSTestCase(
|
|
|
id="ws_subscribe_multi",
|
|
|
name="批量订阅",
|
|
|
description="同时订阅多个标的",
|
|
|
action="subscribe",
|
|
|
symbols=["000001.SZ", "000002.SZ", "CU2504.SHFE"]
|
|
|
),
|
|
|
WSTestCase(
|
|
|
id="ws_subscribe_many",
|
|
|
name="压力测试-大量订阅",
|
|
|
description="订阅大量标的测试性能",
|
|
|
action="subscribe",
|
|
|
symbols=[
|
|
|
"000001.SZ", "000002.SZ", "000063.SZ", "000333.SZ",
|
|
|
"000538.SZ", "000568.SZ", "000651.SZ", "000725.SZ",
|
|
|
"000768.SZ", "000858.SZ"
|
|
|
]
|
|
|
),
|
|
|
WSTestCase(
|
|
|
id="ws_unsubscribe",
|
|
|
name="取消订阅",
|
|
|
description="取消订阅标的",
|
|
|
action="unsubscribe",
|
|
|
symbols=["000001.SZ"]
|
|
|
),
|
|
|
WSTestCase(
|
|
|
id="ws_unsubscribe_all",
|
|
|
name="取消全部订阅",
|
|
|
description="取消所有已订阅标的",
|
|
|
action="unsubscribe",
|
|
|
symbols=["000001.SZ", "000002.SZ", "CU2504.SHFE"]
|
|
|
),
|
|
|
WSTestCase(
|
|
|
id="ws_heartbeat",
|
|
|
name="心跳检测",
|
|
|
description="测试WebSocket连接心跳",
|
|
|
action="subscribe",
|
|
|
symbols=["000001.SZ"]
|
|
|
),
|
|
|
WSTestCase(
|
|
|
id="ws_invalid_symbol",
|
|
|
name="无效标的测试",
|
|
|
description="测试订阅无效标的的错误处理",
|
|
|
action="subscribe",
|
|
|
symbols=["INVALID.CODE"]
|
|
|
),
|
|
|
WSTestCase(
|
|
|
id="ws_empty_symbols",
|
|
|
name="空订阅测试",
|
|
|
description="测试空标的列表的处理",
|
|
|
action="subscribe",
|
|
|
symbols=[]
|
|
|
),
|
|
|
WSTestCase(
|
|
|
id="ws_resubscribe",
|
|
|
name="重新订阅",
|
|
|
description="取消后重新订阅同一标的",
|
|
|
action="subscribe",
|
|
|
symbols=["000001.SZ"]
|
|
|
),
|
|
|
]
|
|
|
|
|
|
return WSTestListData(cases=cases, ws_url="")
|
|
|
|
|
|
async def run_ws_test(self, ws_url: str, req: WSTestRequest) -> WSTestResult:
|
|
|
"""执行WebSocket测试"""
|
|
|
# 获取测试用例
|
|
|
test_list = self.get_ws_test_list()
|
|
|
|
|
|
test_case = None
|
|
|
for item in test_list.cases:
|
|
|
if item.id == req.id:
|
|
|
test_case = item
|
|
|
break
|
|
|
|
|
|
if not test_case:
|
|
|
raise ValueError(f"Test case not found: {req.id}")
|
|
|
|
|
|
# 使用自定义标的
|
|
|
symbols = req.symbols if req.symbols else test_case.symbols
|
|
|
|
|
|
result = WSTestResult(
|
|
|
id=f"ws_{int(datetime.now().timestamp())}",
|
|
|
case_id=req.id,
|
|
|
timestamp=datetime.now(),
|
|
|
messages=[]
|
|
|
)
|
|
|
|
|
|
# 连接WebSocket
|
|
|
start_time = datetime.now()
|
|
|
|
|
|
try:
|
|
|
async with websockets.connect(
|
|
|
ws_url,
|
|
|
extra_headers={"X-API-Key": "test-api-key"}
|
|
|
) as ws:
|
|
|
result.latency = int((datetime.now() - start_time).total_seconds() * 1000)
|
|
|
result.success = True
|
|
|
|
|
|
# 发送订阅消息
|
|
|
msg = {
|
|
|
"action": test_case.action,
|
|
|
"symbols": symbols
|
|
|
}
|
|
|
await ws.send(json.dumps(msg))
|
|
|
|
|
|
# 等待响应（最多3条消息）
|
|
|
for _ in range(3):
|
|
|
try:
|
|
|
msg_data = await asyncio.wait_for(ws.recv(), timeout=5)
|
|
|
result.messages.append(WSMessage(
|
|
|
type="received",
|
|
|
data=json.loads(msg_data),
|
|
|
timestamp=datetime.now()
|
|
|
))
|
|
|
except asyncio.TimeoutError:
|
|
|
break
|
|
|
|
|
|
except Exception as e:
|
|
|
result.latency = int((datetime.now() - start_time).total_seconds() * 1000)
|
|
|
result.success = False
|
|
|
result.error = str(e)
|
|
|
|
|
|
self._add_ws_history(result)
|
|
|
return result
|
|
|
|
|
|
def get_test_history(self, req: TestHistoryRequest) -> TestHistoryData:
|
|
|
"""获取测试历史"""
|
|
|
with self.lock:
|
|
|
limit = req.limit or 20
|
|
|
|
|
|
api_tests = []
|
|
|
ws_tests = []
|
|
|
|
|
|
if not req.type or req.type == "api":
|
|
|
api_tests = self.api_history[-limit:]
|
|
|
|
|
|
if not req.type or req.type == "ws":
|
|
|
ws_tests = self.ws_history[-limit:]
|
|
|
|
|
|
return TestHistoryData(api_tests=api_tests, ws_tests=ws_tests)
|
|
|
|
|
|
def _add_api_history(self, result: APITestResult):
|
|
|
"""添加API测试历史"""
|
|
|
with self.lock:
|
|
|
self.api_history.append(result)
|
|
|
if len(self.api_history) > self.history_size:
|
|
|
self.api_history = self.api_history[-self.history_size:]
|
|
|
|
|
|
def _add_ws_history(self, result: WSTestResult):
|
|
|
"""添加WebSocket测试历史"""
|
|
|
with self.lock:
|
|
|
self.ws_history.append(result)
|
|
|
if len(self.ws_history) > self.history_size:
|
|
|
self.ws_history = self.ws_history[-self.history_size:]
|
