[kimi] Split scorecard_service.py into focused modules (#1406) (#1461)

This commit is contained in:
2026-03-24 20:06:31 +00:00
parent 4d2aeb937f
commit fe7e14b10e
12 changed files with 766 additions and 620 deletions

View File

@@ -1,10 +1,10 @@
"""Unit tests for dashboard/services/scorecard_service.py.
"""Unit tests for dashboard/services/scorecard package.
Focuses on edge cases and scenarios not covered in test_scorecards.py:
- _aggregate_metrics: test.execution events, PR-closed-without-merge,
- aggregate_metrics: test.execution events, PR-closed-without-merge,
push default commit count, untracked agent with agent_id passthrough
- _detect_patterns: boundary conditions (< 3 PRs, exactly 3, exactly 80%)
- _generate_narrative_bullets: singular/plural forms
- detect_patterns: boundary conditions (< 3 PRs, exactly 3, exactly 80%)
- generate_narrative_bullets: singular/plural forms
- generate_scorecard: token augmentation max() logic
- ScorecardSummary.to_dict(): ISO timestamp format, tests_affected count
"""
@@ -18,31 +18,31 @@ import pytest
pytestmark = pytest.mark.unit
from dashboard.services.scorecard_service import (
from dashboard.services.scorecard import (
AgentMetrics,
PeriodType,
ScorecardSummary,
_aggregate_metrics,
_detect_patterns,
_generate_narrative_bullets,
generate_scorecard,
)
from dashboard.services.scorecard.aggregators import aggregate_metrics
from dashboard.services.scorecard.calculators import detect_patterns
from dashboard.services.scorecard.formatters import generate_narrative_bullets
from infrastructure.events.bus import Event
# ---------------------------------------------------------------------------
# _aggregate_metrics — edge cases
# aggregate_metrics — edge cases
# ---------------------------------------------------------------------------
class TestAggregateMetricsEdgeCases:
"""Edge cases for _aggregate_metrics not covered in test_scorecards.py."""
"""Edge cases for aggregate_metrics not covered in test_scorecards.py."""
def test_push_event_defaults_to_one_commit(self):
"""Push event with no num_commits key should count as 1 commit."""
events = [
Event(type="gitea.push", source="gitea", data={"actor": "claude"}),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
assert result["claude"].commits == 1
@@ -55,7 +55,7 @@ class TestAggregateMetricsEdgeCases:
data={"actor": "kimi", "pr_number": 99, "action": "closed", "merged": False},
),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
# PR was not merged — should not be in prs_merged
assert "kimi" in result
@@ -77,7 +77,7 @@ class TestAggregateMetricsEdgeCases:
},
),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
assert "gemini" in result
assert "tests/test_alpha.py" in result["gemini"].tests_affected
@@ -92,7 +92,7 @@ class TestAggregateMetricsEdgeCases:
data={"agent_id": "kimi", "tests_affected": [], "token_reward": 5},
),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
# kimi is tracked and agent_id is present in data
assert "kimi" in result
@@ -107,7 +107,7 @@ class TestAggregateMetricsEdgeCases:
data={"actor": "anon-bot", "num_commits": 10},
),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
assert "anon-bot" not in result
@@ -120,7 +120,7 @@ class TestAggregateMetricsEdgeCases:
data={"actor": "hermes", "issue_number": 0},
),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
assert "hermes" in result
assert len(result["hermes"].issues_touched) == 0
@@ -134,7 +134,7 @@ class TestAggregateMetricsEdgeCases:
data={"actor": "manus", "issue_number": 0},
),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
assert "manus" in result
assert result["manus"].comments == 1
@@ -149,7 +149,7 @@ class TestAggregateMetricsEdgeCases:
data={"agent_id": "claude", "tests_affected": [], "token_reward": 20},
),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
assert "claude" in result
assert len(result["claude"].tests_affected) == 0
@@ -161,7 +161,7 @@ class TestAggregateMetricsEdgeCases:
Event(type="gitea.push", source="gitea", data={"actor": "claude", "num_commits": 3}),
Event(type="gitea.push", source="gitea", data={"actor": "gemini", "num_commits": 7}),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
assert result["claude"].commits == 3
assert result["gemini"].commits == 7
@@ -175,7 +175,7 @@ class TestAggregateMetricsEdgeCases:
data={"actor": "kimi", "pr_number": 0, "action": "opened"},
),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
assert "kimi" in result
assert len(result["kimi"].prs_opened) == 0
@@ -192,7 +192,7 @@ class TestDetectPatternsBoundaries:
def test_no_patterns_with_empty_metrics(self):
"""Empty metrics should not trigger any patterns."""
metrics = AgentMetrics(agent_id="kimi")
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
assert patterns == []
@@ -203,7 +203,7 @@ class TestDetectPatternsBoundaries:
prs_opened={1, 2},
prs_merged={1, 2}, # 100% rate but only 2 PRs
)
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
# Should NOT trigger high-merge-rate pattern (< 3 PRs)
assert not any("High merge rate" in p for p in patterns)
@@ -216,7 +216,7 @@ class TestDetectPatternsBoundaries:
prs_opened={1, 2, 3},
prs_merged={1, 2, 3}, # 100% rate, 3 PRs
)
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
assert any("High merge rate" in p for p in patterns)
@@ -227,7 +227,7 @@ class TestDetectPatternsBoundaries:
prs_opened={1, 2, 3, 4, 5},
prs_merged={1, 2, 3, 4}, # 80%
)
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
assert any("High merge rate" in p for p in patterns)
@@ -238,7 +238,7 @@ class TestDetectPatternsBoundaries:
prs_opened={1, 2, 3, 4, 5, 6, 7}, # 7 PRs
prs_merged={1, 2, 3, 4, 5}, # ~71.4% — below 80%
)
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
assert not any("High merge rate" in p for p in patterns)
@@ -249,7 +249,7 @@ class TestDetectPatternsBoundaries:
commits=10,
prs_opened=set(),
)
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
assert not any("High commit volume" in p for p in patterns)
@@ -260,27 +260,27 @@ class TestDetectPatternsBoundaries:
commits=11,
prs_opened=set(),
)
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
assert any("High commit volume without PRs" in p for p in patterns)
def test_token_accumulation_exact_boundary(self):
"""Net tokens = 100 does NOT trigger accumulation pattern (must be > 100)."""
metrics = AgentMetrics(agent_id="kimi", tokens_earned=100, tokens_spent=0)
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
assert not any("Strong token accumulation" in p for p in patterns)
def test_token_spend_exact_boundary(self):
"""Net tokens = -50 does NOT trigger high spend pattern (must be < -50)."""
metrics = AgentMetrics(agent_id="kimi", tokens_earned=0, tokens_spent=50)
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
assert not any("High token spend" in p for p in patterns)
# ---------------------------------------------------------------------------
# _generate_narrative_bullets — singular/plural
# generate_narrative_bullets — singular/plural
# ---------------------------------------------------------------------------
@@ -290,7 +290,7 @@ class TestGenerateNarrativeSingularPlural:
def test_singular_commit(self):
"""One commit should use singular form."""
metrics = AgentMetrics(agent_id="kimi", commits=1)
bullets = _generate_narrative_bullets(metrics, PeriodType.daily)
bullets = generate_narrative_bullets(metrics, PeriodType.daily)
activity = next((b for b in bullets if "Active across" in b), None)
assert activity is not None
@@ -300,7 +300,7 @@ class TestGenerateNarrativeSingularPlural:
def test_singular_pr_opened(self):
"""One opened PR should use singular form."""
metrics = AgentMetrics(agent_id="kimi", prs_opened={1})
bullets = _generate_narrative_bullets(metrics, PeriodType.daily)
bullets = generate_narrative_bullets(metrics, PeriodType.daily)
activity = next((b for b in bullets if "Active across" in b), None)
assert activity is not None
@@ -309,7 +309,7 @@ class TestGenerateNarrativeSingularPlural:
def test_singular_pr_merged(self):
"""One merged PR should use singular form."""
metrics = AgentMetrics(agent_id="kimi", prs_merged={1})
bullets = _generate_narrative_bullets(metrics, PeriodType.daily)
bullets = generate_narrative_bullets(metrics, PeriodType.daily)
activity = next((b for b in bullets if "Active across" in b), None)
assert activity is not None
@@ -318,7 +318,7 @@ class TestGenerateNarrativeSingularPlural:
def test_singular_issue_touched(self):
"""One issue touched should use singular form."""
metrics = AgentMetrics(agent_id="kimi", issues_touched={42})
bullets = _generate_narrative_bullets(metrics, PeriodType.daily)
bullets = generate_narrative_bullets(metrics, PeriodType.daily)
activity = next((b for b in bullets if "Active across" in b), None)
assert activity is not None
@@ -327,7 +327,7 @@ class TestGenerateNarrativeSingularPlural:
def test_singular_comment(self):
"""One comment should use singular form."""
metrics = AgentMetrics(agent_id="kimi", comments=1)
bullets = _generate_narrative_bullets(metrics, PeriodType.daily)
bullets = generate_narrative_bullets(metrics, PeriodType.daily)
activity = next((b for b in bullets if "Active across" in b), None)
assert activity is not None
@@ -336,14 +336,14 @@ class TestGenerateNarrativeSingularPlural:
def test_singular_test_file(self):
"""One test file should use singular form."""
metrics = AgentMetrics(agent_id="kimi", tests_affected={"test_foo.py"})
bullets = _generate_narrative_bullets(metrics, PeriodType.daily)
bullets = generate_narrative_bullets(metrics, PeriodType.daily)
assert any("1 test file." in b for b in bullets)
def test_weekly_period_label(self):
"""Weekly period uses 'week' label in no-activity message."""
metrics = AgentMetrics(agent_id="kimi")
bullets = _generate_narrative_bullets(metrics, PeriodType.weekly)
bullets = generate_narrative_bullets(metrics, PeriodType.weekly)
assert any("this week" in b for b in bullets)
@@ -366,11 +366,11 @@ class TestGenerateScorecardTokenAugmentation:
),
]
with patch(
"dashboard.services.scorecard_service._collect_events_for_period",
"dashboard.services.scorecard.core.collect_events_for_period",
return_value=events,
):
with patch(
"dashboard.services.scorecard_service._query_token_transactions",
"dashboard.services.scorecard.core.query_token_transactions",
return_value=(50, 0), # ledger says 50 earned
):
scorecard = generate_scorecard("kimi", PeriodType.daily)
@@ -388,11 +388,11 @@ class TestGenerateScorecardTokenAugmentation:
),
]
with patch(
"dashboard.services.scorecard_service._collect_events_for_period",
"dashboard.services.scorecard.core.collect_events_for_period",
return_value=events,
):
with patch(
"dashboard.services.scorecard_service._query_token_transactions",
"dashboard.services.scorecard.core.query_token_transactions",
return_value=(500, 100), # ledger says 500 earned, 100 spent
):
scorecard = generate_scorecard("kimi", PeriodType.daily)

View File

@@ -3,21 +3,22 @@
from datetime import UTC, datetime, timedelta
from unittest.mock import MagicMock, patch
from dashboard.services.scorecard_service import (
from dashboard.services.scorecard import (
AgentMetrics,
PeriodType,
ScorecardSummary,
_aggregate_metrics,
_detect_patterns,
_extract_actor_from_event,
_generate_narrative_bullets,
_get_period_bounds,
_is_tracked_agent,
_query_token_transactions,
generate_all_scorecards,
generate_scorecard,
get_tracked_agents,
)
from dashboard.services.scorecard.aggregators import aggregate_metrics, query_token_transactions
from dashboard.services.scorecard.calculators import detect_patterns
from dashboard.services.scorecard.formatters import generate_narrative_bullets
from dashboard.services.scorecard.validators import (
extract_actor_from_event,
get_period_bounds,
is_tracked_agent,
)
from infrastructure.events.bus import Event
@@ -27,7 +28,7 @@ class TestPeriodBounds:
def test_daily_period_bounds(self):
"""Test daily period returns correct 24-hour window."""
reference = datetime(2026, 3, 21, 12, 30, 45, tzinfo=UTC)
start, end = _get_period_bounds(PeriodType.daily, reference)
start, end = get_period_bounds(PeriodType.daily, reference)
assert end == datetime(2026, 3, 21, 0, 0, 0, tzinfo=UTC)
assert start == datetime(2026, 3, 20, 0, 0, 0, tzinfo=UTC)
@@ -36,7 +37,7 @@ class TestPeriodBounds:
def test_weekly_period_bounds(self):
"""Test weekly period returns correct 7-day window."""
reference = datetime(2026, 3, 21, 12, 30, 45, tzinfo=UTC)
start, end = _get_period_bounds(PeriodType.weekly, reference)
start, end = get_period_bounds(PeriodType.weekly, reference)
assert end == datetime(2026, 3, 21, 0, 0, 0, tzinfo=UTC)
assert start == datetime(2026, 3, 14, 0, 0, 0, tzinfo=UTC)
@@ -44,7 +45,7 @@ class TestPeriodBounds:
def test_default_reference_date(self):
"""Test default reference date uses current time."""
start, end = _get_period_bounds(PeriodType.daily)
start, end = get_period_bounds(PeriodType.daily)
now = datetime.now(UTC)
# End should be start of current day (midnight)
@@ -70,16 +71,16 @@ class TestTrackedAgents:
def test_is_tracked_agent_true(self):
"""Test _is_tracked_agent returns True for tracked agents."""
assert _is_tracked_agent("kimi") is True
assert _is_tracked_agent("KIMI") is True # case insensitive
assert _is_tracked_agent("claude") is True
assert _is_tracked_agent("hermes") is True
assert is_tracked_agent("kimi") is True
assert is_tracked_agent("KIMI") is True # case insensitive
assert is_tracked_agent("claude") is True
assert is_tracked_agent("hermes") is True
def test_is_tracked_agent_false(self):
"""Test _is_tracked_agent returns False for untracked agents."""
assert _is_tracked_agent("unknown") is False
assert _is_tracked_agent("rockachopa") is False
assert _is_tracked_agent("") is False
assert is_tracked_agent("unknown") is False
assert is_tracked_agent("rockachopa") is False
assert is_tracked_agent("") is False
class TestExtractActor:
@@ -88,22 +89,22 @@ class TestExtractActor:
def test_extract_from_actor_field(self):
"""Test extraction from data.actor field."""
event = Event(type="test", source="system", data={"actor": "kimi"})
assert _extract_actor_from_event(event) == "kimi"
assert extract_actor_from_event(event) == "kimi"
def test_extract_from_agent_id_field(self):
"""Test extraction from data.agent_id field."""
event = Event(type="test", source="system", data={"agent_id": "claude"})
assert _extract_actor_from_event(event) == "claude"
assert extract_actor_from_event(event) == "claude"
def test_extract_from_source_fallback(self):
"""Test fallback to event.source."""
event = Event(type="test", source="gemini", data={})
assert _extract_actor_from_event(event) == "gemini"
assert extract_actor_from_event(event) == "gemini"
def test_actor_priority_over_agent_id(self):
"""Test actor field takes priority over agent_id."""
event = Event(type="test", source="system", data={"actor": "kimi", "agent_id": "claude"})
assert _extract_actor_from_event(event) == "kimi"
assert extract_actor_from_event(event) == "kimi"
class TestAggregateMetrics:
@@ -111,7 +112,7 @@ class TestAggregateMetrics:
def test_empty_events(self):
"""Test aggregation with no events returns empty dict."""
result = _aggregate_metrics([])
result = aggregate_metrics([])
assert result == {}
def test_push_event_aggregation(self):
@@ -120,7 +121,7 @@ class TestAggregateMetrics:
Event(type="gitea.push", source="gitea", data={"actor": "kimi", "num_commits": 3}),
Event(type="gitea.push", source="gitea", data={"actor": "kimi", "num_commits": 2}),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
assert "kimi" in result
assert result["kimi"].commits == 5
@@ -139,7 +140,7 @@ class TestAggregateMetrics:
data={"actor": "claude", "issue_number": 101},
),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
assert "claude" in result
assert len(result["claude"].issues_touched) == 2
@@ -160,7 +161,7 @@ class TestAggregateMetrics:
data={"actor": "gemini", "issue_number": 101},
),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
assert "gemini" in result
assert result["gemini"].comments == 2
@@ -185,7 +186,7 @@ class TestAggregateMetrics:
data={"actor": "kimi", "pr_number": 51, "action": "opened"},
),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
assert "kimi" in result
assert len(result["kimi"].prs_opened) == 2
@@ -199,7 +200,7 @@ class TestAggregateMetrics:
type="gitea.push", source="gitea", data={"actor": "rockachopa", "num_commits": 5}
),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
assert "rockachopa" not in result
@@ -216,7 +217,7 @@ class TestAggregateMetrics:
},
),
]
result = _aggregate_metrics(events)
result = aggregate_metrics(events)
assert "kimi" in result
assert len(result["kimi"].tests_affected) == 2
@@ -253,7 +254,7 @@ class TestDetectPatterns:
prs_opened={1, 2, 3, 4, 5},
prs_merged={1, 2, 3, 4}, # 80% merge rate
)
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
assert any("High merge rate" in p for p in patterns)
@@ -264,7 +265,7 @@ class TestDetectPatterns:
prs_opened={1, 2, 3, 4, 5},
prs_merged={1}, # 20% merge rate
)
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
assert any("low merge rate" in p for p in patterns)
@@ -275,7 +276,7 @@ class TestDetectPatterns:
commits=15,
prs_opened=set(),
)
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
assert any("High commit volume without PRs" in p for p in patterns)
@@ -286,7 +287,7 @@ class TestDetectPatterns:
issues_touched={1, 2, 3, 4, 5, 6},
comments=0,
)
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
assert any("silent worker" in p for p in patterns)
@@ -297,7 +298,7 @@ class TestDetectPatterns:
issues_touched={1, 2}, # 2 issues
comments=10, # 5x comments per issue
)
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
assert any("Highly communicative" in p for p in patterns)
@@ -308,7 +309,7 @@ class TestDetectPatterns:
tokens_earned=150,
tokens_spent=10,
)
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
assert any("Strong token accumulation" in p for p in patterns)
@@ -319,7 +320,7 @@ class TestDetectPatterns:
tokens_earned=10,
tokens_spent=100,
)
patterns = _detect_patterns(metrics)
patterns = detect_patterns(metrics)
assert any("High token spend" in p for p in patterns)
@@ -330,7 +331,7 @@ class TestGenerateNarrative:
def test_empty_metrics_narrative(self):
"""Test narrative for empty metrics mentions no activity."""
metrics = AgentMetrics(agent_id="kimi")
bullets = _generate_narrative_bullets(metrics, PeriodType.daily)
bullets = generate_narrative_bullets(metrics, PeriodType.daily)
assert len(bullets) == 1
assert "No recorded activity" in bullets[0]
@@ -343,7 +344,7 @@ class TestGenerateNarrative:
prs_opened={1, 2},
prs_merged={1},
)
bullets = _generate_narrative_bullets(metrics, PeriodType.daily)
bullets = generate_narrative_bullets(metrics, PeriodType.daily)
activity_bullet = next((b for b in bullets if "Active across" in b), None)
assert activity_bullet is not None
@@ -357,7 +358,7 @@ class TestGenerateNarrative:
agent_id="kimi",
tests_affected={"test_a.py", "test_b.py"},
)
bullets = _generate_narrative_bullets(metrics, PeriodType.daily)
bullets = generate_narrative_bullets(metrics, PeriodType.daily)
assert any("2 test files" in b for b in bullets)
@@ -368,7 +369,7 @@ class TestGenerateNarrative:
tokens_earned=100,
tokens_spent=20,
)
bullets = _generate_narrative_bullets(metrics, PeriodType.daily)
bullets = generate_narrative_bullets(metrics, PeriodType.daily)
assert any("Net earned 80 tokens" in b for b in bullets)
@@ -379,7 +380,7 @@ class TestGenerateNarrative:
tokens_earned=20,
tokens_spent=100,
)
bullets = _generate_narrative_bullets(metrics, PeriodType.daily)
bullets = generate_narrative_bullets(metrics, PeriodType.daily)
assert any("Net spent 80 tokens" in b for b in bullets)
@@ -390,7 +391,7 @@ class TestGenerateNarrative:
tokens_earned=100,
tokens_spent=100,
)
bullets = _generate_narrative_bullets(metrics, PeriodType.daily)
bullets = generate_narrative_bullets(metrics, PeriodType.daily)
assert any("Balanced token flow" in b for b in bullets)
@@ -438,7 +439,7 @@ class TestQueryTokenTransactions:
def test_empty_ledger(self):
"""Test empty ledger returns zero values."""
with patch("lightning.ledger.get_transactions", return_value=[]):
earned, spent = _query_token_transactions("kimi", datetime.now(UTC), datetime.now(UTC))
earned, spent = query_token_transactions("kimi", datetime.now(UTC), datetime.now(UTC))
assert earned == 0
assert spent == 0
@@ -460,7 +461,7 @@ class TestQueryTokenTransactions:
),
]
with patch("lightning.ledger.get_transactions", return_value=mock_tx):
earned, spent = _query_token_transactions(
earned, spent = query_token_transactions(
"kimi", now - timedelta(hours=1), now + timedelta(hours=1)
)
assert earned == 100
@@ -478,7 +479,7 @@ class TestQueryTokenTransactions:
),
]
with patch("lightning.ledger.get_transactions", return_value=mock_tx):
earned, spent = _query_token_transactions(
earned, spent = query_token_transactions(
"kimi", now - timedelta(hours=1), now + timedelta(hours=1)
)
assert earned == 0 # Transaction was for claude, not kimi
@@ -497,7 +498,7 @@ class TestQueryTokenTransactions:
]
with patch("lightning.ledger.get_transactions", return_value=mock_tx):
# Query for today only
earned, spent = _query_token_transactions(
earned, spent = query_token_transactions(
"kimi", now - timedelta(hours=1), now + timedelta(hours=1)
)
assert earned == 0 # Transaction was 2 days ago
@@ -508,11 +509,9 @@ class TestGenerateScorecard:
def test_generate_scorecard_no_activity(self):
"""Test scorecard generation for agent with no activity."""
with patch(
"dashboard.services.scorecard_service._collect_events_for_period", return_value=[]
):
with patch("dashboard.services.scorecard.core.collect_events_for_period", return_value=[]):
with patch(
"dashboard.services.scorecard_service._query_token_transactions",
"dashboard.services.scorecard.core.query_token_transactions",
return_value=(0, 0),
):
scorecard = generate_scorecard("kimi", PeriodType.daily)
@@ -529,10 +528,10 @@ class TestGenerateScorecard:
Event(type="gitea.push", source="gitea", data={"actor": "kimi", "num_commits": 5}),
]
with patch(
"dashboard.services.scorecard_service._collect_events_for_period", return_value=events
"dashboard.services.scorecard.core.collect_events_for_period", return_value=events
):
with patch(
"dashboard.services.scorecard_service._query_token_transactions",
"dashboard.services.scorecard.core.query_token_transactions",
return_value=(100, 20),
):
scorecard = generate_scorecard("kimi", PeriodType.daily)
@@ -548,11 +547,9 @@ class TestGenerateAllScorecards:
def test_generates_for_all_tracked_agents(self):
"""Test all tracked agents get scorecards even with no activity."""
with patch(
"dashboard.services.scorecard_service._collect_events_for_period", return_value=[]
):
with patch("dashboard.services.scorecard.core.collect_events_for_period", return_value=[]):
with patch(
"dashboard.services.scorecard_service._query_token_transactions",
"dashboard.services.scorecard.core.query_token_transactions",
return_value=(0, 0),
):
scorecards = generate_all_scorecards(PeriodType.daily)
@@ -563,11 +560,9 @@ class TestGenerateAllScorecards:
def test_scorecards_sorted(self):
"""Test scorecards are sorted by agent_id."""
with patch(
"dashboard.services.scorecard_service._collect_events_for_period", return_value=[]
):
with patch("dashboard.services.scorecard.core.collect_events_for_period", return_value=[]):
with patch(
"dashboard.services.scorecard_service._query_token_transactions",
"dashboard.services.scorecard.core.query_token_transactions",
return_value=(0, 0),
):
scorecards = generate_all_scorecards(PeriodType.daily)