1 files changed, 79 insertions, 0 deletions
diff --git a/tests/test_aggregator.py b/tests/test_aggregator.py
new file mode 100644
index 0000000..c840cac
--- /dev/null
+++ b/tests/test_aggregator.py
@@ -0,0 +1,79 @@
+from datetime import UTC, datetime, date
+
+from sqlalchemy.orm import Session
+
+from srht_contrib.models import ContributionEvent
+from srht_contrib.services.aggregator import ContributionAggregator
+
+
+def test_aggregator_zero_fills_days(db_session: Session) -> None:
+    db_session.add(
+        ContributionEvent(
+            service="todo",
+            event_type="ticket_created",
+            actor="~ccleberg",
+            repo_name=None,
+            resource_id="1",
+            external_uid="todo:1:created",
+            occurred_at=datetime(2026, 3, 28, 12, 0, tzinfo=UTC),
+            weight=1.0,
+            raw_payload_json=None,
+        )
+    )
+    db_session.commit()
+
+    calendar = ContributionAggregator().build_calendar(
+        db_session,
+        "~ccleberg",
+        date(2026, 3, 28),
+        date(2026, 3, 30),
+    )
+
+    assert [day.model_dump() for day in calendar.days] == [
+        {"date": date(2026, 3, 28), "count": 1, "score": 1.0},
+        {"date": date(2026, 3, 29), "count": 0, "score": 0.0},
+        {"date": date(2026, 3, 30), "count": 0, "score": 0.0},
+    ]
+
+
+def test_stats_calculation(db_session: Session) -> None:
+    db_session.add_all(
+        [
+            ContributionEvent(
+                service="todo",
+                event_type="ticket_created",
+                actor="~ccleberg",
+                repo_name=None,
+                resource_id="1",
+                external_uid="todo:1:created",
+                occurred_at=datetime(2026, 3, 28, 12, 0, tzinfo=UTC),
+                weight=1.0,
+                raw_payload_json=None,
+            ),
+            ContributionEvent(
+                service="todo",
+                event_type="ticket_comment",
+                actor="~ccleberg",
+                repo_name=None,
+                resource_id="1",
+                external_uid="todo:1:comment:1",
+                occurred_at=datetime(2026, 3, 29, 12, 0, tzinfo=UTC),
+                weight=0.5,
+                raw_payload_json=None,
+            ),
+        ]
+    )
+    db_session.commit()
+
+    stats = ContributionAggregator().build_stats(
+        db_session,
+        "~ccleberg",
+        date(2026, 3, 28),
+        date(2026, 3, 30),
+    )
+
+    assert stats.total_events == 2
+    assert stats.total_score == 1.5
+    assert stats.active_days == 2
+    assert stats.longest_streak == 2
+    assert stats.current_streak == 0