feat: testing for update user role #229

tests/time_tracker_api/users/users_namespace_test.py

@@ -1,8 +1,14 @@
-from unittest.mock import Mock
+from unittest.mock import Mock, patch
 from flask import json
 from flask.testing import FlaskClient
 from flask_restplus._http import HTTPStatus
 from utils.azure_users import AzureConnection
+from faker import Faker
+
+fake = Faker()
+
+valid_user_role_data = {'role': 'admin'}
+user_id = fake.random_int(1, 9999)
 
 
 def test_users_response_contains_expected_props(
@@ -23,3 +29,37 @@ def test_users_response_contains_expected_props(
     assert 'name' in json.loads(response.data)[0]
     assert 'email' in json.loads(response.data)[0]
     assert 'role' in json.loads(response.data)[0]
+
+
+def test_update_user_role_response_contains_expected_props(
+    client: FlaskClient,
+    valid_header: dict,
+):
+    AzureConnection.update_user_role = Mock(
+        return_value=[{'name': 'dummy', 'email': 'dummy', 'role': 'dummy'}]
+    )
+
+    response = client.put(
+        f'/users/{user_id}', headers=valid_header, json=valid_user_role_data
+    )
+
+    assert HTTPStatus.OK == response.status_code
+    assert 'name' in json.loads(response.data)[0]
+    assert 'email' in json.loads(response.data)[0]
+    assert 'role' in json.loads(response.data)[0]
+
+
+@patch('utils.azure_users.AzureConnection', new_callable=Mock)
+def test_update_user_role_is_being_called_with_valid_arguments(
+    update_user_role_mock,
+    client: FlaskClient,
+    valid_header: dict,
+):
+
+    response = client.put(
+        f'/users/{user_id}', headers=valid_header, json=valid_user_role_data
+    )
+
+    assert HTTPStatus.OK == response.status_code
+    assert valid_user_role_data['role'] == 'admin'
+    update_user_role_mock.assert_called_once()