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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
# --------------------------------------------------------------------------------------------
# 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, ANY
except ImportError:
# Attempt to load mock (works on Python version below 3.3)
from mock import patch, ANY
from knack.util import CLIError
from msrest.serialization import Model
from azext_devops.devops_sdk.v5_0.git.git_client import GitClient
from azext_devops.dev.common.services import clear_connection_cache
from azext_devops.dev.repos.ref import (list_refs, create_ref, delete_ref, lock_ref, unlock_ref)
from azext_devops.tests.utils.authentication import AuthenticatedTests
from azext_devops.tests.utils.helper import get_client_mock_helper, TEST_DEVOPS_ORG_URL
class MockRef(object):
def __init__(self, object_id):
self.object_id = object_id
class MockRefResponse(Model):
def __init__(self, success, custom_message):
self.success = success
self.custom_message = custom_message
class TestRefMethods(AuthenticatedTests):
def setUp(self):
self.authentication_setup()
self.authenticate()
self.get_refs_patcher = patch('azext_devops.devops_sdk.v5_0.git.git_client.GitClient.get_refs')
self.update_ref_patcher = patch('azext_devops.devops_sdk.v5_0.git.git_client.GitClient.update_ref')
self.update_refs_patcher = patch('azext_devops.devops_sdk.v5_0.git.git_client.GitClient.update_refs')
self.mock_get_refs = self.get_refs_patcher.start()
self.mock_update_ref = self.update_ref_patcher.start()
self.mock_update_refs = self.update_refs_patcher.start()
# Setup mocks for clients
self.get_client = patch('azext_devops.devops_sdk.connection.Connection.get_client', new=get_client_mock_helper)
self.mock_get_client = self.get_client.start()
# clear connection cache before running each test
clear_connection_cache()
def tearDown(self):
patch.stopall()
def test_list_refs(self):
response = list_refs(organization=TEST_DEVOPS_ORG_URL,
project='sample_project')
# assert
self.mock_get_refs.assert_called_once_with(filter=None,
project='sample_project',
repository_id=None)
def test_create_ref(self):
response = create_ref(name='sample_ref',
object_id='1234567890',
organization=TEST_DEVOPS_ORG_URL,
project='sample_project')
# assert
self.mock_update_refs.assert_called_once_with(project='sample_project',
ref_updates=ANY,
repository_id=None)
def test_create_ref2(self):
custom_exception = "Mock exception message"
mockResponse = []
mockResponse.append(MockRefResponse(False,custom_exception))
self.mock_update_refs.return_value = mockResponse
try:
response = create_ref(name='sample_ref',
object_id='1234567890',
organization=TEST_DEVOPS_ORG_URL,
project='sample_project')
self.fail('we should have received an error')
except CLIError as ex:
self.assertEqual(str(ex), custom_exception)
def test_lock_ref(self):
response = lock_ref(name='sample_ref',
organization=TEST_DEVOPS_ORG_URL,
project='sample_project')
# assert
self.mock_update_ref.assert_called_once_with(project='sample_project',
new_ref_info=ANY,
filter='sample_ref',
repository_id=None)
def test_unlock_ref(self):
response = unlock_ref(name='sample_ref',
organization=TEST_DEVOPS_ORG_URL,
project='sample_project')
# assert
self.mock_update_ref.assert_called_once_with(project='sample_project',
new_ref_info=ANY,
filter='sample_ref',
repository_id=None)
def test_delete_ref(self):
response = delete_ref(name='sample_ref',
object_id='1234567890',
organization=TEST_DEVOPS_ORG_URL,
project='sample_project')
# assert
self.mock_update_refs.assert_called_once_with(project='sample_project',
ref_updates=ANY,
repository_id=None)
def test_delete_ref_without_obj_id(self):
refs = []
refs.append(MockRef("0"))
self.mock_get_refs.return_value = refs;
response = delete_ref(name='sample_ref',
organization=TEST_DEVOPS_ORG_URL,
project='sample_project')
# assert
self.mock_update_refs.assert_called_once_with(project='sample_project',
ref_updates=ANY,
repository_id=None)
def test_delete_ref_without_obj_id_invalid_ref_name(self):
sample_invalid_ref = "sample_invalid_ref"
try:
response = delete_ref(name=sample_invalid_ref,
organization=TEST_DEVOPS_ORG_URL,
project='sample_project')
self.fail('we should have received an error')
except CLIError as ex:
self.assertEqual(str(ex), f'Failed to find object_id for ref {sample_invalid_ref}. Please provide object_id.')
if __name__ == '__main__':
unittest.main()
|
