def take_action(self, parsed_args):
column_headers = ('id', 'uuid', 'name', 'description', 'created_by',
'versions', 'tags', 'url', 'is_public', 'start_date',
'end_date')
api = AtmosphereAPI(self.app_args.auth_token,
base_url=self.app_args.base_url,
timeout=self.app_args.api_server_timeout,
verify=self.app_args.verify_cert)
data = api.get_image(parsed_args.id)
image = ()
if data.ok:
message = data.message
versions = ''
if parsed_args.show_all_versions:
versions = '\n'.join([
'{} ({})'.format(value['name'], value['id'])
for value in message['versions']
])
else:
version_data = api.get_image_versions(parsed_args.id)
if version_data.ok:
versions = '\n'.join([
'{} ({})'.format(value['name'], value['id'])
for value in version_data.message['results']
])
start_date = ts_to_isodate(message['start_date'])
end_date = ''
if message['end_date']:
end_date = ts_to_isodate(message['end_date'])
image = (message['id'], message['uuid'], message['name'],
message['description'], message['created_by']['username'],
versions,
', '.join([value['name']
for value in message['tags']]), message['url'],
message['is_public'], start_date, end_date)
return (column_headers, image)
def test_getting_image_when_response_is_ok(self):
api = AtmosphereAPI('token', base_url=self.mock_users_base_url)
response = api.get_image(1)
assert response.ok
assert response.message['id'] == 55 and response.message['name'] == 'BioLinux 8'
def test_getting_image_when_response_is_not_ok(self):
api = AtmosphereAPI('token', base_url=self.mock_users_bad_base_url)
response = api.get_image(1)
assert not response.ok