| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| """ BVT tests for Hosts Maintenance |
| """ |
| |
| # Import Local Modules |
| from marvin.codes import FAILED |
| from marvin.cloudstackTestCase import * |
| from marvin.cloudstackAPI import * |
| from marvin.lib.utils import * |
| from marvin.lib.base import * |
| from marvin.lib.common import * |
| from nose.plugins.attrib import attr |
| |
| from time import sleep |
| |
| _multiprocess_shared_ = False |
| |
| |
| class TestHostMaintenance(cloudstackTestCase): |
| |
| def setUp(self): |
| self.logger = logging.getLogger('TestHM') |
| self.stream_handler = logging.StreamHandler() |
| self.logger.setLevel(logging.DEBUG) |
| self.logger.addHandler(self.stream_handler) |
| self.apiclient = self.testClient.getApiClient() |
| self.hypervisor = self.testClient.getHypervisorInfo() |
| self.dbclient = self.testClient.getDbConnection() |
| self.services = self.testClient.getParsedTestDataConfig() |
| self.zone = get_zone(self.apiclient, self.testClient.getZoneForTests()) |
| self.pod = get_pod(self.apiclient, self.zone.id) |
| self.cleanup = [] |
| self.services = { |
| "service_offering": { |
| "name": "Ultra Tiny Instance", |
| "displaytext": "Ultra Tiny Instance", |
| "cpunumber": 1, |
| "cpuspeed": 100, |
| "memory": 128, |
| }, |
| "vm": { |
| "username": "root", |
| "password": "password", |
| "ssh_port": 22, |
| # Hypervisor type should be same as |
| # hypervisor type of cluster |
| "privateport": 22, |
| "publicport": 22, |
| "protocol": 'TCP', |
| }, |
| "natrule": { |
| "privateport": 22, |
| "publicport": 22, |
| "startport": 22, |
| "endport": 22, |
| "protocol": "TCP", |
| "cidrlist": '0.0.0.0/0', |
| }, |
| "ostype": 'CentOS 5.3 (64-bit)', |
| "sleep": 60, |
| "timeout": 10, |
| } |
| |
| |
| def tearDown(self): |
| try: |
| # Clean up, terminate the created templates |
| cleanup_resources(self.apiclient, self.cleanup) |
| |
| except Exception as e: |
| raise Exception("Warning: Exception during cleanup : %s" % e) |
| |
| return |
| |
| def createVMs(self, hostId, number): |
| |
| self.template = get_test_template( |
| self.apiclient, |
| self.zone.id, |
| self.hypervisor |
| ) |
| |
| if self.template == FAILED: |
| assert False, "get_test_template() failed to return template" |
| |
| self.logger.debug("Using template %s " % self.template.id) |
| |
| self.service_offering = ServiceOffering.create( |
| self.apiclient, |
| self.services["service_offering"] |
| ) |
| self.logger.debug("Using service offering %s " % self.service_offering.id) |
| |
| vms=[] |
| for i in range(0, number): |
| self.services["vm"]["zoneid"] = self.zone.id |
| self.services["vm"]["template"] = self.template.id |
| self.services["vm"]["displayname"] = 'vm' + str(i) |
| self.services["vm"]["hypervisor"] = self.hypervisor |
| vm = VirtualMachine.create( |
| self.apiclient, |
| self.services["vm"], |
| serviceofferingid=self.service_offering.id, |
| hostid=hostId |
| ) |
| vms.append(vm) |
| self.cleanup.append(vm) |
| self.logger.debug("VM create = {}".format(vm.id)) |
| return vms |
| |
| def checkVmMigratingOnHost(self, hostId): |
| vm_migrating=False |
| listVms1 = VirtualMachine.list( |
| self.apiclient, |
| hostid=hostId |
| ) |
| |
| if (listVms1 is not None): |
| self.logger.debug('Vms found = {} '.format(len(listVms1))) |
| for vm in listVms1: |
| if (vm.state == "Migrating"): |
| self.logger.debug('VirtualMachine on Hyp id = {} is in {}'.format(vm.id, vm.state)) |
| vm_migrating=True |
| break |
| |
| return (vm_migrating, None) |
| |
| def checkNoVmMigratingOnHost(self, hostId): |
| no_vm_migrating=True |
| listVms1 = VirtualMachine.list( |
| self.apiclient, |
| hostid=hostId |
| ) |
| |
| if (listVms1 is not None): |
| self.logger.debug('Vms found = {} '.format(len(listVms1))) |
| for vm in listVms1: |
| if (vm.state == "Migrating"): |
| self.logger.debug('VirtualMachine on Hyp id = {} is in {}'.format(vm.id, vm.state)) |
| no_vm_migrating=False |
| break |
| |
| return (no_vm_migrating, None) |
| |
| def noOfVMsOnHost(self, hostId): |
| listVms = VirtualMachine.list( |
| self.apiclient, |
| hostid=hostId |
| ) |
| no_of_vms=0 |
| if (listVms is not None): |
| for vm in listVms: |
| self.logger.debug('VirtualMachine on Hyp 1 = {}'.format(vm.id)) |
| no_of_vms=no_of_vms+1 |
| |
| return no_of_vms |
| |
| def hostPrepareAndCancelMaintenance(self, target_host_id, other_host_id, checkVMMigration): |
| |
| cmd = prepareHostForMaintenance.prepareHostForMaintenanceCmd() |
| cmd.id = target_host_id |
| response = self.apiclient.prepareHostForMaintenance(cmd) |
| |
| self.logger.debug('Host with id {} is in prepareHostForMaintenance'.format(target_host_id)) |
| |
| vm_migrating = wait_until(1, 10, checkVMMigration, other_host_id) |
| |
| cmd = cancelHostMaintenance.cancelHostMaintenanceCmd() |
| cmd.id = target_host_id |
| response = self.apiclient.cancelHostMaintenance(cmd) |
| |
| self.logger.debug('Host with id {} is in cancelHostMaintenance'.format(target_host_id) ) |
| |
| return vm_migrating |
| |
| @attr( |
| tags=[ |
| "advanced", |
| "advancedns", |
| "smoke", |
| "basic", |
| "eip", |
| "sg"], |
| required_hardware="true") |
| def test_01_cancel_host_maintenace_with_no_migration_jobs(self): |
| listHost = Host.list( |
| self.apiclient, |
| type='Routing', |
| zoneid=self.zone.id, |
| podid=self.pod.id, |
| ) |
| for host in listHost: |
| self.logger.debug('1 Hypervisor = {}'.format(host.id)) |
| |
| |
| if (len(listHost) < 2): |
| raise unittest.SkipTest("Cancel host maintenance when VMs are migrating should be tested for 2 or more hosts"); |
| return |
| |
| vm_migrating=False |
| |
| try: |
| |
| vm_migrating = self.hostPrepareAndCancelMaintenance(listHost[0].id, listHost[1].id, self.checkNoVmMigratingOnHost) |
| |
| vm_migrating = self.hostPrepareAndCancelMaintenance(listHost[1].id, listHost[0].id, self.checkNoVmMigratingOnHost) |
| |
| except Exception as e: |
| self.logger.debug("Exception {}".format(e)) |
| self.fail("Cancel host maintenance failed {}".format(e[0])) |
| |
| |
| if (vm_migrating == True): |
| raise unittest.SkipTest("VMs are migrating and the test will not be able to check the conditions the test is intended for"); |
| |
| |
| return |
| |
| |
| |
| |
| @attr( |
| tags=[ |
| "advanced", |
| "advancedns", |
| "smoke", |
| "basic", |
| "eip", |
| "sg"], |
| required_hardware="true") |
| def test_02_cancel_host_maintenace_with_migration_jobs(self): |
| |
| listHost = Host.list( |
| self.apiclient, |
| type='Routing', |
| zoneid=self.zone.id, |
| podid=self.pod.id, |
| ) |
| for host in listHost: |
| self.logger.debug('2 Hypervisor = {}'.format(host.id)) |
| |
| if (len(listHost) != 2): |
| raise unittest.SkipTest("Cancel host maintenance when VMs are migrating can only be tested with 2 hosts"); |
| return |
| |
| |
| no_of_vms = self.noOfVMsOnHost(listHost[0].id) |
| |
| no_of_vms = no_of_vms + self.noOfVMsOnHost(listHost[1].id) |
| |
| if no_of_vms < 5: |
| self.logger.debug("Create VMs as there are not enough vms to check host maintenance") |
| no_vm_req = 5 - no_of_vms |
| if (no_vm_req > 0): |
| self.logger.debug("Creating vms = {}".format(no_vm_req)) |
| self.vmlist = self.createVMs(listHost[0].id, no_vm_req) |
| |
| vm_migrating=False |
| |
| try: |
| |
| vm_migrating = self.hostPrepareAndCancelMaintenance(listHost[0].id, listHost[1].id, self.checkVmMigratingOnHost) |
| |
| vm_migrating = self.hostPrepareAndCancelMaintenance(listHost[1].id, listHost[0].id, self.checkVmMigratingOnHost) |
| |
| except Exception as e: |
| self.logger.debug("Exception {}".format(e)) |
| self.fail("Cancel host maintenance failed {}".format(e[0])) |
| |
| |
| if (vm_migrating == False): |
| raise unittest.SkipTest("No VM is migrating and the test will not be able to check the conditions the test is intended for"); |
| |
| |
| return |
| |
| |
