import json from datetime import datetime, timedelta from io import BytesIO from unittest.mock import patch from fittrackee.users.models import User from fittrackee.workouts.models import Sport, Workout from flask import Flask class TestGetUser: def test_it_gets_single_user_without_workouts( self, app: Flask, user_1: User, user_2: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( f'/api/users/{user_2.username}', content_type='application/json', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert data['status'] == 'success' assert len(data['data']['users']) == 1 user = data['data']['users'][0] assert user['username'] == 'toto' assert user['email'] == 'toto@toto.com' assert user['created_at'] assert not user['admin'] assert user['first_name'] is None assert user['last_name'] is None assert user['birth_date'] is None assert user['bio'] is None assert user['location'] is None assert user['timezone'] is None assert user['weekm'] is False assert user['language'] is None assert user['nb_sports'] == 0 assert user['nb_workouts'] == 0 assert user['sports_list'] == [] assert user['total_distance'] == 0 assert user['total_duration'] == '0:00:00' def test_it_gets_single_user_with_workouts( self, app: Flask, user_1: User, sport_1_cycling: Sport, sport_2_running: Sport, workout_cycling_user_1: Workout, workout_running_user_1: Workout, ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( f'/api/users/{user_1.username}', content_type='application/json', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert data['status'] == 'success' assert len(data['data']['users']) == 1 user = data['data']['users'][0] assert user['username'] == 'test' assert user['email'] == 'test@test.com' assert user['created_at'] assert not user['admin'] assert user['first_name'] is None assert user['last_name'] is None assert user['birth_date'] is None assert user['bio'] is None assert user['location'] is None assert user['timezone'] is None assert user['weekm'] is False assert user['language'] is None assert user['nb_sports'] == 2 assert user['nb_workouts'] == 2 assert user['sports_list'] == [1, 2] assert user['total_distance'] == 22 assert user['total_duration'] == '2:40:00' def test_it_returns_error_if_user_does_not_exist( self, app: Flask, user_1: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users/not_existing', content_type='application/json', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 404 assert 'not found' in data['status'] assert 'User does not exist.' in data['message'] class TestGetUsers: def test_it_get_users_list( self, app: Flask, user_1: User, user_2: User, user_3: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 3 assert 'created_at' in data['data']['users'][0] assert 'created_at' in data['data']['users'][1] assert 'created_at' in data['data']['users'][2] assert 'test' in data['data']['users'][0]['username'] assert 'toto' in data['data']['users'][1]['username'] assert 'sam' in data['data']['users'][2]['username'] assert 'test@test.com' in data['data']['users'][0]['email'] assert 'toto@toto.com' in data['data']['users'][1]['email'] assert 'sam@test.com' in data['data']['users'][2]['email'] assert data['data']['users'][0]['timezone'] is None assert data['data']['users'][0]['weekm'] is False assert data['data']['users'][0]['language'] is None assert data['data']['users'][0]['nb_sports'] == 0 assert data['data']['users'][0]['nb_workouts'] == 0 assert data['data']['users'][0]['sports_list'] == [] assert data['data']['users'][0]['total_distance'] == 0 assert data['data']['users'][0]['total_duration'] == '0:00:00' assert data['data']['users'][1]['timezone'] is None assert data['data']['users'][1]['weekm'] is False assert data['data']['users'][1]['language'] is None assert data['data']['users'][1]['nb_sports'] == 0 assert data['data']['users'][1]['nb_workouts'] == 0 assert data['data']['users'][1]['sports_list'] == [] assert data['data']['users'][1]['total_distance'] == 0 assert data['data']['users'][1]['total_duration'] == '0:00:00' assert data['data']['users'][2]['timezone'] is None assert data['data']['users'][2]['weekm'] is True assert data['data']['users'][2]['language'] is None assert data['data']['users'][2]['nb_sports'] == 0 assert data['data']['users'][2]['nb_workouts'] == 0 assert data['data']['users'][2]['sports_list'] == [] assert data['data']['users'][2]['total_distance'] == 0 assert data['data']['users'][2]['total_duration'] == '0:00:00' assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 1, 'total': 3, } def test_it_gets_users_list_with_workouts( self, app: Flask, user_1: User, user_2: User, user_3: User, sport_1_cycling: Sport, workout_cycling_user_1: Workout, sport_2_running: Sport, workout_running_user_1: Workout, workout_cycling_user_2: Workout, ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 3 assert 'created_at' in data['data']['users'][0] assert 'created_at' in data['data']['users'][1] assert 'created_at' in data['data']['users'][2] assert 'test' in data['data']['users'][0]['username'] assert 'toto' in data['data']['users'][1]['username'] assert 'sam' in data['data']['users'][2]['username'] assert 'test@test.com' in data['data']['users'][0]['email'] assert 'toto@toto.com' in data['data']['users'][1]['email'] assert 'sam@test.com' in data['data']['users'][2]['email'] assert data['data']['users'][0]['timezone'] is None assert data['data']['users'][0]['weekm'] is False assert data['data']['users'][0]['nb_sports'] == 2 assert data['data']['users'][0]['nb_workouts'] == 2 assert data['data']['users'][0]['sports_list'] == [1, 2] assert data['data']['users'][0]['total_distance'] == 22.0 assert data['data']['users'][0]['total_duration'] == '2:40:00' assert data['data']['users'][1]['timezone'] is None assert data['data']['users'][1]['weekm'] is False assert data['data']['users'][1]['nb_sports'] == 1 assert data['data']['users'][1]['nb_workouts'] == 1 assert data['data']['users'][1]['sports_list'] == [1] assert data['data']['users'][1]['total_distance'] == 15 assert data['data']['users'][1]['total_duration'] == '1:00:00' assert data['data']['users'][2]['timezone'] is None assert data['data']['users'][2]['weekm'] is True assert data['data']['users'][2]['nb_sports'] == 0 assert data['data']['users'][2]['nb_workouts'] == 0 assert data['data']['users'][2]['sports_list'] == [] assert data['data']['users'][2]['total_distance'] == 0 assert data['data']['users'][2]['total_duration'] == '0:00:00' assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 1, 'total': 3, } @patch('fittrackee.users.users.USER_PER_PAGE', 2) def test_it_gets_first_page_on_users_list( self, app: Flask, user_1: User, user_2: User, user_3: User, ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users?page=1', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 2 assert data['pagination'] == { 'has_next': True, 'has_prev': False, 'page': 1, 'pages': 2, 'total': 3, } @patch('fittrackee.users.users.USER_PER_PAGE', 2) def test_it_gets_next_page_on_users_list( self, app: Flask, user_1: User, user_2: User, user_3: User, ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users?page=2', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 1 assert data['pagination'] == { 'has_next': False, 'has_prev': True, 'page': 2, 'pages': 2, 'total': 3, } def test_it_gets_empty_next_page_on_users_list( self, app: Flask, user_1: User, user_2: User, user_3: User, ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users?page=2', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 0 assert data['pagination'] == { 'has_next': False, 'has_prev': True, 'page': 2, 'pages': 1, 'total': 3, } def test_it_gets_user_list_with_2_per_page( self, app: Flask, user_1: User, user_2: User, user_3: User, ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users?per_page=2', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 2 assert data['pagination'] == { 'has_next': True, 'has_prev': False, 'page': 1, 'pages': 2, 'total': 3, } def test_it_gets_next_page_on_user_list_with_2_per_page( self, app: Flask, user_1: User, user_2: User, user_3: User, ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users?page=2&per_page=2', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 1 assert data['pagination'] == { 'has_next': False, 'has_prev': True, 'page': 2, 'pages': 2, 'total': 3, } def test_it_gets_users_list_ordered_by_username( self, app: Flask, user_1: User, user_2: User, user_3: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users?order_by=username', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 3 assert 'sam' in data['data']['users'][0]['username'] assert 'test' in data['data']['users'][1]['username'] assert 'toto' in data['data']['users'][2]['username'] assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 1, 'total': 3, } def test_it_gets_users_list_ordered_by_username_ascending( self, app: Flask, user_1: User, user_2: User, user_3: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users?order_by=username&order=asc', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 3 assert 'sam' in data['data']['users'][0]['username'] assert 'test' in data['data']['users'][1]['username'] assert 'toto' in data['data']['users'][2]['username'] assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 1, 'total': 3, } def test_it_gets_users_list_ordered_by_username_descending( self, app: Flask, user_1: User, user_2: User, user_3: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users?order_by=username&order=desc', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 3 assert 'toto' in data['data']['users'][0]['username'] assert 'test' in data['data']['users'][1]['username'] assert 'sam' in data['data']['users'][2]['username'] assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 1, 'total': 3, } def test_it_gets_users_list_ordered_by_creation_date( self, app: Flask, user_2: User, user_3: User, user_1_admin: User ) -> None: user_2.created_at = datetime.utcnow() - timedelta(days=1) user_3.created_at = datetime.utcnow() - timedelta(hours=1) user_1_admin.created_at = datetime.utcnow() client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps( dict(email='admin@example.com', password='12345678') ), content_type='application/json', ) response = client.get( '/api/users?order_by=created_at', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 3 assert 'toto' in data['data']['users'][0]['username'] assert 'sam' in data['data']['users'][1]['username'] assert 'admin' in data['data']['users'][2]['username'] assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 1, 'total': 3, } def test_it_gets_users_list_ordered_by_creation_date_ascending( self, app: Flask, user_2: User, user_3: User, user_1_admin: User ) -> None: user_2.created_at = datetime.utcnow() - timedelta(days=1) user_3.created_at = datetime.utcnow() - timedelta(hours=1) user_1_admin.created_at = datetime.utcnow() client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps( dict(email='admin@example.com', password='12345678') ), content_type='application/json', ) response = client.get( '/api/users?order_by=created_at&order=asc', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 3 assert 'toto' in data['data']['users'][0]['username'] assert 'sam' in data['data']['users'][1]['username'] assert 'admin' in data['data']['users'][2]['username'] assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 1, 'total': 3, } def test_it_gets_users_list_ordered_by_creation_date_descending( self, app: Flask, user_2: User, user_3: User, user_1_admin: User ) -> None: user_2.created_at = datetime.utcnow() - timedelta(days=1) user_3.created_at = datetime.utcnow() - timedelta(hours=1) user_1_admin.created_at = datetime.utcnow() client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps( dict(email='admin@example.com', password='12345678') ), content_type='application/json', ) response = client.get( '/api/users?order_by=created_at&order=desc', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 3 assert 'admin' in data['data']['users'][0]['username'] assert 'sam' in data['data']['users'][1]['username'] assert 'toto' in data['data']['users'][2]['username'] assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 1, 'total': 3, } def test_it_gets_users_list_ordered_by_admin_rights( self, app: Flask, user_2: User, user_1_admin: User, user_3: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps( dict(email='admin@example.com', password='12345678') ), content_type='application/json', ) response = client.get( '/api/users?order_by=admin', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 3 assert 'toto' in data['data']['users'][0]['username'] assert 'sam' in data['data']['users'][1]['username'] assert 'admin' in data['data']['users'][2]['username'] assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 1, 'total': 3, } def test_it_gets_users_list_ordered_by_admin_rights_ascending( self, app: Flask, user_2: User, user_1_admin: User, user_3: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps( dict(email='admin@example.com', password='12345678') ), content_type='application/json', ) response = client.get( '/api/users?order_by=admin&order=asc', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 3 assert 'toto' in data['data']['users'][0]['username'] assert 'sam' in data['data']['users'][1]['username'] assert 'admin' in data['data']['users'][2]['username'] assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 1, 'total': 3, } def test_it_gets_users_list_ordered_by_admin_rights_descending( self, app: Flask, user_2: User, user_3: User, user_1_admin: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps( dict(email='admin@example.com', password='12345678') ), content_type='application/json', ) response = client.get( '/api/users?order_by=admin&order=desc', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 3 assert 'admin' in data['data']['users'][0]['username'] assert 'toto' in data['data']['users'][1]['username'] assert 'sam' in data['data']['users'][2]['username'] assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 1, 'total': 3, } def test_it_gets_users_list_ordered_by_workouts_count( self, app: Flask, user_1: User, user_2: User, user_3: User, sport_1_cycling: Sport, workout_cycling_user_2: Workout, ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users?order_by=workouts_count', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 3 assert 'test' in data['data']['users'][0]['username'] assert 0 == data['data']['users'][0]['nb_workouts'] assert 'sam' in data['data']['users'][1]['username'] assert 0 == data['data']['users'][1]['nb_workouts'] assert 'toto' in data['data']['users'][2]['username'] assert 1 == data['data']['users'][2]['nb_workouts'] assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 1, 'total': 3, } def test_it_gets_users_list_ordered_by_workouts_count_ascending( self, app: Flask, user_1: User, user_2: User, user_3: User, sport_1_cycling: Sport, workout_cycling_user_2: Workout, ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users?order_by=workouts_count&order=asc', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 3 assert 'test' in data['data']['users'][0]['username'] assert 0 == data['data']['users'][0]['nb_workouts'] assert 'sam' in data['data']['users'][1]['username'] assert 0 == data['data']['users'][1]['nb_workouts'] assert 'toto' in data['data']['users'][2]['username'] assert 1 == data['data']['users'][2]['nb_workouts'] assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 1, 'total': 3, } def test_it_gets_users_list_ordered_by_workouts_count_descending( self, app: Flask, user_1: User, user_2: User, user_3: User, sport_1_cycling: Sport, workout_cycling_user_2: Workout, ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users?order_by=workouts_count&order=desc', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 3 assert 'toto' in data['data']['users'][0]['username'] assert 1 == data['data']['users'][0]['nb_workouts'] assert 'test' in data['data']['users'][1]['username'] assert 0 == data['data']['users'][1]['nb_workouts'] assert 'sam' in data['data']['users'][2]['username'] assert 0 == data['data']['users'][2]['nb_workouts'] assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 1, 'total': 3, } def test_it_gets_users_list_filtering_on_username( self, app: Flask, user_1: User, user_2: User, user_3: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users?q=toto', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 1 assert 'toto' in data['data']['users'][0]['username'] assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 1, 'total': 1, } def test_it_returns_empty_users_list_filtering_on_username( self, app: Flask, user_1: User, user_2: User, user_3: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users?q=not_existing', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 0 assert data['pagination'] == { 'has_next': False, 'has_prev': False, 'page': 1, 'pages': 0, 'total': 0, } def test_it_users_list_with_complex_query( self, app: Flask, user_1: User, user_2: User, user_3: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.get( '/api/users?order_by=username&order=desc&page=2&per_page=2', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 1 assert 'sam' in data['data']['users'][0]['username'] assert data['pagination'] == { 'has_next': False, 'has_prev': True, 'page': 2, 'pages': 2, 'total': 3, } class TestGetUserPicture: def test_it_return_error_if_user_has_no_picture( self, app: Flask, user_1: User ) -> None: client = app.test_client() response = client.get(f'/api/users/{user_1.username}/picture') data = json.loads(response.data.decode()) assert response.status_code == 404 assert 'not found' in data['status'] assert 'No picture.' in data['message'] def test_it_returns_error_if_user_does_not_exist( self, app: Flask, user_1: User ) -> None: client = app.test_client() response = client.get('/api/users/not_existing/picture') data = json.loads(response.data.decode()) assert response.status_code == 404 assert 'not found' in data['status'] assert 'User does not exist.' in data['message'] class TestUpdateUser: def test_it_adds_admin_rights_to_a_user( self, app: Flask, user_1_admin: User, user_2: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps( dict(email='admin@example.com', password='12345678') ), content_type='application/json', ) response = client.patch( '/api/users/toto', content_type='application/json', data=json.dumps(dict(admin=True)), headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 1 user = data['data']['users'][0] assert user['email'] == 'toto@toto.com' assert user['admin'] is True def test_it_removes_admin_rights_to_a_user( self, app: Flask, user_1_admin: User, user_2: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps( dict(email='admin@example.com', password='12345678') ), content_type='application/json', ) response = client.patch( '/api/users/toto', content_type='application/json', data=json.dumps(dict(admin=False)), headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 200 assert 'success' in data['status'] assert len(data['data']['users']) == 1 user = data['data']['users'][0] assert user['email'] == 'toto@toto.com' assert user['admin'] is False def test_it_returns_error_if_payload_for_admin_rights_is_empty( self, app: Flask, user_1_admin: User, user_2: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps( dict(email='admin@example.com', password='12345678') ), content_type='application/json', ) response = client.patch( '/api/users/toto', content_type='application/json', data=json.dumps(dict()), headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['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_payload_for_admin_rights_is_invalid( self, app: Flask, user_1_admin: User, user_2: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps( dict(email='admin@example.com', password='12345678') ), content_type='application/json', ) response = client.patch( '/api/users/toto', content_type='application/json', data=json.dumps(dict(admin="")), headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 500 assert 'error' in data['status'] assert ( 'Error. Please try again or contact the administrator.' in data['message'] ) def test_it_returns_error_if_user_can_not_change_admin_rights( self, app: Flask, user_1: User, user_2: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.patch( '/api/users/toto', content_type='application/json', data=json.dumps(dict(admin=True)), headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 403 assert 'error' in data['status'] assert 'You do not have permissions.' in data['message'] class TestDeleteUser: def test_user_can_delete_its_own_account( self, app: Flask, user_1: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.delete( '/api/users/test', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) assert response.status_code == 204 def test_user_with_workout_can_delete_its_own_account( self, app: Flask, user_1: User, sport_1_cycling: Sport, gpx_file: str ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) client.post( '/api/workouts', data=dict( file=(BytesIO(str.encode(gpx_file)), 'example.gpx'), data='{"sport_id": 1}', ), headers=dict( content_type='multipart/form-data', Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'], ), ) response = client.delete( '/api/users/test', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) assert response.status_code == 204 def test_user_with_picture_can_delete_its_own_account( self, app: Flask, user_1: User, sport_1_cycling: Sport, gpx_file: str ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) client.post( '/api/auth/picture', data=dict(file=(BytesIO(b'avatar'), 'avatar.png')), headers=dict( content_type='multipart/form-data', authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'], ), ) response = client.delete( '/api/users/test', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) assert response.status_code == 204 def test_user_can_not_delete_another_user_account( self, app: Flask, user_1: User, user_2: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.delete( '/api/users/toto', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 403 assert 'error' in data['status'] assert 'You do not have permissions.' in data['message'] def test_it_returns_error_when_deleting_non_existing_user( self, app: Flask, user_1: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps(dict(email='test@test.com', password='12345678')), content_type='application/json', ) response = client.delete( '/api/users/not_existing', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 404 assert 'not found' in data['status'] assert 'User does not exist.' in data['message'] def test_admin_can_delete_another_user_account( self, app: Flask, user_1_admin: User, user_2: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps( dict(email='admin@example.com', password='12345678') ), content_type='application/json', ) response = client.delete( '/api/users/toto', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) assert response.status_code == 204 def test_admin_can_delete_its_own_account( self, app: Flask, user_1_admin: User, user_2_admin: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps( dict(email='admin@example.com', password='12345678') ), content_type='application/json', ) response = client.delete( '/api/users/admin', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) assert response.status_code == 204 def test_admin_can_not_delete_its_own_account_if_no_other_admin( self, app: Flask, user_1_admin: User, user_2: User ) -> None: client = app.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps( dict(email='admin@example.com', password='12345678') ), content_type='application/json', ) response = client.delete( '/api/users/admin', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) data = json.loads(response.data.decode()) assert response.status_code == 403 assert 'error' in data['status'] assert ( 'You can not delete your account, no other user has admin rights.' in data['message'] ) def test_it_enables_registration_on_user_delete( self, app_no_config: Flask, app_config: Flask, user_1_admin: User, user_2: User, user_3: User, ) -> None: app_config.max_users = 3 client = app_no_config.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps( dict(email='admin@example.com', password='12345678') ), content_type='application/json', ) client.delete( '/api/users/toto', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) response = client.post( '/api/auth/register', data=json.dumps( dict( username='justatest', email='test@test.com', password='12345678', password_conf='12345678', ) ), content_type='application/json', ) assert response.status_code == 201 def test_it_does_not_enable_registration_on_user_delete( self, app_no_config: Flask, app_config: Flask, user_1_admin: User, user_2: User, user_3: User, ) -> None: app_config.max_users = 2 client = app_no_config.test_client() resp_login = client.post( '/api/auth/login', data=json.dumps( dict(email='admin@example.com', password='12345678') ), content_type='application/json', ) client.delete( '/api/users/toto', headers=dict( Authorization='Bearer ' + json.loads(resp_login.data.decode())['auth_token'] ), ) response = client.post( '/api/auth/register', data=json.dumps( dict( username='justatest', email='test@test.com', password='12345678', password_conf='12345678', ) ), content_type='application/json', ) assert response.status_code == 403 data = json.loads(response.data.decode()) assert data['status'] == 'error' assert data['message'] == 'Error. Registration is disabled.'