def test_create_default_keypair_already_exists(self): request_body = {"keypair": {"is_default": "true"}} self.mox.StubOutWithMock(db, "keypair_get_all") db.keypair_get_all( IsA(context.RequestContext), GID, filters=IsA(dict))\ .AndReturn([{}]) self.mox.ReplayAll() url = '/v1/groups/' + GID + '/keypairs' req = get_request(url, 'POST', request_body) res = req.get_response(self.app) self.assertEqual(res.status_code, 400)
def index(self, req, gid): try: self._uuid_check(gid=gid) except exception.NotFound as e: raise webob.exc.HTTPNotFound(explanation=e.format_message()) filters = {} keypair_id = req.params.get("keypair_id") nova_keypair_id = req.params.get("nova_keypair_id") name = req.params.get("name") status = req.params.get("status") is_default = req.params.get("is_default") if keypair_id: filters["keypair_id"] = keypair_id if nova_keypair_id: filters["nova_keypair_id"] = nova_keypair_id if name: filters["display_name"] = name if status: filters["status"] = status if is_default: filters["is_default"] = is_default context = req.environ["rack.context"] keypair_list = db.keypair_get_all(context, gid, filters) return self._view_builder.index(keypair_list)
def _validate(context, body, gid): if not self.is_valid_body(body, 'keypair'): msg = _("Invalid request body") raise exception.InvalidInput(reason=msg) self._uuid_check(gid) values = body["keypair"] name = values.get("name") is_default = values.get("is_default") if is_default: try: is_default = strutils.bool_from_string(is_default, strict=True) keypairs = db.keypair_get_all(context, gid, filters={"is_default": True}) if keypairs: msg = _("Default keypair already exists in the " "group %s" % gid) raise exception.InvalidInput(reason=msg) except ValueError: msg = _("is_default must be a boolean") raise exception.InvalidInput(reason=msg) else: is_default = False valid_values = {} valid_values["gid"] = gid valid_values["display_name"] = name valid_values["is_default"] = is_default return valid_values
def test_create(self): request_body = { "keypair": { "name": "test_keypair", "is_default": "true", } } self.mox.StubOutWithMock(db, "keypair_get_all") self.mox.StubOutWithMock(manager.ResourceOperator, "keypair_create") self.mox.StubOutWithMock(uuid, 'uuid4') mock_id = "1234-5678" uuid.uuid4().AndReturn(mock_id) uuid.uuid4().AndReturn(mock_id) db.keypair_get_all(IsA(context.RequestContext), GID, filters=IsA(dict)).AndReturn([]) manager.ResourceOperator.keypair_create(IsA(context.RequestContext), IsA(unicode)).AndReturn({ "nova_keypair_id": "keypair-" + mock_id, "private_key": "private-key-1234" }) self.mox.ReplayAll() expect = { "keypair": { "keypair_id": mock_id, "nova_keypair_id": "keypair-" + mock_id, "user_id": "noauth", "project_id": "noauth", "gid": GID, "name": "test_keypair", "private_key": "private-key-1234", "is_default": True } } url = '/v1/groups/' + GID + '/keypairs' req = get_request(url, 'POST', request_body) res = req.get_response(self.app) body = jsonutils.loads(res.body) for key in body["keypair"]: self.assertEqual(expect["keypair"][key], body["keypair"][key]) self.assertEqual(res.status_code, 201)
def index(self, req, gid): try: self._uuid_check(gid=gid) except exception.NotFound as e: raise webob.exc.HTTPNotFound(explanation=e.format_message()) context = req.environ['rack.context'] keypair_list = db.keypair_get_all(context, gid) keypair_list = self.manager.keypair_list(context, keypair_list) return self._view_builder.index(keypair_list)
def test_keypair_get_all(self): gid = "12345678-1234-5678-9123-123456789012" self._create_group(gid) values = [ {"keypair_id": unicode(uuid.uuid4()), "display_name": "test_keypair1"}, {"keypair_id": unicode(uuid.uuid4()), "display_name": "test_keypair2"}, {"keypair_id": unicode(uuid.uuid4()), "display_name": "test_keypair3"}, ] keypairs = [self._create_keypair(gid, value) for value in values] expected_keypairs = db.keypair_get_all(self.user_ctxt, gid) self._assertEqualListsOfObjects(keypairs, expected_keypairs)
def delete(self, req, gid): def _validate(gid): if not uuidutils.is_uuid_like(gid): raise exception.GroupNotFound(gid=gid) try: _validate(gid) context = req.environ['rack.context'] keypairs = db.keypair_get_all(context, gid) if keypairs: raise exception.GroupInUse(gid=gid) securitygroups = db.securitygroup_get_all(context, gid) if securitygroups: raise exception.GroupInUse(gid=gid) networks = db.network_get_all(context, gid) if networks: raise exception.GroupInUse(gid=gid) processes = db.process_get_all(context, gid) if processes: raise exception.GroupInUse(gid=gid) db.group_delete(context, gid) except exception.NotFound as e: raise webob.exc.HTTPNotFound(explanation=e.format_message()) except exception.GroupInUse as e: raise webob.exc.HTTPConflict(explanation=e.format_message()) except Exception as e: LOG.warn(e) raise exception.GroupDeleteFailed()
def _validate(context, body, gid, is_proxy=False): proxy = db.process_get_all(context, gid, filters={"is_proxy": True}) if is_proxy: if len(proxy) > 0: msg = _("Proxy process already exists in the group %s" % gid) raise exception.InvalidInput(reason=msg) else: if len(proxy) != 1: msg = _("Proxy process does not exist in the group %s" % gid) raise webob.exc.HTTPBadRequest(explanation=msg) keyname = "proxy" if is_proxy else "process" if not self.is_valid_body(body, keyname): msg = _("Invalid request body") raise exception.InvalidInput(reason=msg) values = body[keyname] ppid = values.get("ppid") name = values.get("name") keypair_id = values.get("keypair_id") securitygroup_ids = values.get("securitygroup_ids") glance_image_id = values.get("glance_image_id") nova_flavor_id = values.get("nova_flavor_id") userdata = values.get("userdata") args = values.get("args") self._uuid_check(gid, ppid, keypair_id) pid = unicode(uuid.uuid4()) if not name: prefix = "proxy-" if is_proxy else "process-" name = prefix + pid if ppid: parent_process = db.process_get_by_pid(context, gid, ppid) nova_keypair_id = None if keypair_id: keypair = db.keypair_get_by_keypair_id(context, gid, keypair_id) nova_keypair_id = keypair["nova_keypair_id"] elif ppid: keypair_id = parent_process.get("keypair_id") if keypair_id: keypair = db.keypair_get_by_keypair_id( context, gid, keypair_id) nova_keypair_id = keypair["nova_keypair_id"] else: default_keypair = db.keypair_get_all( context, gid, filters={"is_default": True}) if default_keypair: keypair_id = default_keypair[0]["keypair_id"] nova_keypair_id = default_keypair[0]["nova_keypair_id"] if securitygroup_ids is not None and\ not isinstance(securitygroup_ids, list): msg = _("securitygroupids must be a list") raise exception.InvalidInput(reason=msg) elif securitygroup_ids: neutron_securitygroup_ids = [] for id in securitygroup_ids: self._uuid_check(securitygroup_id=id) securitygroup = db.securitygroup_get_by_securitygroup_id( context, gid, id) neutron_securitygroup_ids.append( securitygroup["neutron_securitygroup_id"]) elif ppid: securitygroups = parent_process.get("securitygroups") securitygroup_ids =\ [securitygroup["securitygroup_id"] for securitygroup in securitygroups] neutron_securitygroup_ids =\ [securitygroup["neutron_securitygroup_id"] for securitygroup in securitygroups] else: default_securitygroups = db.securitygroup_get_all( context, gid, filters={"is_default": True}) if default_securitygroups: securitygroup_ids =\ [securitygroup["securitygroup_id"] for securitygroup in default_securitygroups] neutron_securitygroup_ids =\ [securitygroup["neutron_securitygroup_id"] for securitygroup in default_securitygroups] else: msg = _("securitygroup_ids is required. Default \ securitygroup_ids are not registered.") raise exception.InvalidInput(reason=msg) if not glance_image_id and ppid: glance_image_id = parent_process.get("glance_image_id") if not nova_flavor_id and ppid: nova_flavor_id = parent_process.get("nova_flavor_id") if userdata: try: base64.b64decode(userdata) except TypeError: msg = _("userdadta must be a base64 encoded value.") raise exception.InvalidInput(reason=msg) networks = db.network_get_all(context, gid) if not networks: msg = _("Netwoks does not exist in the group %s" % gid) raise webob.exc.HTTPBadRequest(explanation=msg) network_ids =\ [network["network_id"] for network in networks] neutron_network_ids =\ [network["neutron_network_id"] for network in networks] nics = [] for id in neutron_network_ids: nics.append({"net-id": id}) if args is None: args = {} elif args is not None and\ not isinstance(args, dict): msg = _("args must be a dict.") raise exception.InvalidInput(reason=msg) else: for key in args.keys(): args[key] = str(args[key]) default_args = { "gid": gid, "pid": pid, } if ppid: default_args["ppid"] = ppid if is_proxy: default_args["rackapi_ip"] = cfg.CONF.my_ip default_args["os_username"] = cfg.CONF.os_username default_args["os_password"] = cfg.CONF.os_password default_args["os_tenant_name"] = cfg.CONF.os_tenant_name default_args["os_auth_url"] = cfg.CONF.os_auth_url default_args["os_region_name"] = cfg.CONF.os_region_name else: proxy_instance_id = proxy[0]["nova_instance_id"] default_args["proxy_ip"] = self.manager.get_process_address( context, proxy_instance_id) args.update(default_args) valid_values = {} valid_values["gid"] = gid valid_values["ppid"] = ppid valid_values["pid"] = pid valid_values["display_name"] = name valid_values["keypair_id"] = keypair_id valid_values["securitygroup_ids"] = securitygroup_ids valid_values["glance_image_id"] = glance_image_id valid_values["nova_flavor_id"] = nova_flavor_id valid_values["userdata"] = userdata valid_values["args"] = json.dumps(args) valid_values["is_proxy"] = True if is_proxy else False valid_values["network_ids"] = network_ids if is_proxy: ipc_endpoint = values.get("ipc_endpoint") shm_endpoint = values.get("shm_endpoint") fs_endpoint = values.get("fs_endpoint") if ipc_endpoint: utils.check_string_length(ipc_endpoint, 'ipc_endpoint', min_length=1, max_length=255) if shm_endpoint: utils.check_string_length(shm_endpoint, 'shm_endpoint', min_length=1, max_length=255) if fs_endpoint: utils.check_string_length(fs_endpoint, 'fs_endpoint', min_length=1, max_length=255) valid_values["ipc_endpoint"] = ipc_endpoint valid_values["shm_endpoint"] = shm_endpoint valid_values["fs_endpoint"] = fs_endpoint boot_values = {} boot_values["name"] = name boot_values["key_name"] = nova_keypair_id boot_values["security_groups"] = neutron_securitygroup_ids boot_values["image"] = glance_image_id boot_values["flavor"] = nova_flavor_id boot_values["userdata"] = userdata boot_values["meta"] = args boot_values["nics"] = nics return valid_values, boot_values