7
7
8
8
fake = Faker ()
9
9
10
- valid_user_role_data = {'role' : 'admin' }
11
- user_id = fake .random_int (1 , 9999 )
12
-
13
10
14
11
def test_users_response_contains_expected_props (
15
12
client : FlaskClient ,
@@ -34,32 +31,38 @@ def test_users_response_contains_expected_props(
34
31
def test_update_user_role_response_contains_expected_props (
35
32
client : FlaskClient ,
36
33
valid_header : dict ,
34
+ user_id : dict ,
37
35
):
36
+ valid_user_role_data = {'role' : 'admin' }
38
37
AzureConnection .update_user_role = Mock (
39
- return_value = [ {'name' : 'dummy' , 'email' : 'dummy' , 'role' : 'dummy' }]
38
+ return_value = {'name' : 'dummy' , 'email' : 'dummy' , 'role' : 'dummy' }
40
39
)
41
40
42
41
response = client .put (
43
42
f'/users/{ user_id } ' , headers = valid_header , json = valid_user_role_data
44
43
)
45
44
46
45
assert HTTPStatus .OK == response .status_code
47
- assert 'name' in json .loads (response .data )[ 0 ]
48
- assert 'email' in json .loads (response .data )[ 0 ]
49
- assert 'role' in json .loads (response .data )[ 0 ]
46
+ assert 'name' in json .loads (response .data )
47
+ assert 'email' in json .loads (response .data )
48
+ assert 'role' in json .loads (response .data )
50
49
51
50
52
- @patch ('utils.azure_users.AzureConnection' , new_callable = Mock )
51
+ @patch ('utils.azure_users.AzureConnection.update_user_role ' , new_callable = Mock )
53
52
def test_update_user_role_is_being_called_with_valid_arguments (
54
53
update_user_role_mock ,
55
54
client : FlaskClient ,
56
55
valid_header : dict ,
56
+ user_id : dict ,
57
57
):
58
58
59
+ valid_user_role_data = {'role' : 'admin' }
59
60
response = client .put (
60
61
f'/users/{ user_id } ' , headers = valid_header , json = valid_user_role_data
61
62
)
62
63
63
64
assert HTTPStatus .OK == response .status_code
64
65
assert valid_user_role_data ['role' ] == 'admin'
65
- update_user_role_mock .assert_called_once ()
66
+ update_user_role_mock .assert_called_once_with (
67
+ user_id , valid_user_role_data ['role' ]
68
+ )
0 commit comments