zhanghongyu_0626
70aa861983
- Added new modules for advisor compliance, KYC sessions, and script templates, enhancing the advisor agent's capabilities. - Implemented a comprehensive API structure under the `/api/advisor-agent` prefix, ensuring clear organization and access to new features. - Established database models and repositories for compliance rules and KYC sessions, facilitating robust data management. - Integrated exception handling and response models to improve error management and user feedback. - Updated settings to include new configurations for compliance and KYC features, ensuring flexibility and adaptability. This update significantly expands the advisor agent's functionality, providing essential tools for compliance and customer interaction while maintaining a structured API design.
32 lines
976 B
Python
32 lines
976 B
Python
"""顾问 Agent sprint 测试共用：merger 鉴权与 token 助手。"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from fastapi.testclient import TestClient
|
|
|
|
STAFF_ADVISOR = "STAFF-10086"
|
|
STAFF_COMPLIANCE = "STAFF-40001"
|
|
|
|
|
|
def login_staff_token(client: TestClient, *, actor_id: str = STAFF_ADVISOR) -> str:
|
|
response = client.post(
|
|
"/api/auth/login",
|
|
json={"actor_id": actor_id, "token_type": "staff"},
|
|
)
|
|
assert response.status_code == 200, response.text
|
|
return response.json()["data"]["access_token"]
|
|
|
|
|
|
def login_token(username: str, password: str) -> str:
|
|
"""兼容旧顾问测试签名：username 映射到 merger mock 账号。"""
|
|
_ = password
|
|
mapping = {
|
|
"advisor_test": STAFF_ADVISOR,
|
|
"compliance_test": STAFF_COMPLIANCE,
|
|
"admin_test": STAFF_COMPLIANCE,
|
|
}
|
|
actor_id = mapping.get(username, STAFF_ADVISOR)
|
|
from app.main import app
|
|
|
|
return login_staff_token(TestClient(app), actor_id=actor_id)
|