chris fattarsi | 5098fa2 | 2012-04-17 13:27:00 -0700 | [diff] [blame] | 1 | from tempest.common.rest_client import RestClient |
Daryl Walleck | 1465d61 | 2011-11-02 02:22:15 -0500 | [diff] [blame] | 2 | import json |
Daryl Walleck | 1465d61 | 2011-11-02 02:22:15 -0500 | [diff] [blame] | 3 | |
| 4 | |
Tiago Mello | eda03b5 | 2012-08-22 23:47:29 -0300 | [diff] [blame] | 5 | class FlavorsClientJSON(RestClient): |
Daryl Walleck | 1465d61 | 2011-11-02 02:22:15 -0500 | [diff] [blame] | 6 | |
Daryl Walleck | 587385b | 2012-03-03 13:00:26 -0600 | [diff] [blame] | 7 | def __init__(self, config, username, password, auth_url, tenant_name=None): |
Tiago Mello | eda03b5 | 2012-08-22 23:47:29 -0300 | [diff] [blame] | 8 | super(FlavorsClientJSON, self).__init__(config, username, password, |
| 9 | auth_url, tenant_name) |
chris fattarsi | 5098fa2 | 2012-04-17 13:27:00 -0700 | [diff] [blame] | 10 | self.service = self.config.compute.catalog_type |
Daryl Walleck | 1465d61 | 2011-11-02 02:22:15 -0500 | [diff] [blame] | 11 | |
| 12 | def list_flavors(self, params=None): |
| 13 | url = 'flavors' |
Zhongyue Luo | e471d6e | 2012-09-17 17:02:43 +0800 | [diff] [blame] | 14 | if params is not None: |
Daryl Walleck | 1465d61 | 2011-11-02 02:22:15 -0500 | [diff] [blame] | 15 | param_list = [] |
| 16 | for param, value in params.iteritems(): |
| 17 | param_list.append("%s=%s&" % (param, value)) |
| 18 | |
| 19 | url = "flavors?" + "".join(param_list) |
| 20 | |
chris fattarsi | 5098fa2 | 2012-04-17 13:27:00 -0700 | [diff] [blame] | 21 | resp, body = self.get(url) |
Daryl Walleck | 1465d61 | 2011-11-02 02:22:15 -0500 | [diff] [blame] | 22 | body = json.loads(body) |
Daryl Walleck | 74d0409 | 2012-01-10 23:33:46 -0600 | [diff] [blame] | 23 | return resp, body['flavors'] |
Daryl Walleck | 1465d61 | 2011-11-02 02:22:15 -0500 | [diff] [blame] | 24 | |
| 25 | def list_flavors_with_detail(self, params=None): |
| 26 | url = 'flavors/detail' |
Zhongyue Luo | e471d6e | 2012-09-17 17:02:43 +0800 | [diff] [blame] | 27 | if params is not None: |
Daryl Walleck | 1465d61 | 2011-11-02 02:22:15 -0500 | [diff] [blame] | 28 | param_list = [] |
| 29 | for param, value in params.iteritems(): |
| 30 | param_list.append("%s=%s&" % (param, value)) |
| 31 | |
| 32 | url = "flavors/detail?" + "".join(param_list) |
| 33 | |
chris fattarsi | 5098fa2 | 2012-04-17 13:27:00 -0700 | [diff] [blame] | 34 | resp, body = self.get(url) |
Daryl Walleck | 1465d61 | 2011-11-02 02:22:15 -0500 | [diff] [blame] | 35 | body = json.loads(body) |
Daryl Walleck | 74d0409 | 2012-01-10 23:33:46 -0600 | [diff] [blame] | 36 | return resp, body['flavors'] |
Daryl Walleck | 1465d61 | 2011-11-02 02:22:15 -0500 | [diff] [blame] | 37 | |
| 38 | def get_flavor_details(self, flavor_id): |
chris fattarsi | 5098fa2 | 2012-04-17 13:27:00 -0700 | [diff] [blame] | 39 | resp, body = self.get("flavors/%s" % str(flavor_id)) |
Daryl Walleck | 1465d61 | 2011-11-02 02:22:15 -0500 | [diff] [blame] | 40 | body = json.loads(body) |
| 41 | return resp, body['flavor'] |
Rohit Karajgi | 0353029 | 2012-04-24 17:00:50 -0700 | [diff] [blame] | 42 | |
| 43 | def create_flavor(self, name, ram, vcpus, disk, ephemeral, flavor_id, |
Zhongyue Luo | 79d8d36 | 2012-09-25 13:49:27 +0800 | [diff] [blame] | 44 | swap, rxtx): |
Rohit Karajgi | 0353029 | 2012-04-24 17:00:50 -0700 | [diff] [blame] | 45 | """Creates a new flavor or instance type""" |
| 46 | post_body = { |
Zhongyue Luo | 30a563f | 2012-09-30 23:43:50 +0900 | [diff] [blame^] | 47 | 'name': name, |
| 48 | 'ram': ram, |
| 49 | 'vcpus': vcpus, |
| 50 | 'disk': disk, |
| 51 | 'OS-FLV-EXT-DATA:ephemeral': ephemeral, |
| 52 | 'id': flavor_id, |
| 53 | 'swap': swap, |
| 54 | 'rxtx_factor': rxtx, |
| 55 | } |
Rohit Karajgi | 0353029 | 2012-04-24 17:00:50 -0700 | [diff] [blame] | 56 | |
| 57 | post_body = json.dumps({'flavor': post_body}) |
| 58 | resp, body = self.post('flavors', post_body, self.headers) |
| 59 | |
| 60 | body = json.loads(body) |
| 61 | return resp, body['flavor'] |
| 62 | |
| 63 | def delete_flavor(self, flavor_id): |
| 64 | """Deletes the given flavor""" |
| 65 | return self.delete("flavors/%s" % str(flavor_id)) |