83 lines
3.0 KiB
Python
83 lines
3.0 KiB
Python
from fastapi.testclient import TestClient
|
|
|
|
from app.db import auth_db
|
|
from app.db.review_center import get_review_center_dashboard
|
|
from app.web import web_server
|
|
|
|
|
|
def _login_user(email: str, admin: bool = False) -> str:
|
|
reg = auth_db.register_user(email, "StrongPass123")
|
|
auth_db.verify_email(email, reg["verification_code"])
|
|
user = auth_db.get_user_by_email(email)
|
|
auth_db.claim_free_trial(user["id"])
|
|
if admin:
|
|
auth_db.set_user_admin(email, True)
|
|
return auth_db.login_user(email, "StrongPass123")["token"]
|
|
|
|
|
|
def test_review_center_page_and_api_require_admin():
|
|
token = _login_user("normal-review-center@example.com")
|
|
client = TestClient(web_server.app)
|
|
client.cookies.set("altcoin_session", token)
|
|
|
|
page = client.get("/review-center")
|
|
api = client.get("/api/review-center/dashboard")
|
|
|
|
assert page.status_code == 403
|
|
assert api.status_code == 403
|
|
|
|
|
|
def test_review_center_admin_can_access_page_and_api():
|
|
token = _login_user("admin-review-center@example.com", admin=True)
|
|
client = TestClient(web_server.app)
|
|
client.cookies.set("altcoin_session", token)
|
|
|
|
page = client.get("/review-center")
|
|
api = client.get("/api/review-center/dashboard")
|
|
|
|
assert page.status_code == 200
|
|
assert "复盘中心" in page.text
|
|
assert api.status_code == 200
|
|
data = api.json()
|
|
assert "opportunity" in data
|
|
assert "paper_trading" in data
|
|
assert "evidence" in data
|
|
assert "iteration" in data
|
|
|
|
|
|
def test_review_center_separates_opportunity_and_paper_pnl(pg_conn):
|
|
pg_conn.execute(
|
|
"""
|
|
INSERT INTO recommendation (
|
|
symbol, rec_time, rec_state, rec_score, entry_price,
|
|
status, execution_status, action_status, display_bucket, entry_triggered
|
|
) VALUES
|
|
('OBS/USDT', '2026-05-16T10:00:00', '观察', 10, 1, 'active', 'observe', '观察', 'watch_pool', 0),
|
|
('BUY/USDT', '2026-05-16T11:00:00', '爆发', 30, 10, 'active', 'buy_now', '可即刻买入', 'realtime', 1)
|
|
"""
|
|
)
|
|
pg_conn.execute(
|
|
"""
|
|
INSERT INTO paper_trades (
|
|
recommendation_id, symbol, side, status, opened_at, closed_at,
|
|
entry_price, exit_price, qty, notional_usdt, margin_usdt, leverage,
|
|
current_price, pnl_pct, realized_pnl_pct, realized_pnl_usdt,
|
|
source_status, source_action, created_at, updated_at
|
|
) VALUES (
|
|
2, 'BUY/USDT', 'long', 'closed', '2026-05-16T11:01:00', '2026-05-16T12:00:00',
|
|
10, 11, 500, 5000, 1000, 5,
|
|
11, 10, 10, 490,
|
|
'buy_now', '可即刻买入', '2026-05-16T11:01:00', '2026-05-16T12:00:00'
|
|
)
|
|
"""
|
|
)
|
|
pg_conn.commit()
|
|
|
|
data = get_review_center_dashboard(days=30)
|
|
|
|
assert data["opportunity"]["summary"]["total_opportunities"] == 2
|
|
assert data["opportunity"]["summary"]["paper_executed_count"] == 1
|
|
assert "total_pnl_usdt" not in data["opportunity"]["summary"]
|
|
assert data["paper_trading"]["summary"]["closed_count"] == 1
|
|
assert data["paper_trading"]["summary"]["realized_pnl_usdt"] == 490
|