FitTrackee/fittrackee/tests/workouts/test_sports_model.py

from typing import Dict, Optional

from flask import Flask

from fittrackee import db
from fittrackee.users.models import User, UserSportPreference
from fittrackee.workouts.models import Sport, Workout


class TestSportModel:
    @staticmethod
    def assert_sport_model(
        sport: Sport, is_admin: Optional[bool] = False
    ) -> Dict:
        assert 1 == sport.id
        assert 'Cycling' == sport.label
        assert '<Sport \'Cycling\'>' == str(sport)

        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:
        serialized_sport = self.assert_sport_model(sport_1_cycling)
        assert 'has_workouts' not in serialized_sport

    def test_sport_model_with_workout(
        self,
        app: Flask,
        sport_1_cycling: Sport,
        user_1: User,
        workout_cycling_user_1: Workout,
    ) -> None:
        serialized_sport = self.assert_sport_model(sport_1_cycling)
        assert 'has_workouts' not in serialized_sport

    def test_sport_model_with_workout_as_admin(
        self,
        app: Flask,
        sport_1_cycling: Sport,
        user_1: User,
        workout_cycling_user_1: Workout,
    ) -> 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