"""画像版本生成器单元测试（`docs/00` §6.4.6 的四条规则）。 覆盖四个方向，都要有牙： 1. **同事务两条事件**：一次生成必须同时投 `MILVUS` 与 `NEO4J` 两条，共用同一 `event_uuid`。 2. **幂等**：`snapshot_hash` 未变 → **不升版本、不投事件**（否则每次调用都灌 outbox）。 3. **旧版本置 0**：清旧当前标记必须发生在插新版本**之前**（唯一键约束）。 4. **失败关闭**：客户主表没有该行 → 抛错，不凭空造画像。 5. **字段对齐**：写入快照必须覆盖读取侧白名单所需字段（否则端点会返回空画像）。 """ from __future__ import annotations import json from datetime import UTC, datetime, timedelta from typing import Any import pytest from app.core.errors import ValidationAgentError from app.core.profile_projection import project_profile from app.service.profile_generation_service import ( REQUIRED_SNAPSHOT_FIELDS, SYNC_OPERATION_UPSERT, SYNC_STATUS_PENDING, SYNC_TARGETS, ProfileGenerationService, build_snapshot, compute_hash, ) NOW = datetime(2026, 9, 11, 12, 0, tzinfo=UTC) # --------------------------------------------------------------------------- # 替身：只实现 ProfileRepository 用到的方法 # --------------------------------------------------------------------------- class FakeRepository: """记录调用顺序与写入内容；`executed` 用来断言"先清旧、后插新"。""" def __init__( self, *, profile: dict[str, Any] | None = None, assessment: dict[str, Any] | None = None, current: dict[str, Any] | None = None, next_version: int = 2, memories: list[dict[str, Any]] | None = None, ) -> None: self._profile = profile self._assessment = assessment self._current = current self._next_version = next_version self._memories = memories or [] self.executed: list[str] = [] self.inserted_snapshot: dict[str, Any] | None = None self.sync_events: list[dict[str, Any]] = [] async def profile_row(self, _cid: int) -> dict[str, Any] | None: self.executed.append("profile_row") return self._profile async def latest_assessment(self, _cid: int) -> dict[str, Any] | None: self.executed.append("latest_assessment") return self._assessment async def current_snapshot(self, _cid: int) -> dict[str, Any] | None: self.executed.append("current_snapshot") return self._current async def active_memories(self, _cid: int) -> list[dict[str, Any]]: self.executed.append("active_memories") return self._memories async def next_version(self, _cid: int) -> int: self.executed.append("next_version") return self._next_version async def clear_current(self, _cid: int, *, now: datetime) -> None: self.executed.append("clear_current") async def insert_snapshot(self, **kwargs: Any) -> None: self.executed.append("insert_snapshot") self.inserted_snapshot = kwargs def add_sync_event(self, **kwargs: Any) -> None: self.executed.append("add_sync_event") self.sync_events.append(kwargs) def service(monkeypatch: pytest.MonkeyPatch, repo: FakeRepository) -> ProfileGenerationService: monkeypatch.setattr( "app.service.profile_generation_service.ProfileRepository", lambda _session: repo ) return ProfileGenerationService(session=object()) # type: ignore[arg-type] def profile_row(**overrides: Any) -> dict[str, Any]: row: dict[str, Any] = { "customer_id": 9102, "investor_type": "C3", "investment_horizon": "medium_term", "trading_frequency": "medium", "preferred_asset_class": ["bond_fund"], "risk_tags": ["balanced"], "behavior_score": 61, "total_asset": "860000.00", "last_active_at": NOW, } row.update(overrides) return row def assessment_row(valid_days: int = 200) -> dict[str, Any]: return { "id": 99102, "questionnaire_version": "v2026.1", "investor_type": "C3", "assessed_at": NOW - timedelta(days=165), "valid_until": (NOW + timedelta(days=valid_days)).replace(tzinfo=None), } # --------------------------------------------------------------------------- # 1. 快照构建 # --------------------------------------------------------------------------- def test_snapshot_covers_every_field_the_read_side_needs() -> None: """写侧必须覆盖读取侧白名单所需字段，否则端点会返回空画像。""" snapshot = build_snapshot(profile_row(), assessment_row(), now=NOW) missing = [f for f in REQUIRED_SNAPSHOT_FIELDS if f not in snapshot] assert not missing, f"快照缺少读取侧需要的字段：{missing}" def test_snapshot_is_projectable_by_the_read_path() -> None: """端到端对齐：生成出来的快照经 `project_profile` 能得到非空投影。""" snapshot = build_snapshot(profile_row(), assessment_row(), now=NOW) projected = project_profile(snapshot, now=NOW) assert projected["investor_type"] == "C3" assert projected["assessment_expired"] is False def test_expired_assessment_is_marked_expired_in_snapshot() -> None: """过期测评必须在快照里标成过期（与读取侧"按当前时间重算"同向）。""" snapshot = build_snapshot(profile_row(), assessment_row(valid_days=-45), now=NOW) assert snapshot["assessment_expired"] is True assert project_profile(snapshot, now=NOW)["assessment_expired"] is True def test_no_assessment_means_no_fabricated_validity() -> None: """没有测评行时**不写**有效期/过期标记，而不是伪造一个。""" snapshot = build_snapshot(profile_row(), None, now=NOW) assert "assessment_valid_until" not in snapshot assert "assessment_expired" not in snapshot def test_hash_is_stable_against_key_order() -> None: """哈希是幂等判据，不能被字典插入顺序影响。""" a = {"investor_type": "C3", "behavior_score": 61} b = {"behavior_score": 61, "investor_type": "C3"} assert compute_hash(a) == compute_hash(b) assert compute_hash({"x": 1}) != compute_hash({"x": 2}) # --------------------------------------------------------------------------- # 2. 首次生成：两条事件 + 顺序 # --------------------------------------------------------------------------- @pytest.mark.asyncio async def test_first_generation_writes_two_events_and_clears_old_current( monkeypatch: pytest.MonkeyPatch, ) -> None: repo = FakeRepository(profile=profile_row(), assessment=assessment_row(), current=None) result = await service(monkeypatch, repo).generate(9102, now=NOW) assert result.changed is True assert result.version == 2 assert result.sync_events == 2 # 两个目标存储各一条，且**共用同一个 event_uuid**（唯一键是 (event_uuid, target_store)） targets = [e["target_store"] for e in repo.sync_events] assert targets == list(SYNC_TARGETS) assert len({e["event_uuid"] for e in repo.sync_events}) == 1 # `aggregate_type='profile'` 由仓储层固定写入（不在 kwargs 里），此处断言服务传入的实体标识 assert len({e["aggregate_uuid"] for e in repo.sync_events}) == 1 assert all(e["aggregate_uuid"] == result.profile_uuid for e in repo.sync_events) # 断言取值本身，且**与消费端领取条件对齐**：outbox worker 只领 `pending`/`failed`， # 写成别的取值事件就永远没人消费。这里不再硬编码字面量（硬编码正是当初跑偏的原因）。 assert all(e["operation"] == SYNC_OPERATION_UPSERT for e in repo.sync_events) assert all(e["status"] == SYNC_STATUS_PENDING for e in repo.sync_events) assert SYNC_OPERATION_UPSERT == "upsert" assert SYNC_STATUS_PENDING == "pending" assert set(SYNC_TARGETS) == {"milvus", "neo4j"} # 顺序：清旧当前标记必须在插新版本之前（否则撞唯一键） assert repo.executed.index("clear_current") < repo.executed.index("insert_snapshot") # 新版本标为当前 assert repo.inserted_snapshot is not None assert repo.inserted_snapshot["version"] == 2 @pytest.mark.asyncio async def test_payload_carries_memory_sources_for_projection( monkeypatch: pytest.MonkeyPatch, ) -> None: """payload 必须带 `memory_sources` 与 `profile_version`。 这是 Milvus 长期记忆投影的输入契约：适配器要 `memory_sources` 才知道往向量库 写什么，要 `profile_version`（或 `version`）才认得出这一批属于哪个画像版本。 缺了它，事件能被领取、却什么也投影不出来——属于"静默空转"，必须由测试挡住。 """ memories = [{ "memory_uuid": "11111111-2222-3333-4444-555555555555", "memory_key": "preference:risk_level", "content": "稳健型", "memory_type": "preference", "confidence": 0.9, "version": 1, "valid_until": None, }] repo = FakeRepository( profile=profile_row(), assessment=assessment_row(), current=None, memories=memories ) result = await service(monkeypatch, repo).generate(9102, now=NOW) for sync_event in repo.sync_events: payload = sync_event["payload"] assert payload["profile_version"] == result.version assert payload["version"] == result.version sources = payload["memory_sources"] assert len(sources) == 1 assert sources[0]["memory_uuid"] == memories[0]["memory_uuid"] assert sources[0]["memory_key"] == "preference:risk_level" @pytest.mark.asyncio async def test_payload_memory_sources_is_empty_without_active_memories( monkeypatch: pytest.MonkeyPatch, ) -> None: """没有有效记忆时给**空列表**（而不是省略该键）。 省略键会让适配器的 `memory_sources is invalid` 报错、事件反复重试直至死信； 空列表是"确实没有要投影的记忆"，语义不同。这里把这个区别钉住。 """ repo = FakeRepository(profile=profile_row(), assessment=assessment_row(), current=None) await service(monkeypatch, repo).generate(9102, now=NOW) for sync_event in repo.sync_events: assert sync_event["payload"]["memory_sources"] == [] @pytest.mark.asyncio async def test_each_version_gets_a_fresh_profile_uuid(monkeypatch: pytest.MonkeyPatch) -> None: """`profile_uuid` 有唯一键，**每个版本必须用新 uuid**（实测撞过 Duplicate entry）。""" old = {"profile_uuid": "old-uuid", "version": 1, "snapshot_hash": "different"} repo = FakeRepository(profile=profile_row(), assessment=assessment_row(), current=old) result = await service(monkeypatch, repo).generate(9102, now=NOW) assert result.profile_uuid != "old-uuid" assert repo.inserted_snapshot is not None assert repo.inserted_snapshot["profile_uuid"] == result.profile_uuid # --------------------------------------------------------------------------- # 3. 幂等 # --------------------------------------------------------------------------- @pytest.mark.asyncio async def test_unchanged_content_does_not_bump_version_or_emit_events( monkeypatch: pytest.MonkeyPatch, ) -> None: snapshot = build_snapshot(profile_row(), assessment_row(), now=NOW) current = { "profile_uuid": "u1", "version": 7, "snapshot_hash": compute_hash(snapshot), } repo = FakeRepository(profile=profile_row(), assessment=assessment_row(), current=current) result = await service(monkeypatch, repo).generate(9102, now=NOW) assert result.changed is False assert result.version == 7 # 不升版本 assert result.sync_events == 0 # 不投事件 assert "insert_snapshot" not in repo.executed assert repo.sync_events == [] @pytest.mark.asyncio async def test_changed_content_bumps_version_and_reemits( monkeypatch: pytest.MonkeyPatch, ) -> None: current = {"profile_uuid": "u1", "version": 3, "snapshot_hash": "stale-hash"} repo = FakeRepository( profile=profile_row(total_asset="999999.00"), assessment=assessment_row(), current=current, next_version=4, ) result = await service(monkeypatch, repo).generate(9102, now=NOW) assert result.changed is True assert result.version == 4 assert result.sync_events == 2 # --------------------------------------------------------------------------- # 4. 失败关闭 # --------------------------------------------------------------------------- @pytest.mark.asyncio async def test_missing_customer_fails_closed(monkeypatch: pytest.MonkeyPatch) -> None: """主表没有该客户 → 抛错，不凭空造画像。""" repo = FakeRepository(profile=None) with pytest.raises(ValidationAgentError, match="客户不存在"): await service(monkeypatch, repo).generate(99999, now=NOW) assert repo.inserted_snapshot is None assert repo.sync_events == [] # --------------------------------------------------------------------------- # 5. 事件载荷 # --------------------------------------------------------------------------- @pytest.mark.asyncio async def test_event_payload_carries_version_and_hash(monkeypatch: pytest.MonkeyPatch) -> None: repo = FakeRepository(profile=profile_row(), assessment=assessment_row(), current=None) result = await service(monkeypatch, repo).generate(9102, now=NOW) payload = repo.sync_events[0]["payload"] assert payload["version"] == result.version assert payload["snapshot_hash"] == result.snapshot_hash assert payload["customer_id"] == "9102" assert payload["snapshot"]["investor_type"] == "C3" # 载荷必须可 JSON 序列化（outbox 是 JSON 列） json.dumps(payload, ensure_ascii=False)