Skip to content

Commit 3d5fe42

Browse files
committed
test: TT-156 add get_grups_and_users test
1 parent 956bfb5 commit 3d5fe42

File tree

2 files changed

+40
-0
lines changed

2 files changed

+40
-0
lines changed

tests/utils/azure_users_test.py

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -96,3 +96,40 @@ def test_is_user_in_group(
9696
azure_connection.is_user_in_group('user_id', payload_mock)
9797
== response_expected
9898
)
99+
100+
101+
@patch('utils.azure_users.AzureConnection.get_msal_client', Mock())
102+
@patch('utils.azure_users.AzureConnection.get_token', Mock())
103+
@patch('requests.get')
104+
def test_get_groups_and_users(get_mock):
105+
response_mock = Mock()
106+
response_mock.status_code = 200
107+
return_value = {
108+
'value': [
109+
{
110+
'displayName': 'test-group-1',
111+
'members': [
112+
{'objectId': 'user-id1'},
113+
{'objectId': 'user-id2'},
114+
],
115+
},
116+
{
117+
'displayName': 'test-group-2',
118+
'members': [
119+
{'objectId': 'user-id3'},
120+
{'objectId': 'user-id4'},
121+
],
122+
},
123+
]
124+
}
125+
response_mock.json = Mock(return_value=return_value)
126+
get_mock.return_value = response_mock
127+
128+
expected_result = [
129+
('test-group-1', ['user-id1', 'user-id2']),
130+
('test-group-2', ['user-id3', 'user-id4']),
131+
]
132+
133+
azure_connection = AzureConnection()
134+
135+
assert azure_connection.get_groups_and_users() == expected_result

utils/azure_users.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -203,6 +203,9 @@ def get_groups_and_users(self):
203203
item['displayName'],
204204
[member['objectId'] for member in item['members']],
205205
)
206+
# from pprint import pprint
207+
# pprint(response.json()['value'])
208+
print(1)
206209
result = list(map(parse_item, response.json()['value']))
207210
return result
208211

0 commit comments

Comments
 (0)