API - workouts refactoring

fittrackee/tests/workouts/test_utils/test_gpx.py

@@ -0,0 +1,88 @@
+from unittest.mock import call, patch
+
+import pytest
+from flask import Flask
+from gpxpy.gpx import MovingData
+from werkzeug.datastructures import FileStorage
+
+from fittrackee.users.models import User, UserSportPreference
+from fittrackee.workouts.models import Sport
+from fittrackee.workouts.utils.workouts import process_files
+
+folders = {
+    'extract_dir': '/tmp/fitTrackee/uploads',
+    'tmp_dir': '/tmp/fitTrackee/uploads/tmp',
+}
+moving_data = MovingData(
+    moving_time=1,
+    stopped_time=1,
+    moving_distance=1,
+    stopped_distance=1,
+    max_speed=1,
+)
+
+
+class TestStoppedSpeedThreshold:
+    @pytest.mark.parametrize(
+        'sport_id, expected_threshold',
+        [(1, 1.0), (2, 0.1)],
+    )
+    def test_it_calls_get_moving_data_with_threshold_depending_on_sport(
+        self,
+        app: Flask,
+        user_1: User,
+        gpx_file_storage: FileStorage,
+        sport_1_cycling: Sport,
+        sport_2_running: Sport,
+        sport_id: int,
+        expected_threshold: float,
+    ) -> None:
+        with patch(
+            'fittrackee.workouts.utils.workouts.get_new_file_path',
+            return_value='/tmp/fitTrackee/uploads/test.png',
+        ), patch(
+            'gpxpy.gpx.GPXTrackSegment.get_moving_data',
+            return_value=moving_data,
+        ) as gpx_track_segment_mock:
+
+            process_files(
+                auth_user=user_1,
+                folders=folders,
+                workout_data={'sport_id': sport_id},
+                workout_file=gpx_file_storage,
+            )
+
+        assert gpx_track_segment_mock.call_args_list[0] == call(
+            stopped_speed_threshold=expected_threshold
+        )
+        gpx_track_segment_mock.assert_called_with(expected_threshold)
+
+    def test_it_calls_get_moving_data_with_threshold_depending_from_user_preference(  # noqa
+        self,
+        app: Flask,
+        user_1: User,
+        gpx_file_storage: FileStorage,
+        sport_1_cycling: Sport,
+        user_sport_1_preference: UserSportPreference,
+    ) -> None:
+        expected_threshold = 0.7
+        user_sport_1_preference.stopped_speed_threshold = expected_threshold
+        with patch(
+            'fittrackee.workouts.utils.workouts.get_new_file_path',
+            return_value='/tmp/fitTrackee/uploads/test.png',
+        ), patch(
+            'gpxpy.gpx.GPXTrackSegment.get_moving_data',
+            return_value=moving_data,
+        ) as gpx_track_segment_mock:
+
+            process_files(
+                auth_user=user_1,
+                folders=folders,
+                workout_data={'sport_id': sport_1_cycling.id},
+                workout_file=gpx_file_storage,
+            )
+
+        assert gpx_track_segment_mock.call_args_list[0] == call(
+            stopped_speed_threshold=expected_threshold
+        )
+        gpx_track_segment_mock.assert_called_with(expected_threshold)

fittrackee/tests/workouts/test_utils/test_workouts.py

@@ -0,0 +1,32 @@
+from statistics import mean
+from typing import List
+
+import pytest
+
+from fittrackee.workouts.utils.workouts import get_average_speed
+
+
+class TestWorkoutAverageSpeed:
+    @pytest.mark.parametrize(
+        'ave_speeds_list',
+        [
+            ([0]),
+            ([10]),
+            ([0, 0]),
+            ([10, 20]),
+            ([10, 0, 20, 10]),
+            ([1.5, 2, 3.7, 4.2]),
+        ],
+    )
+    def test_it_calculates_average_speed(self, ave_speeds_list: List) -> None:
+        nb_workouts = len(ave_speeds_list)
+        total_average_speed = (
+            sum(ave_speeds_list[:-1]) / (len(ave_speeds_list) - 1)
+            if len(ave_speeds_list) > 1
+            else ave_speeds_list[0]
+        )
+        workout_average_speed = ave_speeds_list[-1]
+
+        assert get_average_speed(
+            nb_workouts, total_average_speed, workout_average_speed
+        ) == mean(ave_speeds_list)