1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
# --------------------------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
# --------------------------------------------------------------------------------------------
import unittest
try:
# Attempt to load mock (works on Python 3.3 and above)
from unittest.mock import patch
except ImportError:
# Attempt to load mock (works on Python version below 3.3)
from mock import patch
from azext_devops.devops_sdk.v5_0.member_entitlement_management.member_entitlement_management_client import (MemberEntitlementManagementClient)
from azext_devops.dev.team.user import (get_user_entitlements,
get_user_entitlement,
add_user_entitlement,
delete_user_entitlement,
update_user_entitlement)
from azext_devops.dev.common.services import clear_connection_cache
from azext_devops.tests.utils.authentication import AuthenticatedTests
class TestUserMethods(AuthenticatedTests):
_TEST_DEVOPS_ORGANIZATION = 'https://someorganization.visualstudio.com'
_TEST_PROJECT_NAME = 'sample_project'
_TOP_VALUE = 10
_SKIP_VALUE = 2
_TEST_USER_ID = 'adda517c-0398-42dc-b2a8-0d3f240757f9'
_USER_MGMT_CLIENT_LOCATION = 'azext_devops.devops_sdk.v5_0.member_entitlement_management.member_entitlement_management_client.MemberEntitlementManagementClient.'
def setUp(self):
self.authentication_setup()
self.get_client = patch('azext_devops.devops_sdk.connection.Connection.get_client')
self.get_patch_op_patcher = patch('azext_devops.dev.team.user._create_patch_operation')
self.list_user_patcher = patch(self._USER_MGMT_CLIENT_LOCATION + 'get_user_entitlements')
self.get_user_patcher = patch(self._USER_MGMT_CLIENT_LOCATION + 'get_user_entitlement')
self.add_user_patcher = patch(self._USER_MGMT_CLIENT_LOCATION + 'update_user_entitlements')
self.remove_user_patcher = patch(self._USER_MGMT_CLIENT_LOCATION + 'delete_user_entitlement')
self.update_user_patcher = patch(self._USER_MGMT_CLIENT_LOCATION + 'update_user_entitlement')
self.mock_get_client = self.get_client.start()
self.mock_get_users = self.list_user_patcher.start()
self.mock_add_user = self.add_user_patcher.start()
self.mock_get_user = self.get_user_patcher.start()
self.mock_remove_user = self.remove_user_patcher.start()
self.mock_update_user = self.update_user_patcher.start()
#set return values
self.mock_get_client.return_value = MemberEntitlementManagementClient(base_url=self._TEST_DEVOPS_ORGANIZATION)
#clear connection cache before running each test
clear_connection_cache()
def tearDown(self):
patch.stopall()
def test_list_user(self):
get_user_entitlements(top=self._TOP_VALUE, skip=self._SKIP_VALUE,
organization=self._TEST_DEVOPS_ORGANIZATION)
#assert
self.mock_get_users.assert_called_once()
list_user_param = self.mock_get_users.call_args_list[0][1]
self.assertEqual(10,list_user_param['top'])
self.assertEqual(2, list_user_param['skip'])
def test_show_user(self):
get_user_entitlement(user=self._TEST_USER_ID, organization=self._TEST_DEVOPS_ORGANIZATION)
#assert
self.mock_get_user.assert_called_once_with(user_id = 'adda517c-0398-42dc-b2a8-0d3f240757f9')
def test_add_user(self):
add_user_entitlement(email_id='someuser@xyz.com', license_type='stakeholder',
organization=self._TEST_DEVOPS_ORGANIZATION)
#assert
self.mock_add_user.assert_called_once()
add_user_param = self.mock_add_user.call_args_list[0][1]
add_user_param_document = add_user_param['document'][0].value
self.assertEqual(False, add_user_param['do_not_send_invite_for_new_users'])
self.assertEqual('stakeholder', add_user_param_document['accessLevel'].account_license_type)
self.assertEqual('user', add_user_param_document['user'].subject_kind)
self.assertEqual('someuser@xyz.com', add_user_param_document['user'].principal_name)
def test_remove_user(self):
delete_user_entitlement(user=self._TEST_USER_ID, organization= self._TEST_DEVOPS_ORGANIZATION)
#assert
self.mock_remove_user.assert_called_once()
def test_update_user(self):
update_user_entitlement(user=self._TEST_USER_ID, license_type='express',
organization=self._TEST_DEVOPS_ORGANIZATION)
#assert
self.mock_update_user.assert_called_once()
update_user_param = self.mock_update_user.call_args_list[0][1]
update_user_param_document = update_user_param['document'][0].value
print(update_user_param_document)
self.assertEqual('express', update_user_param_document['accountLicenseType'])
self.assertEqual('adda517c-0398-42dc-b2a8-0d3f240757f9', update_user_param['user_id'])
|
