| Attila Fazekas | 34b75a1 | 2014-02-28 16:18:01 +0100 | [diff] [blame] | 1 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 2 | # you may not use this file except in compliance with the License. |
| 3 | # You may obtain a copy of the License at |
| 4 | # |
| 5 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 6 | # |
| 7 | # Unless required by applicable law or agreed to in writing, software |
| 8 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 9 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 10 | # See the License for the specific language governing permissions and |
| 11 | # limitations under the License. |
| 12 | |
| 13 | from tempest.common.utils import data_utils |
| 14 | from tempest.common.utils.linux import remote_client |
| 15 | from tempest import config |
| 16 | import tempest.stress.stressaction as stressaction |
| 17 | import tempest.test |
| 18 | |
| 19 | import re |
| 20 | CONF = config.CONF |
| 21 | |
| 22 | |
| 23 | class VolumeVerifyStress(stressaction.StressAction): |
| 24 | |
| 25 | def _create_keypair(self): |
| 26 | keyname = data_utils.rand_name("key") |
| 27 | resp, self.key = self.manager.keypairs_client.create_keypair(keyname) |
| 28 | assert(resp.status == 200) |
| 29 | |
| 30 | def _delete_keypair(self): |
| 31 | resp, _ = self.manager.keypairs_client.delete_keypair(self.key['name']) |
| 32 | assert(resp.status == 202) |
| 33 | |
| 34 | def _create_vm(self): |
| 35 | self.name = name = data_utils.rand_name("instance") |
| 36 | servers_client = self.manager.servers_client |
| 37 | self.logger.info("creating %s" % name) |
| 38 | vm_args = self.vm_extra_args.copy() |
| 39 | vm_args['security_groups'] = [self.sec_grp] |
| 40 | vm_args['key_name'] = self.key['name'] |
| 41 | resp, server = servers_client.create_server(name, self.image, |
| 42 | self.flavor, |
| 43 | **vm_args) |
| 44 | self.server_id = server['id'] |
| 45 | assert(resp.status == 202) |
| 46 | self.manager.servers_client.wait_for_server_status(self.server_id, |
| 47 | 'ACTIVE') |
| 48 | |
| 49 | def _destroy_vm(self): |
| 50 | self.logger.info("deleting server: %s" % self.server_id) |
| 51 | resp, _ = self.manager.servers_client.delete_server(self.server_id) |
| 52 | assert(resp.status == 204) # It cannot be 204 if I had to wait.. |
| 53 | self.manager.servers_client.wait_for_server_termination(self.server_id) |
| 54 | self.logger.info("deleted server: %s" % self.server_id) |
| 55 | |
| 56 | def _create_sec_group(self): |
| 57 | sec_grp_cli = self.manager.security_groups_client |
| 58 | s_name = data_utils.rand_name('sec_grp-') |
| 59 | s_description = data_utils.rand_name('desc-') |
| 60 | _, self.sec_grp = sec_grp_cli.create_security_group(s_name, |
| 61 | s_description) |
| 62 | create_rule = sec_grp_cli.create_security_group_rule |
| 63 | create_rule(self.sec_grp['id'], 'tcp', 22, 22) |
| 64 | create_rule(self.sec_grp['id'], 'icmp', -1, -1) |
| 65 | |
| 66 | def _destroy_sec_grp(self): |
| 67 | sec_grp_cli = self.manager.security_groups_client |
| 68 | sec_grp_cli.delete_security_group(self.sec_grp['id']) |
| 69 | |
| 70 | def _create_floating_ip(self): |
| 71 | floating_cli = self.manager.floating_ips_client |
| 72 | _, self.floating = floating_cli.create_floating_ip(self.floating_pool) |
| 73 | |
| 74 | def _destroy_floating_ip(self): |
| 75 | cli = self.manager.floating_ips_client |
| 76 | cli.delete_floating_ip(self.floating['id']) |
| 77 | cli.wait_for_resource_deletion(self.floating['id']) |
| 78 | self.logger.info("Deleted Floating IP %s", str(self.floating['ip'])) |
| 79 | |
| 80 | def _create_volume(self): |
| 81 | name = data_utils.rand_name("volume") |
| 82 | self.logger.info("creating volume: %s" % name) |
| 83 | volumes_client = self.manager.volumes_client |
| 84 | resp, self.volume = volumes_client.create_volume(size=1, |
| 85 | display_name= |
| 86 | name) |
| 87 | assert(resp.status == 200) |
| 88 | volumes_client.wait_for_volume_status(self.volume['id'], |
| 89 | 'available') |
| 90 | self.logger.info("created volume: %s" % self.volume['id']) |
| 91 | |
| 92 | def _delete_volume(self): |
| 93 | self.logger.info("deleting volume: %s" % self.volume['id']) |
| 94 | volumes_client = self.manager.volumes_client |
| 95 | resp, _ = volumes_client.delete_volume(self.volume['id']) |
| 96 | assert(resp.status == 202) |
| 97 | volumes_client.wait_for_resource_deletion(self.volume['id']) |
| 98 | self.logger.info("deleted volume: %s" % self.volume['id']) |
| 99 | |
| 100 | def _wait_disassociate(self): |
| 101 | cli = self.manager.floating_ips_client |
| 102 | |
| 103 | def func(): |
| 104 | _, floating = cli.get_floating_ip_details(self.floating['id']) |
| 105 | return floating['instance_id'] is None |
| 106 | |
| 107 | if not tempest.test.call_until_true(func, CONF.compute.build_timeout, |
| 108 | CONF.compute.build_interval): |
| 109 | raise RuntimeError("IP disassociate timeout!") |
| 110 | |
| 111 | def new_server_ops(self): |
| 112 | self._create_vm() |
| 113 | cli = self.manager.floating_ips_client |
| 114 | cli.associate_floating_ip_to_server(self.floating['ip'], |
| 115 | self.server_id) |
| 116 | if self.ssh_test_before_attach and self.enable_ssh_verify: |
| 117 | self.logger.info("Scanning for block devices via ssh on %s" |
| 118 | % self.server_id) |
| 119 | self.part_wait(self.detach_match_count) |
| 120 | |
| 121 | def setUp(self, **kwargs): |
| 122 | """Note able configuration combinations: |
| 123 | Closest options to the test_stamp_pattern: |
| 124 | new_server = True |
| 125 | new_volume = True |
| 126 | enable_ssh_verify = True |
| 127 | ssh_test_before_attach = False |
| 128 | Just attaching: |
| 129 | new_server = False |
| 130 | new_volume = False |
| 131 | enable_ssh_verify = True |
| 132 | ssh_test_before_attach = True |
| 133 | Mostly API load by repeated attachment: |
| 134 | new_server = False |
| 135 | new_volume = False |
| 136 | enable_ssh_verify = False |
| 137 | ssh_test_before_attach = False |
| 138 | Minimal Nova load, but cinder load not decreased: |
| 139 | new_server = False |
| 140 | new_volume = True |
| 141 | enable_ssh_verify = True |
| 142 | ssh_test_before_attach = True |
| 143 | """ |
| 144 | self.image = CONF.compute.image_ref |
| 145 | self.flavor = CONF.compute.flavor_ref |
| 146 | self.vm_extra_args = kwargs.get('vm_extra_args', {}) |
| 147 | self.floating_pool = kwargs.get('floating_pool', None) |
| 148 | self.new_volume = kwargs.get('new_volume', True) |
| 149 | self.new_server = kwargs.get('new_server', False) |
| 150 | self.enable_ssh_verify = kwargs.get('enable_ssh_verify', True) |
| 151 | self.ssh_test_before_attach = kwargs.get('ssh_test_before_attach', |
| 152 | False) |
| 153 | self.part_line_re = re.compile(kwargs.get('part_line_re', '.*vd.*')) |
| 154 | self.detach_match_count = kwargs.get('detach_match_count', 1) |
| 155 | self.attach_match_count = kwargs.get('attach_match_count', 2) |
| 156 | self.part_name = kwargs.get('part_name', '/dev/vdc') |
| 157 | |
| 158 | self._create_floating_ip() |
| 159 | self._create_sec_group() |
| 160 | self._create_keypair() |
| 161 | private_key = self.key['private_key'] |
| 162 | username = CONF.compute.image_ssh_user |
| 163 | self.remote_client = remote_client.RemoteClient(self.floating['ip'], |
| 164 | username, |
| 165 | pkey=private_key) |
| 166 | if not self.new_volume: |
| 167 | self._create_volume() |
| 168 | if not self.new_server: |
| 169 | self.new_server_ops() |
| 170 | |
| 171 | # now we just test is number of partition increased or decrised |
| 172 | def part_wait(self, num_match): |
| 173 | def _part_state(): |
| 174 | self.partitions = self.remote_client.get_partitions().split('\n') |
| 175 | matching = 0 |
| 176 | for part_line in self.partitions[1:]: |
| 177 | if self.part_line_re.match(part_line): |
| 178 | matching += 1 |
| 179 | return matching == num_match |
| 180 | if tempest.test.call_until_true(_part_state, |
| 181 | CONF.compute.build_timeout, |
| 182 | CONF.compute.build_interval): |
| 183 | return |
| 184 | else: |
| 185 | raise RuntimeError("Unexpected partitions: %s", |
| 186 | str(self.partitions)) |
| 187 | |
| 188 | def run(self): |
| 189 | if self.new_server: |
| 190 | self.new_server_ops() |
| 191 | if self.new_volume: |
| 192 | self._create_volume() |
| 193 | servers_client = self.manager.servers_client |
| 194 | self.logger.info("attach volume (%s) to vm %s" % |
| 195 | (self.volume['id'], self.server_id)) |
| 196 | resp, body = servers_client.attach_volume(self.server_id, |
| 197 | self.volume['id'], |
| 198 | self.part_name) |
| 199 | assert(resp.status == 200) |
| 200 | self.manager.volumes_client.wait_for_volume_status(self.volume['id'], |
| 201 | 'in-use') |
| 202 | if self.enable_ssh_verify: |
| 203 | self.logger.info("Scanning for new block device on %s" |
| 204 | % self.server_id) |
| 205 | self.part_wait(self.attach_match_count) |
| 206 | |
| 207 | resp, body = servers_client.detach_volume(self.server_id, |
| 208 | self.volume['id']) |
| 209 | assert(resp.status == 202) |
| 210 | self.manager.volumes_client.wait_for_volume_status(self.volume['id'], |
| 211 | 'available') |
| 212 | if self.enable_ssh_verify: |
| 213 | self.logger.info("Scanning for block device disapperance on %s" |
| 214 | % self.server_id) |
| 215 | self.part_wait(self.detach_match_count) |
| 216 | if self.new_volume: |
| 217 | self._delete_volume() |
| 218 | if self.new_server: |
| 219 | self._destroy_vm() |
| 220 | |
| 221 | def tearDown(self): |
| 222 | cli = self.manager.floating_ips_client |
| 223 | cli.disassociate_floating_ip_from_server(self.floating['ip'], |
| 224 | self.server_id) |
| 225 | self._wait_disassociate() |
| 226 | if not self.new_server: |
| 227 | self._destroy_vm() |
| 228 | self._delete_keypair() |
| 229 | self._destroy_floating_ip() |
| 230 | self._destroy_sec_grp() |
| 231 | if not self.new_volume: |
| 232 | self._delete_volume() |