袁聪的第一次提交，包含nl2sql,行情数据,场外申购

tests/unit/service/test_financial_nl2sql_service.py

@@ -0,0 +1,72 @@
+import pytest
+
+from app.core.contracts import RequestContext
+from app.core.nl2sql_contracts import FinancialNL2SQLInput
+from app.service.financial_nl2sql_service import FinancialNL2SQLService
+
+
+def context(**updates):
+    base = RequestContext(
+        user_id="1",
+        trace_id="trace-nl2sql",
+        roles=("advisor",),
+        permissions=("financial:nl2sql:read",),
+        data_scope="all",
+    )
+    return base.model_copy(update=updates)
+
+
+@pytest.mark.asyncio
+async def test_generates_read_only_market_sql_with_product_filter() -> None:
+    result = await FinancialNL2SQLService().query(
+        FinancialNL2SQLInput(question="查询159511近30天行情收盘价", dry_run=True),
+        context(),
+    )
+    assert result["status"] == "ready"
+    assert result["sql"].startswith("SELECT ")
+    assert "fin_market_price" in result["sql"]
+    assert "DROP" not in result["sql"]
+    assert result["parameters"]["filter_0"] == "159511"
+    assert result["audit"]["permission_check"]["status"] == "passed"
+
+
+@pytest.mark.asyncio
+async def test_low_confidence_question_requires_confirmation() -> None:
+    result = await FinancialNL2SQLService().query(
+        FinancialNL2SQLInput(question="帮我看看这个情况", dry_run=True),
+        context(),
+    )
+    assert result["status"] == "need_confirmation"
+    assert result["query_plan"]["confidence"] < 0.85
+
+
+@pytest.mark.asyncio
+async def test_as_of_query_rejects_current_snapshot_tables() -> None:
+    result = await FinancialNL2SQLService().query(
+        FinancialNL2SQLInput(question="截至某日客户当前持仓市值", dry_run=True),
+        context(),
+    )
+    assert result["status"] == "rejected"
+    assert "历史版本" in result["message"]
+
+
+@pytest.mark.asyncio
+async def test_customer_scope_is_injected_for_non_all_scope() -> None:
+    result = await FinancialNL2SQLService().query(
+        FinancialNL2SQLInput(question="查询客户账户余额", dry_run=True),
+        context(data_scope="own_customers", customer_ids=("7", "8")),
+    )
+    assert "customer_id IN" in result["sql"]
+    assert result["parameters"]["scope_customer_0"] == 7
+    assert result["parameters"]["scope_customer_1"] == 8
+
+
+@pytest.mark.asyncio
+async def test_cash_ledger_is_in_nl2sql_scope() -> None:
+    result = await FinancialNL2SQLService().query(
+        FinancialNL2SQLInput(question="查询近7天资金流水变化", dry_run=True),
+        context(),
+    )
+    assert result["status"] == "ready"
+    assert "fin_cash_ledger" in result["sql"]
+    assert result["query_plan"]["intent"] == "cash_ledger_query"