Fixed bug with deconstruction

tests/app/dao/test_services_get_specific_hours.py

@@ -1,5 +1,5 @@
+from collections import namedtuple
 from datetime import datetime
-from unittest.mock import Mock
 
 import pytest
 
@@ -7,15 +7,12 @@ from app.dao.services_dao import get_specific_hours_stats
 from app.enums import StatisticsType
 from app.models import TemplateType
 
+NotificationRow = namedtuple("NotificationRow", ["notification_type", "status", "timestamp", "count"])
+
 
 def generate_expected_hourly_output(requested_sms_hours):
-    """
-    Generates expected output only for hours where notifications exist.
-    Removes empty hours from the output to match function behavior.
-    """
-    output = {}
-    for hour in requested_sms_hours:
-        output[hour] = {
+    return {
+        hour: {
             TemplateType.SMS: {
                 StatisticsType.REQUESTED: 1,
                 StatisticsType.DELIVERED: 0,
@@ -29,23 +26,23 @@ def generate_expected_hourly_output(requested_sms_hours):
                 StatisticsType.PENDING: 0,
             },
         }
-    return output
+        for hour in requested_sms_hours
+    }
 
 
 def create_mock_notification(notification_type, status, timestamp, count=1):
     """
-    Creates a mock notification object with the required attributes.
+    Creates a named tuple with the attributes required by format_statistics.
     """
-    mock = Mock()
-    mock.notification_type = notification_type
-    mock.status = status
-    mock.timestamp = timestamp.replace(minute=0, second=0, microsecond=0)
-    mock.count = count
-    return mock
+    return NotificationRow(
+        notification_type=notification_type,
+        status=status,
+        timestamp=timestamp.replace(minute=0, second=0, microsecond=0),
+        count=count
+    )
 
 
 test_cases = [
-    # Single notification at 14:00 (Only 14:00 is expected in output)
     (
         [create_mock_notification(
             TemplateType.SMS,
@@ -54,11 +51,8 @@ test_cases = [
         )],
         datetime(2025, 2, 18, 12, 0),
         6,
-        generate_expected_hourly_output(
-            ["2025-02-18T14:00:00Z"]
-        ),
+        generate_expected_hourly_output(["2025-02-18T14:00:00Z"]),
     ),
-    # Notification at 17:59 (Only 17:00 is expected in output)
     (
         [create_mock_notification(
             TemplateType.SMS,
@@ -67,18 +61,9 @@ test_cases = [
         )],
         datetime(2025, 2, 18, 15, 0),
         3,
-        generate_expected_hourly_output(
-            ["2025-02-18T17:00:00Z"]
-        ),
+        generate_expected_hourly_output(["2025-02-18T17:00:00Z"]),
     ),
-    # No notifications at all (Expect empty `{}`)
-    (
-        [],
-        datetime(2025, 2, 18, 10, 0),
-        4,
-        {},
-    ),
-    # Two notifications at 09:00 and 11:00 (Only those hours expected)
+    ([], datetime(2025, 2, 18, 10, 0), 4, {}),
     (
         [
             create_mock_notification(TemplateType.SMS, StatisticsType.REQUESTED, datetime(2025, 2, 18, 9, 30, 0)),
@@ -86,25 +71,16 @@ test_cases = [
         ],
         datetime(2025, 2, 18, 8, 0),
         5,
-        generate_expected_hourly_output(
-            ["2025-02-18T09:00:00Z", "2025-02-18T11:00:00Z"]
-        ),
+        generate_expected_hourly_output(["2025-02-18T09:00:00Z", "2025-02-18T11:00:00Z"]),
     ),
 ]
 
 
-@pytest.mark.parametrize(
-    "mocked_notifications, start_date, hours, expected_output",
-    test_cases,
-)
+@pytest.mark.parametrize("mocked_notifications, start_date, hours, expected_output", test_cases)
 def test_get_specific_hours(mocked_notifications, start_date, hours, expected_output):
-    """
-    Tests get_specific_hours_stats to ensure it correctly aggregates hourly statistics.
-    """
     results = get_specific_hours_stats(
         mocked_notifications,
         start_date,
         hours=hours
     )
-
     assert results == expected_output, f"Expected {expected_output}, but got {results}"