mirror of
				https://github.com/apache/cloudstack.git
				synced 2025-11-04 00:02:37 +01:00 
			
		
		
		
	This PR prepares marvin and tests for python3. it was part of #4479, until nose2 was decided to be abandoned from that PR. Re-PR of #4543 and #3730 to enable cooperation Co-authored-by: Daan Hoogland <dahn@onecht.net> Co-authored-by: Gabriel Beims Bräscher <gabriel@apache.org> Co-authored-by: Rohit Yadav <rohit.yadav@shapeblue.com>
		
			
				
	
	
		
			275 lines
		
	
	
		
			9.4 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			275 lines
		
	
	
		
			9.4 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
# 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
 | 
						|
import 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)
 |