API - get sports with authenticated user preferences

fittrackee/tests/workouts/test_sports_model.py

@@ -2,7 +2,8 @@ from typing import Dict, Optional
 
 from flask import Flask
 
-from fittrackee.users.models import User
+from fittrackee import db
+from fittrackee.users.models import User, UserSportPreference
 from fittrackee.workouts.models import Sport, Workout
 
 
@@ -15,10 +16,13 @@ class TestSportModel:
         assert 'Cycling' == sport.label
         assert '<Sport \'Cycling\'>' == str(sport)
 
-        serialized_sport = sport.serialize(is_admin)
+        serialized_sport = sport.serialize(is_admin=is_admin)
         assert 1 == serialized_sport['id']
         assert 'Cycling' == serialized_sport['label']
         assert serialized_sport['is_active'] is True
+        assert serialized_sport['is_active_for_user'] is True
+        assert serialized_sport['color'] is None
+        assert serialized_sport['stopped_speed_threshold'] == 1
         return serialized_sport
 
     def test_sport_model(self, app: Flask, sport_1_cycling: Sport) -> None:
@@ -44,3 +48,87 @@ class TestSportModel:
     ) -> None:
         serialized_sport = self.assert_sport_model(sport_1_cycling, True)
         assert serialized_sport['has_workouts'] is True
+
+
+class TestSportModelWithPreferences:
+    def test_sport_model_with_color_preference(
+        self,
+        app: Flask,
+        user_1: User,
+        sport_1_cycling: Sport,
+        user_sport_1_preference: UserSportPreference,
+    ) -> None:
+        user_sport_1_preference.color = '#00000'
+
+        serialized_sport = sport_1_cycling.serialize(
+            sport_preferences=user_sport_1_preference.serialize()
+        )
+        assert serialized_sport['id'] == 1
+        assert serialized_sport['label'] == 'Cycling'
+        assert serialized_sport['is_active'] is True
+        assert serialized_sport['is_active_for_user'] is True
+        assert serialized_sport['color'] == '#00000'
+        assert serialized_sport['stopped_speed_threshold'] == 1
+        assert 'has_workouts' not in serialized_sport
+
+    def test_sport_model_with_is_active_preference(
+        self,
+        app: Flask,
+        user_1: User,
+        sport_1_cycling: Sport,
+        user_sport_1_preference: UserSportPreference,
+    ) -> None:
+        user_sport_1_preference.is_active = False
+
+        serialized_sport = sport_1_cycling.serialize(
+            sport_preferences=user_sport_1_preference.serialize()
+        )
+        assert serialized_sport['id'] == 1
+        assert serialized_sport['label'] == 'Cycling'
+        assert serialized_sport['is_active'] is True
+        assert serialized_sport['is_active_for_user'] is False
+        assert serialized_sport['color'] is None
+        assert serialized_sport['stopped_speed_threshold'] == 1
+        assert 'has_workouts' not in serialized_sport
+
+    def test_inactive_sport_model_with_is_active_preference(
+        self,
+        app: Flask,
+        user_1: User,
+        sport_1_cycling: Sport,
+        user_sport_1_preference: UserSportPreference,
+    ) -> None:
+        sport_1_cycling.is_active = False
+        user_sport_1_preference.is_active = True
+
+        serialized_sport = sport_1_cycling.serialize(
+            sport_preferences=user_sport_1_preference.serialize()
+        )
+        assert serialized_sport['id'] == 1
+        assert serialized_sport['label'] == 'Cycling'
+        assert serialized_sport['is_active'] is False
+        assert serialized_sport['is_active_for_user'] is False
+        assert serialized_sport['color'] is None
+        assert serialized_sport['stopped_speed_threshold'] == 1
+        assert 'has_workouts' not in serialized_sport
+
+    def test_sport_model_with_threshold_preference(
+        self,
+        app: Flask,
+        user_1: User,
+        sport_1_cycling: Sport,
+        user_sport_1_preference: UserSportPreference,
+    ) -> None:
+        user_sport_1_preference.stopped_speed_threshold = 0.5
+        db.session.commit()
+
+        serialized_sport = sport_1_cycling.serialize(
+            sport_preferences=user_sport_1_preference.serialize()
+        )
+        assert serialized_sport['id'] == 1
+        assert serialized_sport['label'] == 'Cycling'
+        assert serialized_sport['is_active'] is True
+        assert serialized_sport['is_active_for_user'] is True
+        assert serialized_sport['color'] is None
+        assert serialized_sport['stopped_speed_threshold'] == 0.5
+        assert 'has_workouts' not in serialized_sport