| from tempest import exceptions |
| from tempest.common import rest_client |
| import json |
| import time |
| |
| |
| class ServersClient(object): |
| |
| def __init__(self, config, username, key, auth_url, tenant_name=None): |
| self.config = config |
| self.client = rest_client.RestClient(config, username, key, |
| auth_url, 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_server(self, name, image_ref, flavor_ref, meta=None, |
| personality=None, accessIPv4=None, accessIPv6=None, |
| adminPass=None): |
| """ |
| Creates an instance of a server. |
| name: The name of the server. |
| image_ref: The reference to the image used to build the server. |
| flavor_ref: The flavor used to build the server. |
| adminPass: Sets the initial root password. |
| meta: A dictionary of values to be used as metadata. |
| personality: A list of dictionaries for files to be injected into |
| the server. |
| accessIPv4: The IPv4 access address for the server. |
| accessIPv6: The IPv6 access address for the server. |
| """ |
| |
| post_body = { |
| 'name': name, |
| 'imageRef': image_ref, |
| 'flavorRef': flavor_ref, |
| } |
| |
| if meta != None: |
| post_body['metadata'] = meta |
| |
| if personality != None: |
| post_body['personality'] = personality |
| |
| if adminPass != None: |
| post_body['adminPass'] = adminPass |
| |
| if accessIPv4 != None: |
| post_body['accessIPv4'] = accessIPv4 |
| |
| if accessIPv6 != None: |
| post_body['accessIPv6'] = accessIPv6 |
| |
| post_body = json.dumps({'server': post_body}) |
| resp, body = self.client.post('servers', post_body, self.headers) |
| body = json.loads(body) |
| return resp, body['server'] |
| |
| def update_server(self, server_id, name=None, meta=None, accessIPv4=None, |
| accessIPv6=None): |
| """ |
| Updates the properties of an existing server. |
| server_id: The id of an existing server. |
| name: The name of the server. |
| personality: A list of files to be injected into the server. |
| accessIPv4: The IPv4 access address for the server. |
| accessIPv6: The IPv6 access address for the server. |
| """ |
| |
| post_body = {} |
| |
| if meta != None: |
| post_body['metadata'] = meta |
| |
| if name != None: |
| post_body['name'] = name |
| |
| if accessIPv4 != None: |
| post_body['accessIPv4'] = accessIPv4 |
| |
| if accessIPv6 != None: |
| post_body['accessIPv6'] = accessIPv6 |
| |
| post_body = json.dumps({'server': post_body}) |
| resp, body = self.client.put("servers/%s" % str(server_id), |
| post_body, self.headers) |
| body = json.loads(body) |
| return resp, body['server'] |
| |
| def get_server(self, server_id): |
| """Returns the details of an existing server""" |
| resp, body = self.client.get("servers/%s" % str(server_id)) |
| body = json.loads(body) |
| return resp, body['server'] |
| |
| def delete_server(self, server_id): |
| """Deletes the given server""" |
| return self.client.delete("servers/%s" % str(server_id)) |
| |
| def list_servers(self, params=None): |
| """Lists all servers for a user""" |
| |
| url = 'servers' |
| if params != None: |
| param_list = [] |
| for param, value in params.iteritems(): |
| param_list.append("%s=%s&" % (param, value)) |
| |
| url = "servers?" + "".join(param_list) |
| |
| resp, body = self.client.get(url) |
| body = json.loads(body) |
| return resp, body |
| |
| def list_servers_with_detail(self, params=None): |
| """Lists all servers in detail for a user""" |
| |
| url = 'servers/detail' |
| if params != None: |
| param_list = [] |
| for param, value in params.iteritems(): |
| param_list.append("%s=%s&" % (param, value)) |
| |
| url = "servers/detail?" + "".join(param_list) |
| |
| resp, body = self.client.get(url) |
| body = json.loads(body) |
| return resp, body |
| |
| def wait_for_server_status(self, server_id, status): |
| """Waits for a server to reach a given status""" |
| resp, body = self.get_server(server_id) |
| server_status = body['status'] |
| start = int(time.time()) |
| |
| while(server_status != status): |
| time.sleep(self.build_interval) |
| resp, body = self.get_server(server_id) |
| server_status = body['status'] |
| |
| if(server_status == 'ERROR'): |
| raise exceptions.BuildErrorException |
| |
| if (int(time.time()) - start >= self.build_timeout): |
| raise exceptions.TimeoutException |
| |
| def list_addresses(self, server_id): |
| """Lists all addresses for a server""" |
| resp, body = self.client.get("servers/%s/ips" % str(server_id)) |
| body = json.loads(body) |
| return resp, body['addresses'] |
| |
| def list_addresses_by_network(self, server_id, network_id): |
| """Lists all addresses of a specific network type for a server""" |
| resp, body = self.client.get("servers/%s/ips/%s" % |
| (str(server_id), network_id)) |
| body = json.loads(body) |
| return resp, body |
| |
| def change_password(self, server_id, password): |
| """Changes the root password for the server""" |
| post_body = { |
| 'changePassword': { |
| 'adminPass': password, |
| } |
| } |
| |
| post_body = json.dumps(post_body) |
| return self.client.post('servers/%s/action' % str(server_id), |
| post_body, self.headers) |
| |
| def reboot(self, server_id, reboot_type): |
| """Reboots a server""" |
| post_body = { |
| 'reboot': { |
| 'type': reboot_type, |
| } |
| } |
| |
| post_body = json.dumps(post_body) |
| return self.client.post('servers/%s/action' % str(server_id), |
| post_body, self.headers) |
| |
| def rebuild(self, server_id, image_ref, name=None, meta=None, |
| personality=None, adminPass=None): |
| """Rebuilds a server with a new image""" |
| post_body = { |
| 'imageRef': image_ref, |
| } |
| |
| if name != None: |
| post_body['name'] = name |
| |
| if adminPass != None: |
| post_body['adminPass'] = adminPass |
| |
| if meta != None: |
| post_body['metadata'] = meta |
| |
| if personality != None: |
| post_body['personality'] = personality |
| |
| post_body = json.dumps({'rebuild': post_body}) |
| resp, body = self.client.post('servers/%s/action' % |
| str(server_id), post_body, |
| self.headers) |
| body = json.loads(body) |
| return resp, body['server'] |
| |
| def resize(self, server_id, flavor_ref): |
| """Changes the flavor of a server.""" |
| post_body = { |
| 'resize': { |
| 'flavorRef': flavor_ref, |
| } |
| } |
| |
| 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 confirm_resize(self, server_id): |
| """Confirms the flavor change for a server""" |
| post_body = { |
| 'confirmResize': None, |
| } |
| |
| 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 revert_resize(self, server_id): |
| """Reverts a server back to its original flavor""" |
| post_body = { |
| 'revertResize': None, |
| } |
| |
| 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 create_image(self, server_id, image_name): |
| """Creates an image of the given server""" |
| post_body = { |
| 'createImage': { |
| 'name': image_name, |
| } |
| } |
| |
| post_body = json.dumps(post_body) |
| resp, body = self.client.post('servers/%s/action' % |
| str(server_id), post_body, self.headers) |
| body = json.loads(body) |
| return resp, body |
| |
| def list_server_metadata(self, server_id): |
| resp, body = self.client.get("servers/%s/metadata" % str(server_id)) |
| body = json.loads(body) |
| return resp, body['metadata'] |
| |
| def set_server_metadata(self, server_id, meta): |
| post_body = json.dumps({'metadata': meta}) |
| resp, body = self.client.put('servers/%s/metadata' % |
| str(server_id), post_body, self.headers) |
| body = json.loads(body) |
| return resp, body['metadata'] |
| |
| def update_server_metadata(self, server_id, meta): |
| post_body = json.dumps({'metadata': meta}) |
| resp, body = self.client.post('servers/%s/metadata' % |
| str(server_id), post_body, self.headers) |
| body = json.loads(body) |
| return resp, body['metadata'] |
| |
| def get_server_metadata_item(self, server_id, key): |
| resp, body = self.client.get("servers/%s/metadata/%s" % |
| (str(server_id), key)) |
| body = json.loads(body) |
| return resp, body['meta'] |
| |
| def set_server_metadata_item(self, server_id, key, meta): |
| post_body = json.dumps({'meta': meta}) |
| resp, body = self.client.put('servers/%s/metadata/%s' % |
| (str(server_id), key), |
| post_body, self.headers) |
| body = json.loads(body) |
| return resp, body['meta'] |
| |
| def delete_server_metadata_item(self, server_id, key): |
| resp, body = self.client.delete("servers/%s/metadata/%s" % |
| (str(server_id), key)) |
| return resp, body |