feat: add animal cohort creation projection and service

Implements Step 3.3: Animal Cohort Creation - Add AnimalRegistryProjection for animal_registry and live_animals_by_location - Add EventAnimalsProjection for event_animals link table - Add IntervalProjection for location and attribute intervals - Add AnimalService with create_cohort() for coordinating event + projections - Add seeded_db fixture to conftest.py - Update projections/__init__.py with new exports All operations atomic within single transaction. Includes validation for location (exists, active) and species (exists, active). 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
2025-12-29 06:52:23 +00:00
parent bd09c99366
commit 876e8174ee
11 changed files with 1637 additions and 1 deletions
--- a/tests/test_projection_event_animals.py
+++ b/tests/test_projection_event_animals.py
@@ -0,0 +1,178 @@
+# ABOUTME: Tests for EventAnimalsProjection.
+# ABOUTME: Validates event_animals link table updates on animal events.
+
+from animaltrack.events.types import ANIMAL_COHORT_CREATED
+from animaltrack.models.events import Event
+from animaltrack.projections.event_animals import EventAnimalsProjection
+
+
+def make_cohort_event(
+    event_id: str,
+    animal_ids: list[str],
+    location_id: str = "01ARZ3NDEKTSV4RRFFQ69G5FAV",
+    ts_utc: int = 1704067200000,
+) -> Event:
+    """Create a test AnimalCohortCreated event."""
+    return Event(
+        id=event_id,
+        type=ANIMAL_COHORT_CREATED,
+        ts_utc=ts_utc,
+        actor="test_user",
+        entity_refs={
+            "location_id": location_id,
+            "animal_ids": animal_ids,
+        },
+        payload={
+            "species": "duck",
+            "count": len(animal_ids),
+            "life_stage": "adult",
+            "sex": "unknown",
+            "location_id": location_id,
+            "origin": "purchased",
+            "notes": None,
+        },
+        version=1,
+    )
+
+
+class TestEventAnimalsProjectionEventTypes:
+    """Tests for get_event_types method."""
+
+    def test_handles_animal_cohort_created(self, seeded_db):
+        """Projection handles AnimalCohortCreated event type."""
+        projection = EventAnimalsProjection(seeded_db)
+        assert ANIMAL_COHORT_CREATED in projection.get_event_types()
+
+
+class TestEventAnimalsProjectionApply:
+    """Tests for apply()."""
+
+    def test_creates_event_animal_link_for_each_animal(self, seeded_db):
+        """Apply creates one row in event_animals per animal_id."""
+        row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
+        location_id = row[0]
+
+        animal_ids = [
+            "01ARZ3NDEKTSV4RRFFQ69G5A01",
+            "01ARZ3NDEKTSV4RRFFQ69G5A02",
+            "01ARZ3NDEKTSV4RRFFQ69G5A03",
+        ]
+        event_id = "01ARZ3NDEKTSV4RRFFQ69G5001"
+
+        projection = EventAnimalsProjection(seeded_db)
+        event = make_cohort_event(event_id, animal_ids, location_id=location_id)
+        projection.apply(event)
+
+        # Check event_animals has 3 rows
+        count = seeded_db.execute("SELECT COUNT(*) FROM event_animals").fetchone()[0]
+        assert count == 3
+
+        # Check each animal_id is linked
+        for animal_id in animal_ids:
+            row = seeded_db.execute(
+                "SELECT event_id FROM event_animals WHERE animal_id = ?",
+                (animal_id,),
+            ).fetchone()
+            assert row is not None
+            assert row[0] == event_id
+
+    def test_event_animal_link_has_correct_event_id(self, seeded_db):
+        """Event animal link has correct event_id."""
+        row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
+        location_id = row[0]
+
+        animal_ids = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
+        event_id = "01ARZ3NDEKTSV4RRFFQ69G5001"
+
+        projection = EventAnimalsProjection(seeded_db)
+        event = make_cohort_event(event_id, animal_ids, location_id=location_id)
+        projection.apply(event)
+
+        row = seeded_db.execute(
+            "SELECT event_id FROM event_animals WHERE animal_id = ?",
+            (animal_ids[0],),
+        ).fetchone()
+        assert row[0] == event_id
+
+    def test_event_animal_link_has_correct_ts_utc(self, seeded_db):
+        """Event animal link has correct ts_utc."""
+        row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
+        location_id = row[0]
+
+        animal_ids = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
+        event_id = "01ARZ3NDEKTSV4RRFFQ69G5001"
+        ts_utc = 1704067200000
+
+        projection = EventAnimalsProjection(seeded_db)
+        event = make_cohort_event(event_id, animal_ids, location_id=location_id, ts_utc=ts_utc)
+        projection.apply(event)
+
+        row = seeded_db.execute(
+            "SELECT ts_utc FROM event_animals WHERE animal_id = ?",
+            (animal_ids[0],),
+        ).fetchone()
+        assert row[0] == ts_utc
+
+
+class TestEventAnimalsProjectionRevert:
+    """Tests for revert()."""
+
+    def test_removes_event_animal_links(self, seeded_db):
+        """Revert deletes rows from event_animals."""
+        row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
+        location_id = row[0]
+
+        animal_ids = [
+            "01ARZ3NDEKTSV4RRFFQ69G5A01",
+            "01ARZ3NDEKTSV4RRFFQ69G5A02",
+        ]
+        event_id = "01ARZ3NDEKTSV4RRFFQ69G5001"
+
+        projection = EventAnimalsProjection(seeded_db)
+        event = make_cohort_event(event_id, animal_ids, location_id=location_id)
+        projection.apply(event)
+
+        # Verify rows exist
+        count = seeded_db.execute("SELECT COUNT(*) FROM event_animals").fetchone()[0]
+        assert count == 2
+
+        # Revert
+        projection.revert(event)
+
+        # Verify rows removed
+        count = seeded_db.execute("SELECT COUNT(*) FROM event_animals").fetchone()[0]
+        assert count == 0
+
+    def test_revert_only_affects_specific_event(self, seeded_db):
+        """Revert only removes links for the specific event."""
+        row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
+        location_id = row[0]
+
+        # Create first event
+        animal_ids_1 = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
+        event_id_1 = "01ARZ3NDEKTSV4RRFFQ69G5001"
+        projection = EventAnimalsProjection(seeded_db)
+        event1 = make_cohort_event(event_id_1, animal_ids_1, location_id=location_id)
+        projection.apply(event1)
+
+        # Create second event
+        animal_ids_2 = ["01ARZ3NDEKTSV4RRFFQ69G5A02"]
+        event_id_2 = "01ARZ3NDEKTSV4RRFFQ69G5002"
+        event2 = make_cohort_event(
+            event_id_2, animal_ids_2, location_id=location_id, ts_utc=1704067300000
+        )
+        projection.apply(event2)
+
+        # Verify both exist
+        count = seeded_db.execute("SELECT COUNT(*) FROM event_animals").fetchone()[0]
+        assert count == 2
+
+        # Revert only event1
+        projection.revert(event1)
+
+        # Event2's link should still exist
+        count = seeded_db.execute("SELECT COUNT(*) FROM event_animals").fetchone()[0]
+        assert count == 1
+
+        row = seeded_db.execute("SELECT event_id FROM event_animals").fetchone()
+        assert row[0] == event_id_2