|
6 | 6 | from pytest import mark |
7 | 7 |
|
8 | 8 |
|
| 9 | +@patch('msal.ConfidentialClientApplication', Mock()) |
| 10 | +@patch('utils.azure_users.AzureConnection.get_token', Mock()) |
| 11 | +@patch( |
| 12 | + 'utils.azure_users.AzureConnection.is_test_user', Mock(return_value=True) |
| 13 | +) |
9 | 14 | @patch( |
10 | 15 | 'commons.feature_toggles.feature_toggle_manager.FeatureToggleManager.get_azure_app_configuration_client' |
11 | 16 | ) |
@@ -183,46 +188,6 @@ def test_update_role_is_called_properly_on_each_action( |
183 | 188 | ) |
184 | 189 |
|
185 | 190 |
|
186 | | -@patch('msal.ConfidentialClientApplication') |
187 | | -@patch('utils.azure_users.AzureConnection.get_token') |
188 | | -@patch('utils.azure_users.AzureConnection._get_user') |
189 | | -@mark.parametrize( |
190 | | - 'field_name,field_value,expected', |
191 | | - [ |
192 | | - (ROLE_FIELD_VALUES['test'][0], ROLE_FIELD_VALUES['test'][1], True), |
193 | | - (ROLE_FIELD_VALUES['test'][0], None, False), |
194 | | - ], |
195 | | -) |
196 | | -def test_azure_connection_is_test_user( |
197 | | - _get_user_mock, |
198 | | - get_token_mock, |
199 | | - msal_client_mock, |
200 | | - field_name, |
201 | | - field_value, |
202 | | - expected, |
203 | | -): |
204 | | - _get_user_mock.return_value = {field_name: field_value} |
205 | | - test_user_id = 'test-user-id' |
206 | | - |
207 | | - az_conn = AzureConnection() |
208 | | - assert az_conn.is_test_user(test_user_id) == expected |
209 | | - |
210 | | - |
211 | | -@patch('msal.ConfidentialClientApplication') |
212 | | -@patch('utils.azure_users.AzureConnection.get_token') |
213 | | -@patch('utils.azure_users.AzureConnection._get_test_user_ids') |
214 | | -def test_azure_connection_get_test_user_ids( |
215 | | - _get_test_user_ids_mock, get_token_mock, msal_client_mock, |
216 | | -): |
217 | | - _get_test_user_ids_mock.return_value = [ |
218 | | - {'objectId': 'ID1'}, |
219 | | - {'objectId': 'ID2'}, |
220 | | - ] |
221 | | - ids = ['ID1', 'ID2'] |
222 | | - az_conn = AzureConnection() |
223 | | - assert az_conn.get_test_user_ids() == ids |
224 | | - |
225 | | - |
226 | 191 | @patch('msal.ConfidentialClientApplication') |
227 | 192 | @patch('utils.azure_users.AzureConnection.get_token') |
228 | 193 | @patch('utils.azure_users.AzureConnection.get_test_user_ids') |
|
0 commit comments