335 lines
11 KiB
Python
335 lines
11 KiB
Python
import json
|
|
|
|
from flask import Flask
|
|
|
|
from fittrackee.users.models import User
|
|
from fittrackee.workouts.models import Sport, Workout
|
|
|
|
from ..api_test_case import ApiTestCaseMixin
|
|
|
|
expected_sport_1_cycling_result = {
|
|
'id': 1,
|
|
'label': 'Cycling',
|
|
'is_active': True,
|
|
}
|
|
expected_sport_1_cycling_admin_result = expected_sport_1_cycling_result.copy()
|
|
expected_sport_1_cycling_admin_result['has_workouts'] = False
|
|
|
|
expected_sport_2_running_result = {
|
|
'id': 2,
|
|
'label': 'Running',
|
|
'is_active': True,
|
|
}
|
|
expected_sport_2_running_admin_result = expected_sport_2_running_result.copy()
|
|
expected_sport_2_running_admin_result['has_workouts'] = False
|
|
|
|
expected_sport_1_cycling_inactive_result = {
|
|
'id': 1,
|
|
'label': 'Cycling',
|
|
'is_active': False,
|
|
}
|
|
expected_sport_1_cycling_inactive_admin_result = (
|
|
expected_sport_1_cycling_inactive_result.copy()
|
|
)
|
|
expected_sport_1_cycling_inactive_admin_result['has_workouts'] = False
|
|
|
|
|
|
class TestGetSports(ApiTestCaseMixin):
|
|
def test_it_gets_all_sports(
|
|
self,
|
|
app: Flask,
|
|
user_1: User,
|
|
sport_1_cycling: Sport,
|
|
sport_2_running: Sport,
|
|
) -> None:
|
|
client, auth_token = self.get_test_client_and_auth_token(app)
|
|
|
|
response = client.get(
|
|
'/api/sports',
|
|
headers=dict(Authorization=f'Bearer {auth_token}'),
|
|
)
|
|
|
|
data = json.loads(response.data.decode())
|
|
assert response.status_code == 200
|
|
assert 'success' in data['status']
|
|
assert len(data['data']['sports']) == 2
|
|
assert data['data']['sports'][0] == expected_sport_1_cycling_result
|
|
assert data['data']['sports'][1] == expected_sport_2_running_result
|
|
|
|
def test_it_gets_all_sports_with_inactive_one(
|
|
self,
|
|
app: Flask,
|
|
user_1: User,
|
|
sport_1_cycling_inactive: Sport,
|
|
sport_2_running: Sport,
|
|
) -> None:
|
|
client, auth_token = self.get_test_client_and_auth_token(app)
|
|
|
|
response = client.get(
|
|
'/api/sports',
|
|
headers=dict(Authorization=f'Bearer {auth_token}'),
|
|
)
|
|
|
|
data = json.loads(response.data.decode())
|
|
assert response.status_code == 200
|
|
assert 'success' in data['status']
|
|
assert len(data['data']['sports']) == 2
|
|
assert (
|
|
data['data']['sports'][0]
|
|
== expected_sport_1_cycling_inactive_result
|
|
)
|
|
assert data['data']['sports'][1] == expected_sport_2_running_result
|
|
|
|
def test_it_gets_all_sports_with_admin_rights(
|
|
self,
|
|
app: Flask,
|
|
user_1_admin: User,
|
|
sport_1_cycling_inactive: Sport,
|
|
sport_2_running: Sport,
|
|
) -> None:
|
|
client, auth_token = self.get_test_client_and_auth_token(
|
|
app, as_admin=True
|
|
)
|
|
|
|
response = client.get(
|
|
'/api/sports',
|
|
headers=dict(Authorization=f'Bearer {auth_token}'),
|
|
)
|
|
|
|
data = json.loads(response.data.decode())
|
|
assert response.status_code == 200
|
|
assert 'success' in data['status']
|
|
assert len(data['data']['sports']) == 2
|
|
assert (
|
|
data['data']['sports'][0]
|
|
== expected_sport_1_cycling_inactive_admin_result
|
|
)
|
|
assert (
|
|
data['data']['sports'][1] == expected_sport_2_running_admin_result
|
|
)
|
|
|
|
|
|
class TestGetSport(ApiTestCaseMixin):
|
|
def test_it_gets_a_sport(
|
|
self, app: Flask, user_1: User, sport_1_cycling: Sport
|
|
) -> None:
|
|
client, auth_token = self.get_test_client_and_auth_token(app)
|
|
|
|
response = client.get(
|
|
'/api/sports/1',
|
|
headers=dict(Authorization=f'Bearer {auth_token}'),
|
|
)
|
|
|
|
data = json.loads(response.data.decode())
|
|
assert response.status_code == 200
|
|
assert 'success' in data['status']
|
|
assert len(data['data']['sports']) == 1
|
|
assert data['data']['sports'][0] == expected_sport_1_cycling_result
|
|
|
|
def test_it_returns_404_if_sport_does_not_exist(
|
|
self, app: Flask, user_1: User
|
|
) -> None:
|
|
client, auth_token = self.get_test_client_and_auth_token(app)
|
|
|
|
response = client.get(
|
|
'/api/sports/1',
|
|
headers=dict(Authorization=f'Bearer {auth_token}'),
|
|
)
|
|
|
|
data = json.loads(response.data.decode())
|
|
assert response.status_code == 404
|
|
assert 'not found' in data['status']
|
|
assert len(data['data']['sports']) == 0
|
|
|
|
def test_it_gets_a_inactive_sport(
|
|
self, app: Flask, user_1: User, sport_1_cycling_inactive: Sport
|
|
) -> None:
|
|
client, auth_token = self.get_test_client_and_auth_token(app)
|
|
|
|
response = client.get(
|
|
'/api/sports/1',
|
|
headers=dict(Authorization=f'Bearer {auth_token}'),
|
|
)
|
|
|
|
data = json.loads(response.data.decode())
|
|
assert response.status_code == 200
|
|
assert 'success' in data['status']
|
|
assert len(data['data']['sports']) == 1
|
|
assert (
|
|
data['data']['sports'][0]
|
|
== expected_sport_1_cycling_inactive_result
|
|
)
|
|
|
|
def test_it_get_an_inactive_sport_with_admin_rights(
|
|
self, app: Flask, user_1_admin: User, sport_1_cycling_inactive: Sport
|
|
) -> None:
|
|
client, auth_token = self.get_test_client_and_auth_token(
|
|
app, as_admin=True
|
|
)
|
|
|
|
response = client.get(
|
|
'/api/sports/1',
|
|
headers=dict(Authorization=f'Bearer {auth_token}'),
|
|
)
|
|
|
|
data = json.loads(response.data.decode())
|
|
assert response.status_code == 200
|
|
assert 'success' in data['status']
|
|
assert len(data['data']['sports']) == 1
|
|
assert (
|
|
data['data']['sports'][0]
|
|
== expected_sport_1_cycling_inactive_admin_result
|
|
)
|
|
|
|
|
|
class TestUpdateSport(ApiTestCaseMixin):
|
|
def test_it_disables_a_sport(
|
|
self, app: Flask, user_1_admin: User, sport_1_cycling: Sport
|
|
) -> None:
|
|
client, auth_token = self.get_test_client_and_auth_token(
|
|
app, as_admin=True
|
|
)
|
|
|
|
response = client.patch(
|
|
'/api/sports/1',
|
|
content_type='application/json',
|
|
data=json.dumps(dict(is_active=False)),
|
|
headers=dict(Authorization=f'Bearer {auth_token}'),
|
|
)
|
|
|
|
data = json.loads(response.data.decode())
|
|
assert response.status_code == 200
|
|
assert 'success' in data['status']
|
|
assert len(data['data']['sports']) == 1
|
|
assert data['data']['sports'][0]['is_active'] is False
|
|
assert data['data']['sports'][0]['has_workouts'] is False
|
|
|
|
def test_it_enables_a_sport(
|
|
self, app: Flask, user_1_admin: User, sport_1_cycling: Sport
|
|
) -> None:
|
|
sport_1_cycling.is_active = False
|
|
client, auth_token = self.get_test_client_and_auth_token(
|
|
app, as_admin=True
|
|
)
|
|
|
|
response = client.patch(
|
|
'/api/sports/1',
|
|
content_type='application/json',
|
|
data=json.dumps(dict(is_active=True)),
|
|
headers=dict(Authorization=f'Bearer {auth_token}'),
|
|
)
|
|
|
|
data = json.loads(response.data.decode())
|
|
assert response.status_code == 200
|
|
assert 'success' in data['status']
|
|
assert len(data['data']['sports']) == 1
|
|
assert data['data']['sports'][0]['is_active'] is True
|
|
assert data['data']['sports'][0]['has_workouts'] is False
|
|
|
|
def test_it_disables_a_sport_with_workouts(
|
|
self,
|
|
app: Flask,
|
|
user_1_admin: User,
|
|
sport_1_cycling: Sport,
|
|
workout_cycling_user_1: Workout,
|
|
) -> None:
|
|
client, auth_token = self.get_test_client_and_auth_token(
|
|
app, as_admin=True
|
|
)
|
|
|
|
response = client.patch(
|
|
'/api/sports/1',
|
|
content_type='application/json',
|
|
data=json.dumps(dict(is_active=False)),
|
|
headers=dict(Authorization=f'Bearer {auth_token}'),
|
|
)
|
|
|
|
data = json.loads(response.data.decode())
|
|
assert response.status_code == 200
|
|
assert 'success' in data['status']
|
|
assert len(data['data']['sports']) == 1
|
|
assert data['data']['sports'][0]['is_active'] is False
|
|
assert data['data']['sports'][0]['has_workouts'] is True
|
|
|
|
def test_it_enables_a_sport_with_workouts(
|
|
self,
|
|
app: Flask,
|
|
user_1_admin: User,
|
|
sport_1_cycling: Sport,
|
|
workout_cycling_user_1: Workout,
|
|
) -> None:
|
|
sport_1_cycling.is_active = False
|
|
client, auth_token = self.get_test_client_and_auth_token(
|
|
app, as_admin=True
|
|
)
|
|
|
|
response = client.patch(
|
|
'/api/sports/1',
|
|
content_type='application/json',
|
|
data=json.dumps(dict(is_active=True)),
|
|
headers=dict(Authorization=f'Bearer {auth_token}'),
|
|
)
|
|
|
|
data = json.loads(response.data.decode())
|
|
assert response.status_code == 200
|
|
assert 'success' in data['status']
|
|
assert len(data['data']['sports']) == 1
|
|
assert data['data']['sports'][0]['is_active'] is True
|
|
assert data['data']['sports'][0]['has_workouts'] is True
|
|
|
|
def test_returns_error_if_user_has_no_admin_rights(
|
|
self, app: Flask, user_1: User, sport_1_cycling: Sport
|
|
) -> None:
|
|
client, auth_token = self.get_test_client_and_auth_token(app)
|
|
|
|
response = client.patch(
|
|
'/api/sports/1',
|
|
content_type='application/json',
|
|
data=json.dumps(dict(is_active=False)),
|
|
headers=dict(Authorization=f'Bearer {auth_token}'),
|
|
)
|
|
|
|
data = json.loads(response.data.decode())
|
|
assert response.status_code == 403
|
|
assert 'success' not in data['status']
|
|
assert 'error' in data['status']
|
|
assert 'you do not have permissions' in data['message']
|
|
|
|
def test_returns_error_if_payload_is_invalid(
|
|
self, app: Flask, user_1_admin: User
|
|
) -> None:
|
|
client, auth_token = self.get_test_client_and_auth_token(
|
|
app, as_admin=True
|
|
)
|
|
|
|
response = client.patch(
|
|
'/api/sports/1',
|
|
content_type='application/json',
|
|
data=json.dumps(dict()),
|
|
headers=dict(Authorization=f'Bearer {auth_token}'),
|
|
)
|
|
|
|
data = json.loads(response.data.decode())
|
|
assert response.status_code == 400
|
|
assert 'error' in data['status']
|
|
assert 'invalid payload' in data['message']
|
|
|
|
def test_it_returns_error_if_sport_does_not_exist(
|
|
self, app: Flask, user_1_admin: User
|
|
) -> None:
|
|
client, auth_token = self.get_test_client_and_auth_token(
|
|
app, as_admin=True
|
|
)
|
|
|
|
response = client.patch(
|
|
'/api/sports/1',
|
|
content_type='application/json',
|
|
data=json.dumps(dict(is_active=False)),
|
|
headers=dict(Authorization=f'Bearer {auth_token}'),
|
|
)
|
|
|
|
data = json.loads(response.data.decode())
|
|
assert response.status_code == 404
|
|
assert 'not found' in data['status']
|
|
assert len(data['data']['sports']) == 0
|