diff --git a/test/integration/component/find_hosts_for_migration.py b/test/integration/component/find_hosts_for_migration.py new file mode 100644 index 00000000000..50789b2bf3b --- /dev/null +++ b/test/integration/component/find_hosts_for_migration.py @@ -0,0 +1,273 @@ +# 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. + +""" tests for find Host for migration both suitable and not suitable +""" +#Import Local Modules +from nose.plugins.attrib import attr +from marvin.cloudstackTestCase import cloudstackTestCase, unittest +from marvin.cloudstackAPI import (migrateVirtualMachine, + prepareHostForMaintenance, + cancelHostMaintenance, + findHostsForMigration) +from marvin.lib.utils import cleanup_resources +from marvin.lib.base import (Account, + VirtualMachine, + ServiceOffering, + Cluster, + Host, + Configurations) +from marvin.lib.common import (get_zone, + get_domain, + get_template, + list_hosts, + list_virtual_machines, + list_service_offering, + findSuitableHostForMigration) +import time + + +class Services: + + def __init__(self): + self.services = { + "account": { + "email": "test@test.com", + "firstname": "HA", + "lastname": "HA", + "username": "HA", + # Random characters are appended for unique + # username + "password": "password", + }, + "service_offering_with_tag": { + "name": "Tiny Instance With hosttag", + "displaytext": "Tiny Instance", + "cpunumber": 1, + "cpuspeed": 100, # in MHz + "memory": 128, # In MBs + "hosttags": "PREMIUM", + }, + "virtual_machine": { + "displayname": "VM", + "username": "root", + "password": "password", + "ssh_port": 22, + "hypervisor": 'XenServer', + # Hypervisor type should be same as + # hypervisor type of cluster + "privateport": 22, + "publicport": 22, + "protocol": 'TCP', + }, + "ostype": 'CentOS 5.3 (64-bit)', + "timeout": 100, + } + + +class TestHostsForMigration(cloudstackTestCase): + + @classmethod + def setUpClass(cls): + cls.testClient = super(TestHostsForMigration, cls).getClsTestClient() + cls.api_client = cls.testClient.getApiClient() + + cls.services = Services().services + # Get Zone, Domain and templates + cls.domain = get_domain(cls.api_client) + cls.zone = get_zone(cls.api_client, cls.testClient.getZoneForTests()) + + cls.template = get_template( + cls.api_client, + cls.zone.id, + cls.services["ostype"] + ) + + + clusterWithSufficientHosts = None + clusters = Cluster.list(cls.api_client, zoneid=cls.zone.id) + for cluster in clusters: + cls.hosts = Host.list(cls.api_client, clusterid=cluster.id, type="Routing") + if len(cls.hosts) >= 2: + clusterWithSufficientHosts = cluster + break + + if clusterWithSufficientHosts is None: + raise unittest.SkipTest("No Cluster with 2 hosts found") + + + Host.update(cls.api_client, id=cls.hosts[1].id, hosttags="PREMIUM") + + cls.services["virtual_machine"]["zoneid"] = cls.zone.id + cls.services["virtual_machine"]["template"] = cls.template.id + + cls.service_offering_with_tag = ServiceOffering.create( + cls.api_client, + cls.services["service_offering_with_tag"] + ) + + cls._cleanup = [ + cls.service_offering_with_tag, + ] + return + + @classmethod + def tearDownClass(cls): + try: + # Host.update(cls.api_client, id=cls.hosts[1].id, hosttags="") + #Cleanup resources used + cleanup_resources(cls.api_client, cls._cleanup) + except Exception as e: + raise Exception("Warning: Exception during cleanup : %s" % e) + return + + def setUp(self): + self.apiclient = self.testClient.getApiClient() + self.dbclient = self.testClient.getDbConnection() + self.account = Account.create( + self.apiclient, + self.services["account"], + admin=True, + domainid=self.domain.id + ) + self.cleanup = [self.account] + return + + def tearDown(self): + try: + #Clean up, terminate the created accounts, domains etc + cleanup_resources(self.apiclient, self.cleanup) + except Exception as e: + raise Exception("Warning: Exception during cleanup : %s" % e) + return + + @attr(required_hardware="false") + @attr(tags=["advanced", "basic", "simulator"]) + def test_01_find_hosts_for_migration(self): + """ Test find suitable and not-suitable list of hosts for migration """ + + # Steps, + #1. Create a Compute service offering with the tag . + #2. Create a Guest VM with the compute service offering created above. + #3. find hosts to migrate the vm crated above + + # Validations, + #1. Ensure that the offering is created with the tag + #The listServiceOffering API should list show tag + #2. Select the newly created VM and ensure that the Compute offering field value lists the compute service offering that was selected. + #3. findHostsForMigration cmd should list both suitable and not-suitable hosts + + list_service_response = list_service_offering( + self.apiclient, + id=self.service_offering_with_tag.id + ) + self.assertEqual( + isinstance(list_service_response, list), + True, + "listServiceOfferings returned invalid object in response." + ) + self.assertNotEqual( + len(list_service_response), + 0, + "listServiceOfferings returned empty list." + ) + self.assertEqual( + list_service_response[0].hosttags, + "PREMIUM", + "The service offering having tag" + ) + + #create virtual machine with the service offering with Ha enabled + virtual_machine = VirtualMachine.create( + self.apiclient, + self.services["virtual_machine"], + accountid=self.account.name, + domainid=self.account.domainid, + serviceofferingid=self.service_offering_with_tag.id + ) + vms = VirtualMachine.list( + self.apiclient, + id=virtual_machine.id, + listall=True + ) + self.assertEqual( + isinstance(vms, list), + True, + "listVirtualMachines returned invalid object in response." + ) + self.assertNotEqual( + len(vms), + 0, + "listVirtualMachines returned empty list." + ) + self.debug("Deployed VM on host: %s" % vms[0].hostid) + + #verify that the virtual machine created on the host with tag + list_hosts_response = list_hosts( + self.apiclient, + id=virtual_machine.hostid + ) + self.assertEqual( + isinstance(list_hosts_response, list), + True, + "listHosts returned invalid object in response." + ) + + self.assertNotEqual( + len(list_hosts_response), + 0, + "listHosts returned empty list." + ) + + host = list_hosts_response[0] + + self.assertEqual( + host.hosttags, + "PREMIUM", + "VM is created on a host having appropriate tag. %s" % host.uuid + ) + + try: + list_hosts_response = Host.listForMigration(self.apiclient, virtualmachineid=virtual_machine.id, + ) + except Exception as e: + raise Exception("Exception while getting hosts list suitable for migration: %s" % e) + + + self.assertEqual( + isinstance(list_hosts_response, list), + True, + "listHosts returned invalid object in response." + ) + + self.assertNotEqual( + len(list_hosts_response), + 0, + "listHosts returned empty response." + ) + suitableHost = set() + notSuitableHost = set() + + for host in list_hosts_response: + if host.suitableformigration: + suitableHost.add(host) + else: + notSuitableHost.add(host) + + self.assertTrue(notSuitableHost is not None, "notsuitablehost should not be None") + self.debug("Suitable Hosts: %s" % suitableHost) + self.debug("Not suitable Hosts: %s" % notSuitableHost) \ No newline at end of file