Files
group_fqcd_jr/tests/unit/infrastructure/test_neo4j_profile_projection.py
T
张胜宇 e239eb778b docs: 品牌全量口径统一为「南方基金」+ 作废文档清理
1) 客服 Agent 四份交付文档 + 构建脚手架:品牌由包装占位 XX科技 / 旧名 南方财富
   统一为南方基金(热线 400-889-8899 / 官网 nffund.com),系统名改为「智能服务系统」;
   同步追加 §0.4 修订记录行,工程记录行保留原占位字面以支撑硬编码扫描验收。
2) 开发文档:清理 28 份已作废/残留文档(14 份移出归档 + 14 份仓库副本),
   新增《文档规整方案与开发前待决事项-2026-09-17》。
3) 客服agent 四份交付文档首次纳入本分支。
2026-09-17 15:15:22 +08:00

93 lines
2.9 KiB
Python

from types import SimpleNamespace
import pytest
from app.infrastructure.neo4j_profile_projection import Neo4jProfileProjection
class FakeDriver:
def __init__(self, *, applied: bool = True) -> None:
self.applied = applied
self.calls: list[tuple[str, dict[str, object]]] = []
async def execute_query(self, query: str, **parameters: object) -> SimpleNamespace:
self.calls.append((query, parameters))
return SimpleNamespace(records=[{"applied": True}] if self.applied else [])
def payload() -> dict[str, object]:
return {
"customer_id": 7,
"profile_uuid": "profile-7-v1",
"profile_version": 1,
"memory_sources": [
{
"memory_uuid": "memory-1",
"memory_key": "preference:risk_level",
"content": "稳健型",
"memory_type": "preference",
"confidence": 0.9,
"version": 2,
"valid_until": None,
},
{
"memory_uuid": "memory-2",
"memory_key": "goal:liquidity",
"content": "保持流动性",
"memory_type": "goal",
"confidence": 0.8,
"version": 1,
"valid_until": None,
},
],
}
@pytest.mark.asyncio
async def test_projects_only_fixed_preference_and_goal_queries() -> None:
driver = FakeDriver()
result = await Neo4jProfileProjection(driver).upsert(payload())
assert result.applied is True
assert len(driver.calls) == 3
assert "MERGE (c:Customer" in driver.calls[0][0]
assert "PREFERS" in driver.calls[1][0]
assert "HAS_GOAL" in driver.calls[2][0]
assert driver.calls[1][1]["items"][0]["memory_uuid"] == "memory-1"
@pytest.mark.asyncio
async def test_lower_profile_version_is_skipped_without_writes() -> None:
driver = FakeDriver(applied=False)
result = await Neo4jProfileProjection(driver).upsert(payload())
assert result.applied is False
assert result.reason == "newer_profile_version_exists"
assert len(driver.calls) == 1
@pytest.mark.asyncio
async def test_sensitive_memory_content_is_redacted_before_projection() -> None:
data = payload()
source = data["memory_sources"][0]
assert isinstance(source, dict)
source["content"] = "我的密码是123456,手机号13800138000"
driver = FakeDriver()
await Neo4jProfileProjection(driver).upsert(data)
projected = driver.calls[1][1]["items"][0]["content"]
assert "123456" not in projected
assert "13800138000" not in projected
@pytest.mark.asyncio
async def test_unknown_memory_key_is_rejected() -> None:
data = payload()
source = data["memory_sources"][0]
assert isinstance(source, dict)
source["memory_key"] = "account:balance"
with pytest.raises(ValueError, match="not projectable"):
await Neo4jProfileProjection(FakeDriver()).upsert(data)