def take_action(self, parsed_args): column_headers = ('id', 'uuid', 'username', 'user_id', 'user_uuid', 'key', 'is_leader', 'provider', 'provider_id', 'provider_uuid', 'usage', 'quota_cpu', 'quota_memory', 'quota_storage', 'quota_floating_ip_count', 'quota_instance_count', 'quota_port_count', 'quota_snapshot_count', 'quota_storage_count') 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_identity(parsed_args.id) identity = () if data.ok: message = data.message identity = ( message['id'], message['uuid'], message['user']['username'], message['user']['id'], message['user']['uuid'], message['key'], message['is_leader'], message['provider']['name'], message['provider']['id'], message['provider']['uuid'], message['usage'], message['quota']['cpu'], message['quota']['memory'], message['quota']['storage'], message['quota']['floating_ip_count'], message['quota']['instance_count'], message['quota']['port_count'], message['quota']['snapshot_count'], message['quota']['storage_count'] ) return (column_headers, identity)
def take_action(self, parsed_args): 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) payload = { "name": parsed_args.name, "identity": parsed_args.identity, "size_alias": parsed_args.size_alias, "project": parsed_args.project, "allocation_source_id": parsed_args.allocation_source_id, "deploy": parsed_args.deploy, "scripts": [], "extra": {} } if parsed_args.image: # get the provider from the identity payload['source_alias'] = None id_data = api.get_identity(parsed_args.identity) if id_data.ok: provider_id = id_data.message['provider']['id'] self.log.debug( 'Found provider id {} for identity uuid {}'.format( provider_id, parsed_args.identity)) # get the provider machine from the latest image version v_data = api.get_image_versions(parsed_args.image) if v_data.ok: # get latest image version latest_version = v_data.message['results'][-1] machines = latest_version['machines'] machine_id = next((m['uuid'] for m in machines if m['provider']['id'] == provider_id), None) self.log.debug( 'Found machine id {} for latest image version {} and provider id {}' .format(machine_id, latest_version['name'], provider_id)) payload['source_alias'] = machine_id else: self.app.stdout.write( 'Error, instance not created. Could not lookup latest image version from image uuid supplied: {}' .format(parsed_args.image)) else: self.app.stdout.write( 'Error, instance not created. Could not lookup provider from identity uuid supplied: {}' .format(parsed_args.identity)) else: payload['source_alias'] = parsed_args.source_alias self.log.debug('INPUT: {}'.format(json.dumps(payload))) instance = () column_headers = ('id', 'uuid', 'name', 'username', 'allocation_source', 'image_id', 'image_version', 'launched', 'image_size', 'provider') if payload['source_alias']: data = api.create_instance(json.dumps(payload)) if data.ok: message = data.message launched = ts_to_isodate(message['start_date']) instance = (message['id'], message['uuid'], message['name'], message['user']['username'], message['allocation_source']['name'], message['image']['id'], message['version']['name'], launched, message['size']['name'], message['provider']['name']) else: self.app.stdout.write( 'Error, instance not created! Make sure to supply name, identity, project, size, source (or image), and allocation_source.' ) return (column_headers, instance)
def test_getting_identity_when_response_is_ok(self): api = AtmosphereAPI('token', base_url=self.mock_users_base_url) response = api.get_identity("32e3354c-03cc-40e1-8c33-02bc7f6be299") assert response.ok assert response.message['id'] == 1403 and response.message['quota'][ 'cpu'] == 132
def test_getting_identity_when_response_is_not_ok(self): api = AtmosphereAPI('token', base_url=self.mock_users_bad_base_url) response = api.get_identity("32e3354c-03cc-40e1-8c33-02bc7f6be299") assert not response.ok