rajalakshmi-ganesan | b446557 | 2012-03-22 01:22:50 +0530 | [diff] [blame] | 1 | from tempest import exceptions |
chris fattarsi | 5098fa2 | 2012-04-17 13:27:00 -0700 | [diff] [blame] | 2 | from tempest.common.rest_client import RestClient |
rajalakshmi-ganesan | ddd9e0e | 2012-03-21 00:49:22 +0530 | [diff] [blame] | 3 | import json |
rajalakshmi-ganesan | b446557 | 2012-03-22 01:22:50 +0530 | [diff] [blame] | 4 | import time |
rajalakshmi-ganesan | ddd9e0e | 2012-03-21 00:49:22 +0530 | [diff] [blame] | 5 | |
| 6 | |
Rohit Karajgi | dd47d7e | 2012-07-31 04:11:01 -0700 | [diff] [blame^] | 7 | class VolumesExtensionsClient(RestClient): |
rajalakshmi-ganesan | ddd9e0e | 2012-03-21 00:49:22 +0530 | [diff] [blame] | 8 | |
chris fattarsi | 5098fa2 | 2012-04-17 13:27:00 -0700 | [diff] [blame] | 9 | def __init__(self, config, username, password, auth_url, tenant_name=None): |
Rohit Karajgi | dd47d7e | 2012-07-31 04:11:01 -0700 | [diff] [blame^] | 10 | super(VolumesExtensionsClient, self).__init__(config, username, |
| 11 | password, auth_url, |
| 12 | tenant_name) |
chris fattarsi | 5098fa2 | 2012-04-17 13:27:00 -0700 | [diff] [blame] | 13 | self.service = self.config.compute.catalog_type |
Rohit Karajgi | dd47d7e | 2012-07-31 04:11:01 -0700 | [diff] [blame^] | 14 | self.build_interval = self.config.volume.build_interval |
| 15 | self.build_timeout = self.config.volume.build_timeout |
rajalakshmi-ganesan | ddd9e0e | 2012-03-21 00:49:22 +0530 | [diff] [blame] | 16 | |
| 17 | def list_volumes(self, params=None): |
| 18 | """List all the volumes created""" |
| 19 | url = 'os-volumes' |
| 20 | if params != None: |
| 21 | param_list = [] |
| 22 | for param, value in params.iteritems(): |
| 23 | param_list.append("%s=%s&" % (param, value)) |
| 24 | |
| 25 | url += '?' + ' '.join(param_list) |
| 26 | |
chris fattarsi | 5098fa2 | 2012-04-17 13:27:00 -0700 | [diff] [blame] | 27 | resp, body = self.get(url) |
rajalakshmi-ganesan | ddd9e0e | 2012-03-21 00:49:22 +0530 | [diff] [blame] | 28 | body = json.loads(body) |
| 29 | return resp, body['volumes'] |
| 30 | |
rajalakshmi-ganesan | b446557 | 2012-03-22 01:22:50 +0530 | [diff] [blame] | 31 | def list_volumes_with_detail(self, params=None): |
| 32 | """List all the details of volumes""" |
| 33 | url = 'os-volumes/detail' |
| 34 | if params != None: |
| 35 | param_list = [] |
| 36 | for param, value in params.iteritems(): |
| 37 | param_list.append("%s=%s&" % (param, value)) |
| 38 | |
| 39 | url = '?' + ' '.join(param_list) |
| 40 | |
chris fattarsi | 5098fa2 | 2012-04-17 13:27:00 -0700 | [diff] [blame] | 41 | resp, body = self.get(url) |
rajalakshmi-ganesan | b446557 | 2012-03-22 01:22:50 +0530 | [diff] [blame] | 42 | body = json.loads(body) |
| 43 | return resp, body['volumes'] |
| 44 | |
rajalakshmi-ganesan | ddd9e0e | 2012-03-21 00:49:22 +0530 | [diff] [blame] | 45 | def get_volume(self, volume_id): |
| 46 | """Returns the details of a single volume""" |
| 47 | url = "os-volumes/%s" % str(volume_id) |
chris fattarsi | 5098fa2 | 2012-04-17 13:27:00 -0700 | [diff] [blame] | 48 | resp, body = self.get(url) |
rajalakshmi-ganesan | ddd9e0e | 2012-03-21 00:49:22 +0530 | [diff] [blame] | 49 | body = json.loads(body) |
| 50 | return resp, body['volume'] |
| 51 | |
rajalakshmi-ganesan | b446557 | 2012-03-22 01:22:50 +0530 | [diff] [blame] | 52 | def create_volume(self, size, **kwargs): |
| 53 | """ |
| 54 | Creates a new Volume. |
| 55 | size(Required): Size of volume in GB. |
| 56 | Following optional keyword arguments are accepted: |
| 57 | display_name: Optional Volume Name. |
| 58 | metadata: A dictionary of values to be used as metadata. |
| 59 | """ |
| 60 | post_body = { |
| 61 | 'size': size, |
| 62 | 'display_name': kwargs.get('display_name'), |
| 63 | 'metadata': kwargs.get('metadata'), |
| 64 | } |
| 65 | |
| 66 | post_body = json.dumps({'volume': post_body}) |
chris fattarsi | 5098fa2 | 2012-04-17 13:27:00 -0700 | [diff] [blame] | 67 | resp, body = self.post('os-volumes', post_body, self.headers) |
rajalakshmi-ganesan | b446557 | 2012-03-22 01:22:50 +0530 | [diff] [blame] | 68 | body = json.loads(body) |
| 69 | return resp, body['volume'] |
| 70 | |
rajalakshmi-ganesan | ddd9e0e | 2012-03-21 00:49:22 +0530 | [diff] [blame] | 71 | def delete_volume(self, volume_id): |
| 72 | """Deletes the Specified Volume""" |
chris fattarsi | 5098fa2 | 2012-04-17 13:27:00 -0700 | [diff] [blame] | 73 | return self.delete("os-volumes/%s" % str(volume_id)) |
rajalakshmi-ganesan | b446557 | 2012-03-22 01:22:50 +0530 | [diff] [blame] | 74 | |
| 75 | def wait_for_volume_status(self, volume_id, status): |
| 76 | """Waits for a Volume to reach a given status""" |
| 77 | resp, body = self.get_volume(volume_id) |
| 78 | volume_name = body['displayName'] |
| 79 | volume_status = body['status'] |
| 80 | start = int(time.time()) |
| 81 | |
| 82 | while volume_status != status: |
| 83 | time.sleep(self.build_interval) |
| 84 | resp, body = self.get_volume(volume_id) |
| 85 | volume_status = body['status'] |
| 86 | if volume_status == 'error': |
rajalakshmi-ganesan | e3bb58f | 2012-05-16 12:01:15 +0530 | [diff] [blame] | 87 | raise exceptions.VolumeBuildErrorException(volume_id=volume_id) |
rajalakshmi-ganesan | b446557 | 2012-03-22 01:22:50 +0530 | [diff] [blame] | 88 | |
| 89 | if int(time.time()) - start >= self.build_timeout: |
| 90 | message = 'Volume %s failed to reach %s status within '\ |
| 91 | 'the required time (%s s).' % (volume_name, status, |
| 92 | self.build_timeout) |
| 93 | raise exceptions.TimeoutException(message) |
David Kranz | 6aceb4a | 2012-06-05 14:05:45 -0400 | [diff] [blame] | 94 | |
| 95 | def is_resource_deleted(self, id): |
| 96 | try: |
| 97 | self.get_volume(id) |
| 98 | except exceptions.NotFound: |
| 99 | return True |
| 100 | return False |