| from tempest.common import rest_client |
| from tempest import exceptions |
| import json |
| import time |
| |
| |
| class ImagesClient(object): |
| |
| def __init__(self, config, username, key, auth_url, tenant_name=None): |
| self.config = config |
| catalog_name = self.config.nova.catalog_name |
| self.client = rest_client.RestClient(config, username, key, |
| auth_url, catalog_name, |
| tenant_name) |
| |
| self.build_interval = self.config.nova.build_interval |
| self.build_timeout = self.config.nova.build_timeout |
| self.headers = {'Content-Type': 'application/json', |
| 'Accept': 'application/json'} |
| |
| def create_image(self, server_id, name, meta=None): |
| """Creates an image of the original server""" |
| |
| post_body = { |
| 'createImage': { |
| 'name': name, |
| } |
| } |
| |
| if meta != None: |
| post_body['createImage']['metadata'] = meta |
| |
| post_body = json.dumps(post_body) |
| resp, body = self.client.post('servers/%s/action' % |
| str(server_id), post_body, self.headers) |
| return resp, body |
| |
| def list_images(self, params=None): |
| """Returns a list of all images filtered by any parameters""" |
| url = 'images' |
| if params != None: |
| param_list = [] |
| for param, value in params.iteritems(): |
| param_list.append("%s=%s&" % (param, value)) |
| |
| url = "images?" + "".join(param_list) |
| |
| resp, body = self.client.get(url) |
| body = json.loads(body) |
| return resp, body['images'] |
| |
| def list_images_with_detail(self, params=None): |
| """Returns a detailed list of images filtered by any parameters""" |
| url = 'images/detail' |
| if params != None: |
| param_list = [] |
| for param, value in params.iteritems(): |
| param_list.append("%s=%s&" % (param, value)) |
| |
| url = "images/detail?" + "".join(param_list) |
| |
| resp, body = self.client.get(url) |
| body = json.loads(body) |
| return resp, body['images'] |
| |
| def get_image(self, image_id): |
| """Returns the details of a single image""" |
| resp, body = self.client.get("images/%s" % str(image_id)) |
| body = json.loads(body) |
| return resp, body['image'] |
| |
| def delete_image(self, image_id): |
| """Deletes the provided image""" |
| return self.client.delete("images/%s" % str(image_id)) |
| |
| def wait_for_image_resp_code(self, image_id, code): |
| """ |
| Waits until the HTTP response code for the request matches the |
| expected value |
| """ |
| resp, body = self.client.get("images/%s" % str(image_id)) |
| start = int(time.time()) |
| |
| while resp.status != code: |
| time.sleep(self.build_interval) |
| resp, body = self.client.get("images/%s" % str(image_id)) |
| |
| if int(time.time()) - start >= self.build_timeout: |
| raise exceptions.BuildErrorException |
| |
| def wait_for_image_status(self, image_id, status): |
| """Waits for an image to reach a given status.""" |
| resp, image = self.get_image(image_id) |
| start = int(time.time()) |
| |
| while image['status'] != status: |
| time.sleep(self.build_interval) |
| resp, image = self.get_image(image_id) |
| |
| if image['status'] == 'ERROR': |
| raise exceptions.TimeoutException |
| |
| if int(time.time()) - start >= self.build_timeout: |
| raise exceptions.BuildErrorException |
| |
| def list_image_metadata(self, image_id): |
| """Lists all metadata items for an image""" |
| resp, body = self.client.get("images/%s/metadata" % str(image_id)) |
| body = json.loads(body) |
| return resp, body['metadata'] |
| |
| def set_image_metadata(self, image_id, meta): |
| """Sets the metadata for an image""" |
| post_body = json.dumps({'metadata': meta}) |
| resp, body = self.client.put('images/%s/metadata' % |
| str(image_id), post_body, self.headers) |
| body = json.loads(body) |
| return resp, body['metadata'] |
| |
| def update_image_metadata(self, image_id, meta): |
| """Updates the metadata for an image""" |
| post_body = json.dumps({'metadata': meta}) |
| resp, body = self.client.post('images/%s/metadata' % |
| str(image_id), post_body, self.headers) |
| body = json.loads(body) |
| return resp, body['metadata'] |
| |
| def get_image_metadata_item(self, image_id, key): |
| """Returns the value for a specific image metadata key""" |
| resp, body = self.client.get("images/%s/metadata/%s" % |
| (str(image_id), key)) |
| body = json.loads(body) |
| return resp, body['meta'] |
| |
| def set_image_metadata_item(self, image_id, key, meta): |
| """Sets the value for a specific image metadata key""" |
| post_body = json.dumps({'meta': meta}) |
| resp, body = self.client.put('images/%s/metadata/%s' % |
| (str(image_id), key), post_body, |
| self.headers) |
| body = json.loads(body) |
| return resp, body['meta'] |
| |
| def delete_image_metadata_item(self, image_id, key): |
| """Deletes a single image metadata key/value pair""" |
| resp, body = self.client.delete("images/%s/metadata/%s" % |
| (str(image_id), key)) |
| return resp, body |