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
|
# --------------------------------------------------------------------------------------------
# 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.dev.pipelines.runs_artifacts import run_artifact_download, run_artifact_upload, run_artifact_list
from azext_devops.dev.common.services import clear_connection_cache
from azext_devops.tests.utils.authentication import AuthenticatedTests
from azext_devops.devops_sdk.v5_0.build.build_client import BuildClient
from azext_devops.dev.common.const import ARTIFACTTOOL_PAT_ENVKEY
class TestPipelinesRunsArtifactsMethods(AuthenticatedTests):
_TEST_DEVOPS_ORGANIZATION = 'https://someorganization.visualstudio.com'
_TEST_DEVOPS_PROJECT = 'MyProject'
_TEST_PAT_TOKEN = 'Some_PAT_Token'
_TEST_ARTIFACT_NAME = 'artifactname'
_TEST_ARTIFACT_PATH = 'D:/Testpath'
def setUp(self):
self.authentication_setup()
self.authenticate()
self.get_client_patcher = patch('azext_devops.devops_sdk.connection.Connection.get_client')
self.get_artifacts_patcher = patch('azext_devops.devops_sdk.v5_0.build.build_client.BuildClient.get_artifacts')
self.run_artifacttool_patcher = patch('azext_devops.dev.common.artifacttool.ArtifactToolInvoker.run_artifacttool')
#start the patchers
self.mock_get_artifacts = self.get_artifacts_patcher.start()
self.mock_get_client = self.get_client_patcher.start()
self.mock_run_artifacttool = self.run_artifacttool_patcher.start()
# Set return values which will be same across tests
self.mock_get_client.return_value = BuildClient(base_url=self._TEST_DEVOPS_ORGANIZATION)
#clear connection cache before running each test
clear_connection_cache()
def tearDown(self):
patch.stopall()
def test_runs_artifacts_list(self):
# set return values
response = run_artifact_list(run_id=12345, organization=self._TEST_DEVOPS_ORGANIZATION,
project=self._TEST_DEVOPS_PROJECT, detect=None)
#assert
self.mock_get_artifacts.assert_called_once_with(build_id=12345, project=self._TEST_DEVOPS_PROJECT)
def test_runs_artifacts_download(self):
# set return values
response = run_artifact_download(
run_id=12345, artifact_name=self._TEST_ARTIFACT_NAME, path=self._TEST_ARTIFACT_PATH,
organization=self._TEST_DEVOPS_ORGANIZATION, project=self._TEST_DEVOPS_PROJECT, detect=None)
#assert
self.mock_run_artifacttool.assert_called_with(self._TEST_DEVOPS_ORGANIZATION,
[
'pipelineartifact',
'download',
'--service', self._TEST_DEVOPS_ORGANIZATION,
'--patvar', ARTIFACTTOOL_PAT_ENVKEY,
'--project', self._TEST_DEVOPS_PROJECT,
'--pipeline-id', 12345,
'--artifact-name', self._TEST_ARTIFACT_NAME,
'--path', self._TEST_ARTIFACT_PATH,
],
'Downloading')
def test_runs_artifacts_upload(self):
# set return values
response = run_artifact_upload(
run_id=12345, artifact_name=self._TEST_ARTIFACT_NAME, path=self._TEST_ARTIFACT_PATH,
organization=self._TEST_DEVOPS_ORGANIZATION, project=self._TEST_DEVOPS_PROJECT, detect=None)
#assert
self.mock_run_artifacttool.assert_called_with(self._TEST_DEVOPS_ORGANIZATION,
[
'pipelineartifact',
'publish',
'--service', self._TEST_DEVOPS_ORGANIZATION,
'--patvar', ARTIFACTTOOL_PAT_ENVKEY,
'--project', self._TEST_DEVOPS_PROJECT,
'--pipeline-id', 12345,
'--artifact-name', self._TEST_ARTIFACT_NAME,
'--path', self._TEST_ARTIFACT_PATH,
],
'Uploading')
if __name__ == '__main__':
unittest.main()
|
