新增异常账户基线写库服务

lsfx-mock-server/tests/test_abnormal_account_baseline_service.py

@@ -0,0 +1,149 @@
+import pytest
+
+from services.abnormal_account_baseline_service import AbnormalAccountBaselineService
+
+
+class FakeCursor:
+    def __init__(self, connection):
+        self.connection = connection
+
+    def execute(self, sql, params=None):
+        self.connection.executed_sql.append(
+            {
+                "sql": sql,
+                "params": params,
+            }
+        )
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, exc_type, exc, tb):
+        return False
+
+
+class FakeConnection:
+    def __init__(self):
+        self.executed_sql = []
+        self.commit_count = 0
+        self.rollback_count = 0
+
+    def cursor(self):
+        return FakeCursor(self)
+
+    def commit(self):
+        self.commit_count += 1
+
+    def rollback(self):
+        self.rollback_count += 1
+
+    def close(self):
+        return None
+
+
+def test_apply_should_skip_when_abnormal_accounts_is_empty():
+    service = AbnormalAccountBaselineService()
+    fake_connection = FakeConnection()
+    service._connect = lambda: fake_connection
+
+    service.apply("330101199001010001", [])
+
+    assert fake_connection.executed_sql == []
+    assert fake_connection.commit_count == 0
+    assert fake_connection.rollback_count == 0
+
+
+def test_apply_should_raise_when_fact_owner_mismatches_staff():
+    service = AbnormalAccountBaselineService()
+
+    with pytest.raises(RuntimeError, match="owner_id_card"):
+        service.apply(
+            "330101199001010001",
+            [
+                {
+                    "account_no": "6222000000000001",
+                    "owner_id_card": "330101199001010099",
+                    "account_name": "测试员工工资卡",
+                    "status": 2,
+                    "effective_date": "2024-01-01",
+                    "invalid_date": "2026-03-20",
+                    "rule_code": "SUDDEN_ACCOUNT_CLOSURE",
+                }
+            ],
+        )
+
+
+def test_apply_should_insert_new_account_fact_by_account_no():
+    service = AbnormalAccountBaselineService()
+    fake_connection = FakeConnection()
+    service._connect = lambda: fake_connection
+
+    service.apply(
+        "330101199001010001",
+        [
+            {
+                "account_no": "6222000000000001",
+                "owner_id_card": "330101199001010001",
+                "account_name": "测试员工工资卡",
+                "status": 2,
+                "effective_date": "2024-01-01",
+                "invalid_date": "2026-03-20",
+                "rule_code": "SUDDEN_ACCOUNT_CLOSURE",
+            }
+        ],
+    )
+
+    assert len(fake_connection.executed_sql) == 1
+    executed = fake_connection.executed_sql[0]
+    assert "INSERT INTO ccdi_account_info" in executed["sql"]
+    assert executed["params"] == (
+        "6222000000000001",
+        "DEBIT",
+        "测试员工工资卡",
+        "EMPLOYEE",
+        "330101199001010001",
+        "兰溪农商银行",
+        "LXNCSY",
+        "CNY",
+        1,
+        "HIGH",
+        2,
+        "2024-01-01",
+        "2026-03-20",
+        "lsfx-mock-server",
+        "lsfx-mock-server",
+    )
+    assert fake_connection.commit_count == 1
+    assert fake_connection.rollback_count == 0
+
+
+def test_apply_should_update_existing_account_fact_by_account_no():
+    service = AbnormalAccountBaselineService()
+    fake_connection = FakeConnection()
+    service._connect = lambda: fake_connection
+
+    service.apply(
+        "330101199001010001",
+        [
+            {
+                "account_no": "6222000000000001",
+                "owner_id_card": "330101199001010001",
+                "account_name": "测试员工结算卡",
+                "status": 1,
+                "effective_date": "2025-01-01",
+                "invalid_date": None,
+                "rule_code": "DORMANT_ACCOUNT_LARGE_ACTIVATION",
+            }
+        ],
+    )
+
+    assert len(fake_connection.executed_sql) == 1
+    executed = fake_connection.executed_sql[0]
+    assert "ON DUPLICATE KEY UPDATE" in executed["sql"]
+    assert executed["params"][0] == "6222000000000001"
+    assert executed["params"][2] == "测试员工结算卡"
+    assert executed["params"][10] == 1
+    assert executed["params"][11] == "2025-01-01"
+    assert executed["params"][12] is None
+    assert fake_connection.commit_count == 1
+    assert fake_connection.rollback_count == 0