Skip to content
Merged
Prev Previous commit
Next Next commit
test: TT-156 parametrize test for get_groups_by_user_id
  • Loading branch information
Angeluz-07 committed Feb 26, 2021
commit 8945be38cab89c2b865526edd27c5d499c7c24db
19 changes: 14 additions & 5 deletions tests/utils/azure_users_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,14 +140,23 @@ def test_get_groups_and_users(get_mock):
@patch('utils.azure_users.AzureConnection.get_msal_client', Mock())
@patch('utils.azure_users.AzureConnection.get_token', Mock())
@patch('utils.azure_users.AzureConnection.get_groups_and_users')
def test_get_groups_by_user_id(get_groups_and_users_mock):
@mark.parametrize(
'user_id,groups_expected_value',
[
('user-id1', ['test-group-1', 'test-group-2']),
('user-id2', ['test-group-1']),
('user-id3', ['test-group-2']),
('user-id4', []),
],
)
def test_get_groups_by_user_id(
get_groups_and_users_mock, user_id, groups_expected_value
):
get_groups_and_users_mock.return_value = [
('test-group-1', ['user-id1', 'user-id2']),
('test-group-2', ['user-id3', 'user-id1']),
]

azure_connection = AzureConnection()
user_id = 'user-id1'
expected_result = ['test-group-1', 'test-group-2']
groups = azure_connection.get_groups_by_user_id('user-id1')
assert groups == expected_result
groups = azure_connection.get_groups_by_user_id(user_id)
assert groups == groups_expected_value