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