first commit

backend/tests/conftest.py

@@ -0,0 +1,19 @@
+import pytest
+import asyncio
+from httpx import AsyncClient, ASGITransport
+
+from app.main import app
+
+
+@pytest.fixture(scope="session")
+def event_loop():
+    loop = asyncio.new_event_loop()
+    yield loop
+    loop.close()
+
+
+@pytest.fixture
+async def client():
+    transport = ASGITransport(app=app)
+    async with AsyncClient(transport=transport, base_url="http://test") as c:
+        yield c

backend/tests/test_api.py

@@ -0,0 +1,19 @@
+"""API integration tests."""
+import pytest
+from httpx import AsyncClient, ASGITransport
+
+from app.main import app
+
+
+@pytest.fixture
+async def client():
+    transport = ASGITransport(app=app)
+    async with AsyncClient(transport=transport, base_url="http://test") as c:
+        yield c
+
+
+@pytest.mark.asyncio
+async def test_health(client: AsyncClient):
+    resp = await client.get("/health")
+    assert resp.status_code == 200
+    assert resp.json()["status"] == "ok"

backend/tests/test_rules.py

@@ -0,0 +1,87 @@
+"""Unit tests for the rules engine using plain objects (no SQLAlchemy session)."""
+from datetime import datetime, timezone
+from types import SimpleNamespace
+from uuid import uuid4
+
+import pytest
+
+from app.models.transaction import Direction
+from app.models.evidence_image import SourceApp
+from app.rules.dedup_rules import is_duplicate_pair
+from app.rules.transit_rules import is_self_transfer
+from app.rules.assessment_rules import classify_transaction
+
+
+def _make_tx(**kwargs):
+    defaults = dict(
+        id=uuid4(), case_id=uuid4(), source_app=SourceApp.alipay,
+        trade_time=datetime(2026, 3, 8, 10, 0, tzinfo=timezone.utc),
+        amount=10000, direction=Direction.out,
+        counterparty_name="测试对手方", counterparty_account="",
+        self_account_tail_no="1234", order_no="ORD001",
+        remark="测试", confidence=0.9, is_duplicate=False, is_transit=False,
+        evidence_image_id=None, cluster_id=None,
+    )
+    defaults.update(kwargs)
+    return SimpleNamespace(**defaults)
+
+
+class TestDedupRules:
+    def test_same_order_no(self):
+        a = _make_tx(order_no="ORD001")
+        b = _make_tx(order_no="ORD001", self_account_tail_no="5678")
+        assert is_duplicate_pair(a, b)
+
+    def test_different_order_no_different_counterparty(self):
+        a = _make_tx(order_no="ORD001", counterparty_name="A", self_account_tail_no="1111")
+        b = _make_tx(order_no="ORD002", counterparty_name="B", self_account_tail_no="2222")
+        assert not is_duplicate_pair(a, b)
+
+    def test_same_amount_close_time_same_tail(self):
+        a = _make_tx(order_no="", amount=5000)
+        b = _make_tx(
+            order_no="",
+            amount=5000,
+            trade_time=datetime(2026, 3, 8, 10, 3, tzinfo=timezone.utc),
+        )
+        assert is_duplicate_pair(a, b)
+
+    def test_same_amount_far_time(self):
+        a = _make_tx(order_no="", amount=5000)
+        b = _make_tx(
+            order_no="",
+            amount=5000,
+            trade_time=datetime(2026, 3, 8, 11, 0, tzinfo=timezone.utc),
+        )
+        assert not is_duplicate_pair(a, b)
+
+
+class TestTransitRules:
+    def test_keyword_match(self):
+        tx = _make_tx(counterparty_name="支付宝-张某", direction=Direction.out)
+        assert is_self_transfer(tx, [])
+
+    def test_known_account_match(self):
+        tx = _make_tx(counterparty_name="我的银行卡")
+        assert is_self_transfer(tx, ["我的银行卡"])
+
+    def test_not_transit(self):
+        tx = _make_tx(counterparty_name="李*华", remark="投资款")
+        assert not is_self_transfer(tx, [])
+
+
+class TestAssessmentRules:
+    def test_transit_classified_as_low(self):
+        tx = _make_tx(is_transit=True)
+        level, reason, _ = classify_transaction(tx)
+        assert level.value == "low"
+
+    def test_high_confidence_fraud_keyword(self):
+        tx = _make_tx(confidence=0.95, remark="投资款")
+        level, reason, _ = classify_transaction(tx)
+        assert level.value == "high"
+
+    def test_income_classified_as_low(self):
+        tx = _make_tx(direction=Direction.in_)
+        level, _, _ = classify_transaction(tx)
+        assert level.value == "low"