修复Mock流水按数据库员工及亲属绑定身份证

lsfx-mock-server/tests/test_statement_service.py

@@ -12,6 +12,15 @@ from services.statement_rule_samples import (
 )
 
 
+class FakeStaffIdentityRepository:
+    def select_random_staff_with_families(self):
+        return {
+            "staff_name": "数据库员工",
+            "staff_id_card": "320101199001010030",
+            "family_id_cards": ["320101199201010051", "320101199301010052"],
+        }
+
+
 def test_generate_statements_should_include_seeded_samples_before_noise():
     """生成流水时必须先混入固定命中样本，而不是纯随机噪声。"""
     service = StatementService()
@@ -89,6 +98,30 @@ def test_generate_statements_should_fill_noise_up_to_requested_count():
     assert len(statements) == 80
 
 
+def test_generate_statements_should_stay_within_single_employee_scope_per_log_id():
+    """同一 logId 的流水只能落在 FileRecord 绑定的员工及亲属身份证内。"""
+    file_service = FileService(staff_identity_repository=FakeStaffIdentityRepository())
+    service = StatementService(file_service=file_service)
+    response = file_service.fetch_inner_flow(
+        {
+            "groupId": 1001,
+            "customerNo": "customer_scope",
+            "dataChannelCode": "test",
+            "requestDateId": 20240101,
+            "dataStartDateId": 20240101,
+            "dataEndDateId": 20240131,
+            "uploadUserId": 902001,
+        }
+    )
+    log_id = response["data"][0]
+    record = file_service.file_records[log_id]
+    allowed_id_cards = {record.staff_id_card, *record.family_id_cards}
+
+    statements = service._generate_statements(group_id=1000, log_id=log_id, count=1600)
+
+    assert {item["cretNo"] for item in statements}.issubset(allowed_id_cards)
+
+
 def test_generate_statements_should_only_use_recognizable_identity_cards():
     """命中样本和随机噪声都只能使用现库可识别的身份证号。"""
     service = StatementService()
@@ -121,7 +154,7 @@ def test_get_bank_statement_should_keep_same_cached_result_for_same_log_id():
 
 def test_get_bank_statement_uses_primary_binding_from_file_service(monkeypatch):
     """同一 logId 的流水记录必须复用 FileService 中的主体与账号绑定。"""
-    file_service = FileService()
+    file_service = FileService(staff_identity_repository=FakeStaffIdentityRepository())
     statement_service = StatementService(file_service=file_service)
 
     monkeypatch.setattr(
@@ -161,7 +194,7 @@ def test_get_bank_statement_uses_primary_binding_from_file_service(monkeypatch):
 
 def test_get_bank_statement_contains_large_transaction_hit_samples(monkeypatch):
     """流水 Mock 首次生成时必须稳定包含可命中大额交易规则的样本簇。"""
-    file_service = FileService()
+    file_service = FileService(staff_identity_repository=FakeStaffIdentityRepository())
     statement_service = StatementService(file_service=file_service)
 
     monkeypatch.setattr(
@@ -182,6 +215,9 @@ def test_get_bank_statement_contains_large_transaction_hit_samples(monkeypatch):
         }
     )
     log_id = response["data"][0]
+    record = file_service.file_records[log_id]
+    staff_id_card = record.staff_id_card
+    family_id_card = record.family_id_cards[0]
 
     statement_response = statement_service.get_bank_statement(
         {
@@ -195,12 +231,7 @@ def test_get_bank_statement_contains_large_transaction_hit_samples(monkeypatch):
 
     assert statements
     assert any(
-        item["cretNo"] in {
-            "330101198801010011",
-            "330101199001010022",
-            "330101198802020033",
-            "330101199202020044",
-        }
+        item["cretNo"] in {staff_id_card, family_id_card}
         for item in statements
     )
     assert any("房产首付款" in item["userMemo"] for item in statements)
@@ -212,7 +243,7 @@ def test_get_bank_statement_contains_large_transaction_hit_samples(monkeypatch):
 
     for item in statements:
         if (
-            item["cretNo"] == "330101198802020033"
+            item["cretNo"] == staff_id_card
             and item["customerName"] == "浙江远望贸易有限公司"
             and item["crAmount"] > 0
         ):