Alo/animaltrack
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: add animal cohort creation projection and service

Browse Source 
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>
This commit is contained in:
Petru
2025-12-29 06:52:23 +00:00
parent bd09c99366
commit 876e8174ee
11 changed files with 1637 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

405
tests/test_projection_animal_registry.py Normal file
View File

@@ -0,0 +1,405 @@
# ABOUTME: Tests for AnimalRegistryProjection.
# ABOUTME: Validates animal_registry and live_animals_by_location updates on cohort creation.
import json
from animaltrack.events.types import ANIMAL_COHORT_CREATED
from animaltrack.models.events import Event
from animaltrack.projections.animal_registry import AnimalRegistryProjection
def make_cohort_event(
animal_ids: list[str],
location_id: str = "01ARZ3NDEKTSV4RRFFQ69G5FAV",
species: str = "duck",
sex: str = "unknown",
life_stage: str = "adult",
origin: str = "purchased",
ts_utc: int = 1704067200000,
) -> Event:
"""Create a test AnimalCohortCreated event."""
return Event(
id="01ARZ3NDEKTSV4RRFFQ69G5001",
type=ANIMAL_COHORT_CREATED,
ts_utc=ts_utc,
actor="test_user",
entity_refs={
"location_id": location_id,
"animal_ids": animal_ids,
},
payload={
"species": species,
"count": len(animal_ids),
"life_stage": life_stage,
"sex": sex,
"location_id": location_id,
"origin": origin,
"notes": None,
},
version=1,
)
class TestAnimalRegistryProjectionEventTypes:
"""Tests for get_event_types method."""
def test_handles_animal_cohort_created(self, seeded_db):
"""Projection handles AnimalCohortCreated event type."""
projection = AnimalRegistryProjection(seeded_db)
assert ANIMAL_COHORT_CREATED in projection.get_event_types()
class TestAnimalRegistryProjectionApply:
"""Tests for apply() on AnimalCohortCreated."""
def test_creates_animal_registry_row_for_each_animal(self, seeded_db):
"""Apply creates one row in animal_registry per animal_id."""
# Get a valid location_id from seeds
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
animal_ids = [
"01ARZ3NDEKTSV4RRFFQ69G5A01",
"01ARZ3NDEKTSV4RRFFQ69G5A02",
"01ARZ3NDEKTSV4RRFFQ69G5A03",
]
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id)
projection.apply(event)
# Check animal_registry has 3 rows
count = seeded_db.execute("SELECT COUNT(*) FROM animal_registry").fetchone()[0]
assert count == 3
# Check each animal_id exists
for animal_id in animal_ids:
row = seeded_db.execute(
"SELECT animal_id FROM animal_registry WHERE animal_id = ?",
(animal_id,),
).fetchone()
assert row is not None
def test_animal_has_correct_species(self, seeded_db):
"""Registry row has correct species_code from payload."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
animal_ids = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id, species="goose")
projection.apply(event)
row = seeded_db.execute(
"SELECT species_code FROM animal_registry WHERE animal_id = ?",
(animal_ids[0],),
).fetchone()
assert row[0] == "goose"
def test_animal_has_correct_sex(self, seeded_db):
"""Registry row has correct sex from payload."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
animal_ids = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id, sex="female")
projection.apply(event)
row = seeded_db.execute(
"SELECT sex FROM animal_registry WHERE animal_id = ?",
(animal_ids[0],),
).fetchone()
assert row[0] == "female"
def test_animal_has_correct_life_stage(self, seeded_db):
"""Registry row has correct life_stage from payload."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
animal_ids = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id, life_stage="juvenile")
projection.apply(event)
row = seeded_db.execute(
"SELECT life_stage FROM animal_registry WHERE animal_id = ?",
(animal_ids[0],),
).fetchone()
assert row[0] == "juvenile"
def test_animal_has_correct_location(self, seeded_db):
"""Registry row has correct location_id from payload."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 2'").fetchone()
location_id = row[0]
animal_ids = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id)
projection.apply(event)
row = seeded_db.execute(
"SELECT location_id FROM animal_registry WHERE animal_id = ?",
(animal_ids[0],),
).fetchone()
assert row[0] == location_id
def test_animal_has_correct_origin(self, seeded_db):
"""Registry row has correct origin from payload."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
animal_ids = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id, origin="rescued")
projection.apply(event)
row = seeded_db.execute(
"SELECT origin FROM animal_registry WHERE animal_id = ?",
(animal_ids[0],),
).fetchone()
assert row[0] == "rescued"
def test_status_is_alive(self, seeded_db):
"""Registry row has status='alive' for new cohort."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
animal_ids = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id)
projection.apply(event)
row = seeded_db.execute(
"SELECT status FROM animal_registry WHERE animal_id = ?",
(animal_ids[0],),
).fetchone()
assert row[0] == "alive"
def test_identified_is_false(self, seeded_db):
"""Registry row has identified=false for new cohort."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
animal_ids = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id)
projection.apply(event)
row = seeded_db.execute(
"SELECT identified FROM animal_registry WHERE animal_id = ?",
(animal_ids[0],),
).fetchone()
assert row[0] == 0 # SQLite stores booleans as 0/1
def test_repro_status_is_unknown(self, seeded_db):
"""Registry row has repro_status='unknown' for new cohort."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
animal_ids = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id)
projection.apply(event)
row = seeded_db.execute(
"SELECT repro_status FROM animal_registry WHERE animal_id = ?",
(animal_ids[0],),
).fetchone()
assert row[0] == "unknown"
def test_first_seen_utc_matches_event(self, seeded_db):
"""Registry row first_seen_utc matches event ts_utc."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
animal_ids = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
ts_utc = 1704067200000
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id, ts_utc=ts_utc)
projection.apply(event)
row = seeded_db.execute(
"SELECT first_seen_utc FROM animal_registry WHERE animal_id = ?",
(animal_ids[0],),
).fetchone()
assert row[0] == ts_utc
def test_last_event_utc_matches_event(self, seeded_db):
"""Registry row last_event_utc matches event ts_utc."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
animal_ids = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
ts_utc = 1704067200000
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id, ts_utc=ts_utc)
projection.apply(event)
row = seeded_db.execute(
"SELECT last_event_utc FROM animal_registry WHERE animal_id = ?",
(animal_ids[0],),
).fetchone()
assert row[0] == ts_utc
def test_creates_live_animal_row_for_each_animal(self, seeded_db):
"""Apply creates one row in live_animals_by_location per animal_id."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
animal_ids = [
"01ARZ3NDEKTSV4RRFFQ69G5A01",
"01ARZ3NDEKTSV4RRFFQ69G5A02",
]
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id)
projection.apply(event)
# Check live_animals_by_location has 2 rows
count = seeded_db.execute("SELECT COUNT(*) FROM live_animals_by_location").fetchone()[0]
assert count == 2
# Check each animal_id exists
for animal_id in animal_ids:
row = seeded_db.execute(
"SELECT animal_id FROM live_animals_by_location WHERE animal_id = ?",
(animal_id,),
).fetchone()
assert row is not None
def test_live_animal_tags_empty_json_array(self, seeded_db):
"""Live animal row has tags='[]' for new cohort."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
animal_ids = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id)
projection.apply(event)
row = seeded_db.execute(
"SELECT tags FROM live_animals_by_location WHERE animal_id = ?",
(animal_ids[0],),
).fetchone()
assert row[0] == "[]"
assert json.loads(row[0]) == []
def test_live_animal_last_move_utc_is_null(self, seeded_db):
"""Live animal row has last_move_utc=NULL for new cohort."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
animal_ids = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id)
projection.apply(event)
row = seeded_db.execute(
"SELECT last_move_utc FROM live_animals_by_location WHERE animal_id = ?",
(animal_ids[0],),
).fetchone()
assert row[0] is None
class TestAnimalRegistryProjectionRevert:
"""Tests for revert() on AnimalCohortCreated."""
def test_removes_animal_registry_rows(self, seeded_db):
"""Revert deletes rows from animal_registry."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
animal_ids = [
"01ARZ3NDEKTSV4RRFFQ69G5A01",
"01ARZ3NDEKTSV4RRFFQ69G5A02",
]
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id)
projection.apply(event)
# Verify rows exist
count = seeded_db.execute("SELECT COUNT(*) FROM animal_registry").fetchone()[0]
assert count == 2
# Revert
projection.revert(event)
# Verify rows removed
count = seeded_db.execute("SELECT COUNT(*) FROM animal_registry").fetchone()[0]
assert count == 0
def test_removes_live_animal_rows(self, seeded_db):
"""Revert deletes rows from live_animals_by_location."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
animal_ids = [
"01ARZ3NDEKTSV4RRFFQ69G5A01",
"01ARZ3NDEKTSV4RRFFQ69G5A02",
]
projection = AnimalRegistryProjection(seeded_db)
event = make_cohort_event(animal_ids, location_id=location_id)
projection.apply(event)
# Verify rows exist
count = seeded_db.execute("SELECT COUNT(*) FROM live_animals_by_location").fetchone()[0]
assert count == 2
# Revert
projection.revert(event)
# Verify rows removed
count = seeded_db.execute("SELECT COUNT(*) FROM live_animals_by_location").fetchone()[0]
assert count == 0
def test_revert_only_affects_event_animals(self, seeded_db):
"""Revert only removes animals from the specific event."""
row = seeded_db.execute("SELECT id FROM locations WHERE name = 'Strip 1'").fetchone()
location_id = row[0]
# Create first cohort
animal_ids_1 = ["01ARZ3NDEKTSV4RRFFQ69G5A01"]
projection = AnimalRegistryProjection(seeded_db)
event1 = make_cohort_event(animal_ids_1, location_id=location_id)
projection.apply(event1)
# Create second cohort with different event
animal_ids_2 = ["01ARZ3NDEKTSV4RRFFQ69G5A02"]
event2 = Event(
id="01ARZ3NDEKTSV4RRFFQ69G5002", # Different event ID
type=ANIMAL_COHORT_CREATED,
ts_utc=1704067300000,
actor="test_user",
entity_refs={
"location_id": location_id,
"animal_ids": animal_ids_2,
},
payload={
"species": "duck",
"count": 1,
"life_stage": "adult",
"sex": "unknown",
"location_id": location_id,
"origin": "purchased",
"notes": None,
},
version=1,
)
projection.apply(event2)
# Verify both exist
count = seeded_db.execute("SELECT COUNT(*) FROM animal_registry").fetchone()[0]
assert count == 2
# Revert only event1
projection.revert(event1)
# Event2's animal should still exist
count = seeded_db.execute("SELECT COUNT(*) FROM animal_registry").fetchone()[0]
assert count == 1
row = seeded_db.execute("SELECT animal_id FROM animal_registry").fetchone()
assert row[0] == animal_ids_2[0]