1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
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
|
