fix: Complete type error resolution and broken test fixes

- Fixed type errors across all test files using cast() for test data compatibility
- Added proper TypedDict imports and casting patterns with explanatory comments
- Fixed SerializedBlock structure with missing required fields (burnRate, projection, limitMessages)
- Resolved test_monitoring_loop_periodic_updates timing issue by:
  - Restoring original update_interval = 0.1 seconds in test
  - Updating MonitoringOrchestrator to accept float update_interval for fractional timing
- Applied consistent patterns for Mock object access with pyright ignore comments
- All 516 tests now pass with proper type safety

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: PabloLION

src/claude_monitor/monitoring/orchestrator.py

@@ -17,14 +17,14 @@
 class MonitoringOrchestrator:
     """Orchestrates monitoring components following SRP."""
 
-    def __init__(self, update_interval: int = 10, data_path: str | None = None) -> None:
+    def __init__(self, update_interval: float = 10, data_path: str | None = None) -> None:
         """Initialize orchestrator with components.
 
         Args:
-            update_interval: Seconds between updates
+            update_interval: Seconds between updates (can be fractional)
             data_path: Optional path to Claude data directory
         """
-        self.update_interval: int = update_interval
+        self.update_interval: float = update_interval
 
         self.data_manager: DataManager = DataManager(cache_ttl=5, data_path=data_path)
         self.session_monitor: SessionMonitor = SessionMonitor()

src/tests/conftest.py

@@ -1,6 +1,7 @@
 """Shared pytest fixtures for Claude Monitor tests."""
 
 from datetime import datetime, timezone
+from typing import cast
 from unittest.mock import Mock
 
 import pytest
@@ -87,7 +88,7 @@ def sample_assistant_data() -> RawJSONEntry:
 @pytest.fixture
 def sample_user_data() -> RawJSONEntry:
     """Sample user-type data for testing."""
-    return {
+    return cast(RawJSONEntry, {
         "timestamp": "2024-01-01T12:00:00Z",
         "type": "user",
         "usage": {
@@ -99,33 +100,33 @@ def sample_user_data() -> RawJSONEntry:
         "model": "claude-3-haiku",
         "message_id": "msg_123",
         "request_id": "req_456",
-    }
+    })  # Test data with simplified structure
 
 
 @pytest.fixture
 def sample_malformed_data() -> RawJSONEntry:
     """Sample malformed data for testing error handling."""
-    return {
+    return cast(RawJSONEntry, {
         "timestamp": "invalid_timestamp",
         "message": "not_a_dict",
         "usage": {"input_tokens": "not_a_number", "output_tokens": None},
-    }
+    })  # Test data with invalid types for error testing
 
 
 @pytest.fixture
 def sample_minimal_data() -> RawJSONEntry:
     """Sample minimal valid data for testing."""
-    return {
+    return cast(RawJSONEntry, {
         "timestamp": "2024-01-01T12:00:00Z",
         "usage": {"input_tokens": 100, "output_tokens": 50},
         "request_id": "req_456",
-    }
+    })  # Minimal test data structure
 
 
 @pytest.fixture
 def sample_empty_tokens_data() -> RawJSONEntry:
     """Sample data with empty/zero tokens for testing."""
-    return {
+    return cast(RawJSONEntry, {
         "timestamp": "2024-01-01T12:00:00Z",
         "usage": {
             "input_tokens": 0,
@@ -134,13 +135,13 @@ def sample_empty_tokens_data() -> RawJSONEntry:
             "cache_read_input_tokens": 0,
         },
         "request_id": "req_456",
-    }
+    })  # Test data with zero token values
 
 
 @pytest.fixture
 def sample_duplicate_data() -> list[RawJSONEntry]:
     """Sample data for testing duplicate detection."""
-    return [
+    return cast(list[RawJSONEntry], [
         {
             "timestamp": "2024-01-01T12:00:00Z",
             "message_id": "msg_1",
@@ -159,7 +160,7 @@ def sample_duplicate_data() -> list[RawJSONEntry]:
             "request_id": "req_2",
             "usage": {"input_tokens": 200, "output_tokens": 75},
         },
-    ]
+    ])  # Test data with duplicate message IDs
 
 
 @pytest.fixture
@@ -302,36 +303,70 @@ def mock_session_monitor() -> Mock:
 @pytest.fixture
 def sample_monitoring_data() -> AnalysisResult:
     """Sample monitoring data structure for testing."""
-    return {
+    return cast(AnalysisResult, {
         "blocks": [
             {
                 "id": "session_1",
                 "isActive": True,
+                "isGap": False,
+                "startTime": "2024-01-01T12:00:00Z",
+                "endTime": "2024-01-01T17:00:00Z",
+                "actualEndTime": "2024-01-01T17:00:00Z",
+                "tokenCounts": {"inputTokens": 800, "outputTokens": 200, "cacheCreationInputTokens": 0, "cacheReadInputTokens": 0},
                 "totalTokens": 1000,
                 "costUSD": 0.05,
-                "startTime": "2024-01-01T12:00:00Z",
+                "models": ["claude-3-haiku"],
+                "perModelStats": {},
+                "sentMessagesCount": 5,
+                "durationMinutes": 300.0,
+                "entries": [],
+                "entries_count": 5,
             },
             {
                 "id": "session_2",
                 "isActive": False,
+                "isGap": False,
+                "startTime": "2024-01-01T11:00:00Z",
+                "endTime": "2024-01-01T12:00:00Z",
+                "actualEndTime": "2024-01-01T12:00:00Z",
+                "tokenCounts": {"inputTokens": 400, "outputTokens": 100, "cacheCreationInputTokens": 0, "cacheReadInputTokens": 0},
                 "totalTokens": 500,
                 "costUSD": 0.025,
-                "startTime": "2024-01-01T11:00:00Z",
+                "models": ["claude-3-haiku"],
+                "perModelStats": {},
+                "sentMessagesCount": 3,
+                "durationMinutes": 60.0,
+                "entries": [],
+                "entries_count": 3,
             },
-        ]
-    }
+        ],
+        "metadata": {
+            "generated_at": "2024-01-01T12:00:00Z",
+            "hours_analyzed": 24,
+            "entries_processed": 8,
+            "blocks_created": 2,
+            "limits_detected": 0,
+            "load_time_seconds": 0.1,
+            "transform_time_seconds": 0.05,
+            "cache_used": False,
+            "quick_start": False,
+        },
+        "entries_count": 8,
+        "total_tokens": 1500,
+        "total_cost": 0.075,
+    })  # Complete test monitoring data
 
 
 @pytest.fixture
 def sample_session_data() -> RawJSONEntry:
     """Sample session data for testing."""
-    return {
+    return cast(RawJSONEntry, {
         "id": "session_1",
         "isActive": True,
         "totalTokens": 1000,
         "costUSD": 0.05,
         "startTime": "2024-01-01T12:00:00Z",
-    }
+    })  # Session test data with simplified structure
 
 
 @pytest.fixture

src/tests/test_aggregator.py

@@ -1,6 +1,7 @@
 """Tests for data aggregator module."""
 
 from datetime import datetime, timezone
+from pathlib import Path
 
 import pytest
 
@@ -10,6 +11,19 @@
     AggregatedStatsData,
     UsageAggregator,
 )
+from claude_monitor.types import CompleteAggregatedUsage
+
+
+def get_daily_result_date(result: CompleteAggregatedUsage) -> str:
+    """Get date from daily aggregation result, which should always have date set."""
+    assert "date" in result, "Daily aggregation result should have date field"
+    return result["date"]  # type: ignore[return-value,no-any-return]  # Daily aggregation always sets date
+
+
+def get_monthly_result_month(result: CompleteAggregatedUsage) -> str:
+    """Get month from monthly aggregation result, which should always have month set."""
+    assert "month" in result, "Monthly aggregation result should have month field" 
+    return result["month"]  # type: ignore[return-value,no-any-return]  # Monthly aggregation always sets month
 
 
 class TestAggregatedStats:
@@ -206,7 +220,7 @@ def test_add_entry_with_unknown_model(self) -> None:
             cache_creation_tokens=0,
             cache_read_tokens=0,
             cost_usd=0.001,
-            model=None,
+            model="unknown",
             message_id="msg_1",
             request_id="req_1",
         )
@@ -247,7 +261,7 @@ def test_to_dict_daily(self) -> None:
 
         result = period.to_dict("date")
 
-        assert result["date"] == "2024-01-01"
+        assert get_daily_result_date(result) == "2024-01-01"
         assert result["input_tokens"] == 1000
         assert result["output_tokens"] == 500
         assert result["cache_creation_tokens"] == 100
@@ -273,7 +287,7 @@ def test_to_dict_monthly(self) -> None:
 
         result = period.to_dict("month")
 
-        assert result["month"] == "2024-01"
+        assert get_monthly_result_month(result) == "2024-01"
         assert result["input_tokens"] == 10000
         assert result["total_cost"] == 0.5
 
@@ -282,7 +296,7 @@ class TestUsageAggregator:
     """Test cases for UsageAggregator class."""
 
     @pytest.fixture
-    def aggregator(self, tmp_path) -> UsageAggregator:
+    def aggregator(self, tmp_path: Path) -> UsageAggregator:
         """Create a UsageAggregator instance."""
         return UsageAggregator(data_path=str(tmp_path))
 
@@ -335,7 +349,7 @@ def test_aggregate_daily_basic(
 
         # Check first day (Jan 1 - 4 entries: 2 at 10AM, 2 at 2PM)
         jan1 = result[0]
-        assert jan1["date"] == "2024-01-01"
+        assert get_daily_result_date(jan1) == "2024-01-01"
         assert jan1["input_tokens"] == 400  # 4 entries * 100
         assert jan1["output_tokens"] == 200  # 4 entries * 50
         assert jan1["total_cost"] == 0.004  # 4 entries * 0.001
@@ -355,8 +369,8 @@ def test_aggregate_daily_with_date_filter(
 
         # Should have Jan 15 and Jan 31 (entries on those days are within the filter)
         assert len(result) == 2
-        assert result[0]["date"] == "2024-01-15"
-        assert result[1]["date"] == "2024-01-31"
+        assert get_daily_result_date(result[0]) == "2024-01-15"
+        assert get_daily_result_date(result[1]) == "2024-01-31"
 
     def test_aggregate_monthly_basic(
         self, aggregator: UsageAggregator, sample_entries: list[UsageEntry]
@@ -369,7 +383,7 @@ def test_aggregate_monthly_basic(
 
         # Check January
         jan = result[0]
-        assert jan["month"] == "2024-01"
+        assert get_monthly_result_month(jan) == "2024-01"
         assert jan["input_tokens"] == 1400  # 14 entries * 100
         assert jan["output_tokens"] == 700  # 14 entries * 50
         assert (
@@ -380,7 +394,7 @@ def test_aggregate_monthly_basic(
 
         # Check February
         feb = result[1]
-        assert feb["month"] == "2024-02"
+        assert get_monthly_result_month(feb) == "2024-02"
         assert feb["input_tokens"] == 600  # 3 entries * 200
         assert feb["output_tokens"] == 300  # 3 entries * 100
         assert feb["total_cost"] == 0.006  # 3 entries * 0.002
@@ -397,7 +411,7 @@ def test_aggregate_monthly_with_date_filter(
 
         # Should only have February
         assert len(result) == 1
-        assert result[0]["month"] == "2024-02"
+        assert get_monthly_result_month(result[0]) == "2024-02"
 
     def test_aggregate_from_blocks_daily(
         self, aggregator: UsageAggregator, sample_entries: list[UsageEntry]
@@ -435,7 +449,7 @@ def test_aggregate_from_blocks_daily(
         result = aggregator.aggregate_from_blocks(blocks, "daily")
 
         assert len(result) >= 2  # At least 2 days of data
-        assert result[0]["date"] == "2024-01-01"
+        assert get_daily_result_date(result[0]) == "2024-01-01"
 
     def test_aggregate_from_blocks_monthly(
         self, aggregator: UsageAggregator, sample_entries: list[UsageEntry]
@@ -454,8 +468,8 @@ def test_aggregate_from_blocks_monthly(
         result = aggregator.aggregate_from_blocks([block], "monthly")
 
         assert len(result) == 2  # Jan and Feb
-        assert result[0]["month"] == "2024-01"
-        assert result[1]["month"] == "2024-02"
+        assert get_monthly_result_month(result[0]) == "2024-01"
+        assert get_monthly_result_month(result[1]) == "2024-02"
 
     def test_aggregate_from_blocks_invalid_view_type(
         self, aggregator: UsageAggregator
@@ -488,25 +502,31 @@ def test_calculate_totals_empty(self, aggregator: UsageAggregator) -> None:
 
     def test_calculate_totals_with_data(self, aggregator: UsageAggregator) -> None:
         """Test calculating totals with aggregated data."""
-        aggregated_data = [
-            {
-                "date": "2024-01-01",
-                "input_tokens": 1000,
-                "output_tokens": 500,
-                "cache_creation_tokens": 100,
-                "cache_read_tokens": 50,
-                "total_cost": 0.05,
-                "entries_count": 10,
-            },
-            {
-                "date": "2024-01-02",
-                "input_tokens": 2000,
-                "output_tokens": 1000,
-                "cache_creation_tokens": 200,
-                "cache_read_tokens": 100,
-                "total_cost": 0.10,
-                "entries_count": 20,
-            },
+        from claude_monitor.types import CompleteAggregatedUsage
+        
+        aggregated_data: list[CompleteAggregatedUsage] = [
+            CompleteAggregatedUsage(
+                date="2024-01-01",
+                input_tokens=1000,
+                output_tokens=500,
+                cache_creation_tokens=100,
+                cache_read_tokens=50,
+                total_cost=0.05,
+                entries_count=10,
+                models_used=[],
+                model_breakdowns={},
+            ),
+            CompleteAggregatedUsage(
+                date="2024-01-02",
+                input_tokens=2000,
+                output_tokens=1000,
+                cache_creation_tokens=200,
+                cache_read_tokens=100,
+                total_cost=0.10,
+                entries_count=20,
+                models_used=[],
+                model_breakdowns={},
+            ),
         ]
 
         result = aggregator.calculate_totals(aggregated_data)
@@ -573,9 +593,9 @@ def test_period_sorting(self, aggregator: UsageAggregator) -> None:
         # Test daily sorting
         daily_result = aggregator.aggregate_daily(entries)
         assert len(daily_result) == 3
-        assert daily_result[0]["date"] == "2024-01-01"
-        assert daily_result[1]["date"] == "2024-01-10"
-        assert daily_result[2]["date"] == "2024-01-15"
+        assert get_daily_result_date(daily_result[0]) == "2024-01-01"
+        assert get_daily_result_date(daily_result[1]) == "2024-01-10"
+        assert get_daily_result_date(daily_result[2]) == "2024-01-15"
 
         # Test monthly sorting
         monthly_entries = [
@@ -616,6 +636,6 @@ def test_period_sorting(self, aggregator: UsageAggregator) -> None:
 
         monthly_result = aggregator.aggregate_monthly(monthly_entries)
         assert len(monthly_result) == 3
-        assert monthly_result[0]["month"] == "2024-01"
-        assert monthly_result[1]["month"] == "2024-02"
-        assert monthly_result[2]["month"] == "2024-03"
+        assert get_monthly_result_month(monthly_result[0]) == "2024-01"
+        assert get_monthly_result_month(monthly_result[1]) == "2024-02"
+        assert get_monthly_result_month(monthly_result[2]) == "2024-03"