API - refactor tests

fittrackee/tests/application/test_app_config_api.py

@@ -4,24 +4,18 @@ from flask import Flask
 
 from fittrackee.users.models import User
 
+from ..api_test_case import ApiTestCaseMixin
 
-class TestGetConfig:
+
+class TestGetConfig(ApiTestCaseMixin):
     def test_it_gets_application_config(
         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',
-        )
+        client, auth_token = self.get_test_client_and_auth_token(app)
 
         response = client.get(
             '/api/config',
-            headers=dict(
-                Authorization='Bearer '
-                + json.loads(resp_login.data.decode())['auth_token']
-            ),
+            headers=dict(Authorization=f'Bearer {auth_token}'),
         )
 
         data = json.loads(response.data.decode())
@@ -41,22 +35,14 @@ class TestGetConfig:
     def test_it_returns_error_if_application_has_no_config(
         self, app_no_config: Flask, user_1_admin: User
     ) -> None:
-        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, auth_token = self.get_test_client_and_auth_token(
+            app_no_config, as_admin=True
         )
 
         response = client.get(
             '/api/config',
             content_type='application/json',
-            headers=dict(
-                Authorization='Bearer '
-                + json.loads(resp_login.data.decode())['auth_token']
-            ),
+            headers=dict(Authorization=f'Bearer {auth_token}'),
         )
 
         data = json.loads(response.data.decode())
@@ -67,22 +53,14 @@ class TestGetConfig:
     def test_it_returns_error_if_application_has_several_config(
         self, app: Flask, app_config: Flask, 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',
+        client, auth_token = self.get_test_client_and_auth_token(
+            app, as_admin=True
         )
 
         response = client.get(
             '/api/config',
             content_type='application/json',
-            headers=dict(
-                Authorization='Bearer '
-                + json.loads(resp_login.data.decode())['auth_token']
-            ),
+            headers=dict(Authorization=f'Bearer {auth_token}'),
         )
 
         data = json.loads(response.data.decode())
@@ -91,26 +69,18 @@ class TestGetConfig:
         assert 'Error on getting configuration.' in data['message']
 
 
-class TestUpdateConfig:
+class TestUpdateConfig(ApiTestCaseMixin):
     def test_it_updates_config_when_user_is_admin(
         self, app: Flask, 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',
+        client, auth_token = self.get_test_client_and_auth_token(
+            app, as_admin=True
         )
         response = client.patch(
             '/api/config',
             content_type='application/json',
             data=json.dumps(dict(gpx_limit_import=100, max_users=10)),
-            headers=dict(
-                Authorization='Bearer '
-                + json.loads(resp_login.data.decode())['auth_token']
-            ),
+            headers=dict(Authorization=f'Bearer {auth_token}'),
         )
         data = json.loads(response.data.decode())
 
@@ -125,13 +95,8 @@ class TestUpdateConfig:
     def test_it_updates_all_config(
         self, app: Flask, 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',
+        client, auth_token = self.get_test_client_and_auth_token(
+            app, as_admin=True
         )
 
         response = client.patch(
@@ -145,10 +110,7 @@ class TestUpdateConfig:
                     max_users=50,
                 )
             ),
-            headers=dict(
-                Authorization='Bearer '
-                + json.loads(resp_login.data.decode())['auth_token']
-            ),
+            headers=dict(Authorization=f'Bearer {auth_token}'),
         )
 
         data = json.loads(response.data.decode())
@@ -163,21 +125,13 @@ class TestUpdateConfig:
     def test_it_returns_403_when_user_is_not_an_admin(
         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',
-        )
+        client, auth_token = self.get_test_client_and_auth_token(app)
 
         response = client.patch(
             '/api/config',
             content_type='application/json',
             data=json.dumps(dict(gpx_limit_import=100, max_users=10)),
-            headers=dict(
-                Authorization='Bearer '
-                + json.loads(resp_login.data.decode())['auth_token']
-            ),
+            headers=dict(Authorization=f'Bearer {auth_token}'),
         )
 
         data = json.loads(response.data.decode())
@@ -189,23 +143,15 @@ class TestUpdateConfig:
     def test_it_returns_400_if_invalid_is_payload(
         self, app: Flask, 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',
+        client, auth_token = self.get_test_client_and_auth_token(
+            app, as_admin=True
         )
 
         response = client.patch(
             '/api/config',
             content_type='application/json',
             data=json.dumps(dict()),
-            headers=dict(
-                Authorization='Bearer '
-                + json.loads(resp_login.data.decode())['auth_token']
-            ),
+            headers=dict(Authorization=f'Bearer {auth_token}'),
         )
 
         data = json.loads(response.data.decode())
@@ -216,23 +162,15 @@ class TestUpdateConfig:
     def test_it_returns_error_on_update_if_application_has_no_config(
         self, app_no_config: Flask, user_1_admin: User
     ) -> None:
-        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, auth_token = self.get_test_client_and_auth_token(
+            app_no_config, as_admin=True
         )
 
         response = client.patch(
             '/api/config',
             content_type='application/json',
             data=json.dumps(dict(gpx_limit_import=100, max_users=10)),
-            headers=dict(
-                Authorization='Bearer '
-                + json.loads(resp_login.data.decode())['auth_token']
-            ),
+            headers=dict(Authorization=f'Bearer {auth_token}'),
         )
 
         data = json.loads(response.data.decode())
@@ -243,13 +181,8 @@ class TestUpdateConfig:
     def test_it_raises_error_if_archive_max_size_is_below_files_max_size(
         self, app: Flask, 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',
+        client, auth_token = self.get_test_client_and_auth_token(
+            app, as_admin=True
         )
 
         response = client.patch(
@@ -263,10 +196,7 @@ class TestUpdateConfig:
                     max_users=50,
                 )
             ),
-            headers=dict(
-                Authorization='Bearer '
-                + json.loads(resp_login.data.decode())['auth_token']
-            ),
+            headers=dict(Authorization=f'Bearer {auth_token}'),
         )
 
         data = json.loads(response.data.decode())
@@ -280,13 +210,8 @@ class TestUpdateConfig:
     def test_it_raises_error_if_archive_max_size_equals_0(
         self, app_with_max_file_size_equals_0: Flask, user_1_admin: User
     ) -> None:
-        client = app_with_max_file_size_equals_0.test_client()
-        resp_login = client.post(
-            '/api/auth/login',
-            data=json.dumps(
-                dict(email='admin@example.com', password='12345678')
-            ),
-            content_type='application/json',
+        client, auth_token = self.get_test_client_and_auth_token(
+            app_with_max_file_size_equals_0, as_admin=True
         )
 
         response = client.patch(
@@ -297,10 +222,7 @@ class TestUpdateConfig:
                     max_zip_file_size=0,
                 )
             ),
-            headers=dict(
-                Authorization='Bearer '
-                + json.loads(resp_login.data.decode())['auth_token']
-            ),
+            headers=dict(Authorization=f'Bearer {auth_token}'),
         )
 
         data = json.loads(response.data.decode())
@@ -314,13 +236,8 @@ class TestUpdateConfig:
     def test_it_raises_error_if_files_max_size_equals_0(
         self, app: Flask, 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',
+        client, auth_token = self.get_test_client_and_auth_token(
+            app, as_admin=True
         )
 
         response = client.patch(
@@ -331,10 +248,7 @@ class TestUpdateConfig:
                     max_single_file_size=0,
                 )
             ),
-            headers=dict(
-                Authorization='Bearer '
-                + json.loads(resp_login.data.decode())['auth_token']
-            ),
+            headers=dict(Authorization=f'Bearer {auth_token}'),
         )
 
         data = json.loads(response.data.decode())
@@ -348,13 +262,8 @@ class TestUpdateConfig:
     def test_it_raises_error_if_gpx_limit_import_equals_0(
         self, app: Flask, 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',
+        client, auth_token = self.get_test_client_and_auth_token(
+            app, as_admin=True
         )
 
         response = client.patch(
@@ -365,10 +274,7 @@ class TestUpdateConfig:
                     gpx_limit_import=0,
                 )
             ),
-            headers=dict(
-                Authorization='Bearer '
-                + json.loads(resp_login.data.decode())['auth_token']
-            ),
+            headers=dict(Authorization=f'Bearer {auth_token}'),
         )
 
         data = json.loads(response.data.decode())