| # Copyright 2012 OpenStack Foundation |
| # All Rights Reserved. |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); you may |
| # not use this file except in compliance with the License. You may obtain |
| # a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| # License for the specific language governing permissions and limitations |
| # under the License. |
| |
| import json |
| import urllib |
| |
| from tempest_lib import exceptions as lib_exc |
| |
| from tempest.api_schema.response.compute.v2_1 import floating_ips as schema |
| from tempest.common import service_client |
| |
| |
| class FloatingIPsClientJSON(service_client.ServiceClient): |
| |
| def list_floating_ips(self, params=None): |
| """Returns a list of all floating IPs filtered by any parameters.""" |
| url = 'os-floating-ips' |
| if params: |
| url += '?%s' % urllib.urlencode(params) |
| |
| resp, body = self.get(url) |
| body = json.loads(body) |
| self.validate_response(schema.list_floating_ips, resp, body) |
| return service_client.ResponseBodyList(resp, body['floating_ips']) |
| |
| def get_floating_ip_details(self, floating_ip_id): |
| """Get the details of a floating IP.""" |
| url = "os-floating-ips/%s" % str(floating_ip_id) |
| resp, body = self.get(url) |
| body = json.loads(body) |
| self.validate_response(schema.floating_ip, resp, body) |
| return service_client.ResponseBody(resp, body['floating_ip']) |
| |
| def create_floating_ip(self, pool_name=None): |
| """Allocate a floating IP to the project.""" |
| url = 'os-floating-ips' |
| post_body = {'pool': pool_name} |
| post_body = json.dumps(post_body) |
| resp, body = self.post(url, post_body) |
| body = json.loads(body) |
| self.validate_response(schema.floating_ip, resp, body) |
| return service_client.ResponseBody(resp, body['floating_ip']) |
| |
| def delete_floating_ip(self, floating_ip_id): |
| """Deletes the provided floating IP from the project.""" |
| url = "os-floating-ips/%s" % str(floating_ip_id) |
| resp, body = self.delete(url) |
| self.validate_response(schema.add_remove_floating_ip, resp, body) |
| return service_client.ResponseBody(resp, body) |
| |
| def associate_floating_ip_to_server(self, floating_ip, server_id): |
| """Associate the provided floating IP to a specific server.""" |
| url = "servers/%s/action" % str(server_id) |
| post_body = { |
| 'addFloatingIp': { |
| 'address': floating_ip, |
| } |
| } |
| |
| post_body = json.dumps(post_body) |
| resp, body = self.post(url, post_body) |
| self.validate_response(schema.add_remove_floating_ip, resp, body) |
| return service_client.ResponseBody(resp, body) |
| |
| def disassociate_floating_ip_from_server(self, floating_ip, server_id): |
| """Disassociate the provided floating IP from a specific server.""" |
| url = "servers/%s/action" % str(server_id) |
| post_body = { |
| 'removeFloatingIp': { |
| 'address': floating_ip, |
| } |
| } |
| |
| post_body = json.dumps(post_body) |
| resp, body = self.post(url, post_body) |
| self.validate_response(schema.add_remove_floating_ip, resp, body) |
| return service_client.ResponseBody(resp, body) |
| |
| def is_resource_deleted(self, id): |
| try: |
| self.get_floating_ip_details(id) |
| except lib_exc.NotFound: |
| return True |
| return False |
| |
| @property |
| def resource_type(self): |
| """Returns the primary type of resource this client works with.""" |
| return 'floating_ip' |
| |
| def list_floating_ip_pools(self, params=None): |
| """Returns a list of all floating IP Pools.""" |
| url = 'os-floating-ip-pools' |
| if params: |
| url += '?%s' % urllib.urlencode(params) |
| |
| resp, body = self.get(url) |
| body = json.loads(body) |
| self.validate_response(schema.floating_ip_pools, resp, body) |
| return service_client.ResponseBodyList(resp, body['floating_ip_pools']) |
| |
| def create_floating_ips_bulk(self, ip_range, pool, interface): |
| """Allocate floating IPs in bulk.""" |
| post_body = { |
| 'ip_range': ip_range, |
| 'pool': pool, |
| 'interface': interface |
| } |
| post_body = json.dumps({'floating_ips_bulk_create': post_body}) |
| resp, body = self.post('os-floating-ips-bulk', post_body) |
| body = json.loads(body) |
| self.validate_response(schema.create_floating_ips_bulk, resp, body) |
| return service_client.ResponseBody(resp, |
| body['floating_ips_bulk_create']) |
| |
| def list_floating_ips_bulk(self): |
| """Returns a list of all floating IPs bulk.""" |
| resp, body = self.get('os-floating-ips-bulk') |
| body = json.loads(body) |
| self.validate_response(schema.list_floating_ips_bulk, resp, body) |
| return service_client.ResponseBodyList(resp, body['floating_ip_info']) |
| |
| def delete_floating_ips_bulk(self, ip_range): |
| """Deletes the provided floating IPs bulk.""" |
| post_body = json.dumps({'ip_range': ip_range}) |
| resp, body = self.put('os-floating-ips-bulk/delete', post_body) |
| body = json.loads(body) |
| self.validate_response(schema.delete_floating_ips_bulk, resp, body) |
| data = body['floating_ips_bulk_delete'] |
| return service_client.ResponseBodyData(resp, data) |