extract contact reconciliation service

tests/test_contact_reconciliation_service.py

@@ -0,0 +1,72 @@
+"""Tests for shared contact/message reconciliation helpers."""
+
+import pytest
+
+from app.repository import ContactNameHistoryRepository, ContactRepository, MessageRepository
+from app.services.contact_reconciliation import (
+    claim_prefix_messages_for_contact,
+    record_contact_name_and_reconcile,
+)
+
+
+@pytest.mark.asyncio
+async def test_claim_prefix_messages_for_contact_promotes_prefix_dm(test_db):
+    public_key = "aa" * 32
+    await ContactRepository.upsert({"public_key": public_key, "name": "Alice", "type": 1})
+
+    await MessageRepository.create(
+        msg_type="PRIV",
+        text="hello",
+        conversation_key=public_key[:12],
+        sender_timestamp=1000,
+        received_at=1000,
+    )
+
+    claimed = await claim_prefix_messages_for_contact(public_key=public_key)
+
+    assert claimed == 1
+    messages = await MessageRepository.get_all(conversation_key=public_key)
+    assert len(messages) == 1
+    assert messages[0].conversation_key == public_key
+
+
+@pytest.mark.asyncio
+async def test_record_contact_name_and_reconcile_records_history_and_backfills(test_db):
+    public_key = "bb" * 32
+    channel_key = "CC" * 16
+    await ContactRepository.upsert({"public_key": public_key, "name": "Alice", "type": 1})
+
+    await MessageRepository.create(
+        msg_type="PRIV",
+        text="dm",
+        conversation_key=public_key[:12],
+        sender_timestamp=1000,
+        received_at=1000,
+    )
+    await MessageRepository.create(
+        msg_type="CHAN",
+        text="Alice: hello",
+        conversation_key=channel_key,
+        sender_timestamp=1001,
+        received_at=1001,
+        sender_name="Alice",
+    )
+
+    claimed, backfilled = await record_contact_name_and_reconcile(
+        public_key=public_key,
+        contact_name="Alice",
+        timestamp=1234,
+    )
+
+    assert claimed == 1
+    assert backfilled == 1
+
+    history = await ContactNameHistoryRepository.get_history(public_key)
+    assert len(history) == 1
+    assert history[0].name == "Alice"
+    assert history[0].first_seen == 1234
+    assert history[0].last_seen == 1234
+
+    messages = await MessageRepository.get_all(msg_type="CHAN", conversation_key=channel_key)
+    assert len(messages) == 1
+    assert messages[0].sender_key == public_key