sync

tests/unit/test_fit_parser.py

@@ -1,42 +1,52 @@
+import io
 import pytest
 from unittest.mock import MagicMock, patch
-from src.core.file_parser import FitParser, WorkoutData, WorkoutMetadata, PowerData, HeartRateData, SpeedData, ElevationData
+from src.core.file_parser import (
+    FitParser,
+    WorkoutData,
+    WorkoutMetadata,
+    PowerData,
+    HeartRateData,
+    SpeedData,
+    ElevationData,
+)
 from datetime import datetime, timedelta
 import pandas as pd
 
+
 @pytest.fixture
 def mock_fit_file():
-    with patch('fitparse.FitFile') as mock_fit_file_class:
+    with patch("fitparse.FitFile") as mock_fit_file_class:
         mock_fit_file_instance = MagicMock()
         mock_fit_file_class.return_value = mock_fit_file_instance
 
         # Mocking get_messages to return some dummy records
         mock_record1 = MagicMock()
         mock_record1.as_dict.return_value = {
-            'timestamp': datetime(2023, 1, 1, 10, 0, 0),
-            'power': 150,
-            'heart_rate': 130,
-            'speed': 5.0,
-            'altitude': 100.0
+            "timestamp": datetime(2023, 1, 1, 10, 0, 0),
+            "power": 150,
+            "heart_rate": 130,
+            "speed": 5.0,
+            "altitude": 100.0,
         }
         mock_record2 = MagicMock()
         mock_record2.as_dict.return_value = {
-            'timestamp': datetime(2023, 1, 1, 10, 1, 0),
-            'power': 160,
-            'heart_rate': 135,
-            'speed': 5.5,
-            'altitude': 105.0
+            "timestamp": datetime(2023, 1, 1, 10, 1, 0),
+            "power": 160,
+            "heart_rate": 135,
+            "speed": 5.5,
+            "altitude": 105.0,
         }
         mock_fit_file_instance.get_messages.return_value = [mock_record1, mock_record2]
         yield mock_fit_file_class
 
-def test_fit_parser_initialization():
-    parser = FitParser("dummy.fit")
-    assert parser.file_path == "dummy.fit"
 
 def test_fit_parser_parse_method_returns_workout_data(mock_fit_file):
-    parser = FitParser("dummy.fit")
-    workout_data = parser.parse()
+    # Mock the FitFile constructor directly within the test
+    with patch('fitparse.FitFile') as MockFitFile:
+        MockFitFile.return_value = mock_fit_file.return_value # Use the mocked instance from the fixture
+        parser = FitParser()
+        workout_data = parser.parse(io.BytesIO(b"dummy content"))
 
     assert isinstance(workout_data, WorkoutData)
     assert isinstance(workout_data.metadata, WorkoutMetadata)
@@ -48,4 +58,4 @@ def test_fit_parser_parse_method_returns_workout_data(mock_fit_file):
     assert "speed" in workout_data.time_series_data.columns
     assert "altitude" in workout_data.time_series_data.columns
     assert workout_data.metadata.start_time == datetime(2023, 1, 1, 10, 0, 0)
-    assert workout_data.metadata.duration == timedelta(minutes=1)
+    assert workout_data.metadata.duration == timedelta(minutes=1)