def test_get_connection_cache_works_mulitple_organization(self):
with patch('azext_devops.dev.common.services._get_credentials') as mock_get_credentials:
with patch('azext_devops.dev.common.telemetry.try_send_telemetry_data') as mock_telemetry_send:
get_connection(self._TEST_DEVOPS_ORGANIZATION)
get_connection(self._TEST_DEVOPS_ORGANIZATION2)
#assert
self.assertEqual(2, mock_get_credentials.call_count)
self.assertEqual(2, mock_telemetry_send.call_count)
def test_get_connection_cache_works(self):
with patch('azext_devops.dev.common.services._get_credentials') as mock_get_credentials:
with patch('azext_devops.dev.common.telemetry.try_send_telemetry_data') as mock_telemetry_send:
get_connection(self._TEST_DEVOPS_ORGANIZATION)
get_connection(self._TEST_DEVOPS_ORGANIZATION.lower())
#assert
mock_telemetry_send.assert_called_once_with(self._TEST_DEVOPS_ORGANIZATION.lower())
mock_get_credentials.assert_called_once()
def _get_current_release(organization, override_version):
connection = get_connection(organization)
client = connection.get_client(
'azext_devops.dev.common.client_tool.client_tool_client.ClientToolClient'
)
logger.debug("Looking up current version of ArtifactTool...")
# Distro returns empty strings on Windows currently, so don't even send
distro_name = distro.id() or None
distro_version = distro.version() or None
os_name = platform.system()
arch = platform.machine()
# For M1 macs, there is no version of artifact tool. However, the x86_64
# version can run under Rosetta, so we use that instead.
if os_name == "Darwin" and arch in ["amd64", "arm64"]:
arch = "x86_64"
# Similarly for Windows ARM64 targets there is no version of artifact tool. However, the x86_64
# version can run under emulation, so we use that instead.
if os_name == "Windows" and arch == "ARM64":
arch = "x86_64"
release = client.get_clienttool_release("ArtifactTool",
os_name=os_name,
arch=arch,
distro_name=distro_name,
distro_version=distro_version,
version=override_version)
return (release.uri, _compute_id(release)) if release is not None else None
def _get_current_release(organization, override_version):
connection = get_connection(organization)
client = connection.get_client(
'azext_devops.dev.common.client_tool.client_tool_client.ClientToolClient'
)
logger.debug("Looking up current version of ArtifactTool...")
# Distro returns empty strings on Windows currently, so don't even send
distro_name = distro.id() or None
distro_version = distro.version() or None
release = client.get_clienttool_release("ArtifactTool",
os_name=platform.system(),
arch=platform.machine(),
distro_name=distro_name,
distro_version=distro_version,
version=override_version)
return (release.uri, _compute_id(release)) if release is not None else None
def invoke(area=None,
resource=None,
route_parameters=None,
query_parameters=None,
api_version='5.0',
http_method='GET',
in_file=None,
encoding='utf-8',
media_type='application/json',
accept_media_type='application/json',
out_file=None,
organization=None,
detect=None):
logger.info('route_parameter received is %s', route_parameters)
version = apiVersionToFloat(api_version)
organization = resolve_instance(detect=detect, organization=organization)
connection = get_connection(organization)
request_body = None
if in_file:
from os import path
if not path.exists(in_file):
raise CLIError('--in-file does not point to a valid file location')
from azext_devops.dev.common.utils import read_file_content
in_file_content = read_file_content(file_path=in_file,
encoding=encoding)
import json
request_body = json.loads(in_file_content)
resource_areas = connection._get_resource_areas(force=True)
if (not area and not resource):
print(
'Please wait a couple of seconds while we fetch all required information.'
)
service_list = []
for x in resource_areas:
if x.location_url not in service_list:
service_list.append(x.location_url)
resource_locations = []
for x in service_list:
try:
logger.info('trying to get locations from %s', x)
clientMock = Client(x, connection._creds)
resource_location_on_this_service = clientMock._get_resource_locations(
all_host_types=True)
resource_locations.extend(resource_location_on_this_service)
except: # pylint: disable=bare-except
logger.info('Failed to get location for %s', x)
return resource_locations
client_url = ''
if not resource_areas:
# this is for on-prem
client_url = connection.base_url
for resource_area in resource_areas:
if resource_area.name.lower() == area.lower():
client_url = resource_area.location_url
if not client_url:
raise CLIError('--area is not present in current organization')
client = Client(client_url, connection._creds)
# there can be multiple resouce/ area with different version so this version comparision is needed
location_id = ''
current_version = 0.0
resource_locations = client._get_resource_locations(all_host_types=True)
for resource_location in resource_locations:
if (resource.lower() == resource_location.resource_name.lower()
and area.lower() == resource_location.area.lower()):
current_maxVersion = float(resource_location.max_version)
if current_maxVersion > current_version and version >= current_version:
location_id = resource_location.id
current_version = current_maxVersion
if not location_id:
raise CLIError(
'--resource and --api-version combination is not correct')
route_values = stringToDict(route_parameters)
query_values = stringToDict(query_parameters)
response = client._send(http_method=http_method,
location_id=location_id,
version=api_version,
query_parameters=query_values,
route_values=route_values,
media_type=media_type,
accept_media_type=accept_media_type,
content=request_body)
logger.info('content type header')
logger.info(response.headers.get("content-type"))
if 'json' in response.headers.get("content-type") and not out_file:
return response.json()
if not out_file:
raise CLIError(
'Response is not json, you need to provide --out-file where it can be written'
)
import os
if os.path.exists(out_file):
raise CLIError('Out file already exists, please give a new name.')
open(out_file, "a").close()
with open(out_file, 'ab') as f:
for chunk in client._client.stream_download(response, callback=None):
f.write(chunk)