cirry
2f9c8bee2b
- adj_factor 覆盖索引 (ts_code,trade_date) INCLUDE (adj_factor)：根治堆碎片化 （单股 6516 行散 6516 块，满载时位图堆扫 1.5s+），Index Only Scan ~2ms； 因子查询全部改 2 列投影，lag() 窗口只取变点（6516→32 行） - preview 冷路径 2 会话 2 波并发，信息卡合并为 LEFT JOIN LATERAL 一条 - 鉴权会话 60s 进程内缓存（登出/全端登出即时失效），全站请求省 ~80ms - pvj/chipsj/stocksj/facetsj 存 model_dump_json 原串直返（与 response_model 字节一致），热路径 230-2190ms → 1-2ms；get_version 本地缓存+bump 即时可见 - 连接池 10+20；smoke_test 适配鉴权缓存
121 lines
4.7 KiB
Python
121 lines
4.7 KiB
Python
"""使用线上 PostgreSQL 跑鉴权与核心 API 自检，临时用户会自动清理。
|
||
|
||
用法：uv run --directory backend python smoke_test.py
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import asyncio
|
||
import json
|
||
import secrets
|
||
from datetime import datetime, timedelta, timezone
|
||
|
||
import httpx
|
||
from sqlalchemy import delete, select
|
||
|
||
from app.auth import drop_session_cache, hash_password, token_digest
|
||
from app.db import async_session, engine
|
||
from app.main import app
|
||
from app.models import AuthSession, Candle, User
|
||
|
||
|
||
async def main() -> None:
|
||
username = f"smoke_{secrets.token_hex(6)}"
|
||
password = secrets.token_urlsafe(24)
|
||
now = datetime.now(timezone.utc)
|
||
|
||
async with async_session() as db:
|
||
db.add(
|
||
User(
|
||
username=username,
|
||
password_hash=hash_password(password),
|
||
password_algo="argon2id",
|
||
is_active=True,
|
||
failed_login_count=0,
|
||
password_changed_at=now,
|
||
created_at=now,
|
||
updated_at=now,
|
||
)
|
||
)
|
||
await db.commit()
|
||
|
||
try:
|
||
transport = httpx.ASGITransport(app=app)
|
||
async with httpx.AsyncClient(transport=transport, base_url="http://testserver") as client:
|
||
health = await client.get("/api/health")
|
||
protected = await client.post("/api/backtest", json={})
|
||
bad_login = await client.post(
|
||
"/api/auth/login", json={"username": username, "password": "wrong-password"}
|
||
)
|
||
login = await client.post(
|
||
"/api/auth/login", json={"username": username, "password": password}
|
||
)
|
||
print("health:", health.status_code)
|
||
print("protected without login:", protected.status_code)
|
||
print("bad login:", bad_login.status_code)
|
||
print("login:", login.status_code)
|
||
assert health.status_code == 200
|
||
assert protected.status_code == 401
|
||
assert bad_login.status_code == 401
|
||
assert login.status_code == 200
|
||
assert client.cookies.get("stock_session")
|
||
|
||
me = await client.get("/api/auth/me")
|
||
assert me.status_code == 200 and me.json()["username"] == username
|
||
|
||
daily = await client.post(
|
||
"/api/backtest",
|
||
json={
|
||
"symbol": "000001",
|
||
"strategy": "macd_cross",
|
||
"params": {"fast": 12, "slow": 26, "signal": 9},
|
||
"initial_cash": 100000.0,
|
||
"fast_mode": False,
|
||
},
|
||
)
|
||
assert daily.status_code == 200, daily.text
|
||
result = daily.json()
|
||
assert len(result["candles"]) > 100
|
||
print("backtest metrics:", json.dumps(result["metrics"], ensure_ascii=False))
|
||
|
||
token = client.cookies.get("stock_session")
|
||
assert token
|
||
async with async_session() as db:
|
||
auth_session = (
|
||
await db.execute(
|
||
select(AuthSession).where(AuthSession.token_hash == token_digest(token))
|
||
)
|
||
).scalar_one()
|
||
auth_session.expires_at = datetime.now(timezone.utc) - timedelta(seconds=1)
|
||
await db.commit()
|
||
# 鉴权会话有 60s 进程内缓存（auth.py），会掩盖库里的过期态；
|
||
# 清缓存模拟 TTL 已过，验证 require_user 对过期会话本身的判定
|
||
drop_session_cache(digest=token_digest(token))
|
||
assert (await client.get("/api/auth/me")).status_code == 401
|
||
print("expired session: 401")
|
||
|
||
login = await client.post(
|
||
"/api/auth/login", json={"username": username, "password": password}
|
||
)
|
||
assert login.status_code == 200
|
||
logout = await client.post("/api/auth/logout")
|
||
assert logout.status_code == 204
|
||
assert (await client.get("/api/auth/me")).status_code == 401
|
||
print("logout and revoke: ok")
|
||
|
||
async with async_session() as db:
|
||
has_market_data = bool(await db.scalar(select(Candle.id).limit(1)))
|
||
print("market data present:", has_market_data)
|
||
finally:
|
||
async with async_session() as db:
|
||
user_id = await db.scalar(select(User.id).where(User.username == username))
|
||
if user_id is not None:
|
||
await db.execute(delete(AuthSession).where(AuthSession.user_id == user_id))
|
||
await db.execute(delete(User).where(User.id == user_id))
|
||
await db.commit()
|
||
await engine.dispose()
|
||
print("temporary user cleaned")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(main())
|