mirror of
				https://github.com/apache/cloudstack.git
				synced 2025-11-04 00:02:37 +01:00 
			
		
		
		
	Fixing some component tests and adding them in travis. Signed-off-by: Rohit Yadav <rohit.yadav@shapeblue.com>
		
			
				
	
	
		
			173 lines
		
	
	
		
			7.9 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			173 lines
		
	
	
		
			7.9 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 primary storage limits during upload volume
 | 
						|
"""
 | 
						|
import unittest
 | 
						|
 | 
						|
from ddt import ddt, data
 | 
						|
from marvin.cloudstackTestCase import cloudstackTestCase
 | 
						|
from marvin.codes import PASS, RESOURCE_PRIMARY_STORAGE, FAIL, USER_ACCOUNT
 | 
						|
from marvin.lib.base import Domain, Account, VirtualMachine, DiskOffering, ServiceOffering, Volume
 | 
						|
from marvin.lib.common import get_domain, get_zone, update_resource_limit, uploadVolume, matchResourceCount, get_template
 | 
						|
from marvin.lib.utils import cleanup_resources, validateList
 | 
						|
from nose.plugins.attrib import attr
 | 
						|
 | 
						|
 | 
						|
@ddt
 | 
						|
class TestPrimaryResourceLimitsVolume(cloudstackTestCase):
 | 
						|
    @classmethod
 | 
						|
    def setUpClass(cls):
 | 
						|
        cloudstacktestclient = super(TestPrimaryResourceLimitsVolume,
 | 
						|
                                     cls).getClsTestClient()
 | 
						|
        cls.api_client = cloudstacktestclient.getApiClient()
 | 
						|
        cls.hypervisor = cloudstacktestclient.getHypervisorInfo()
 | 
						|
        # Fill services from the external config file
 | 
						|
        cls.services = cloudstacktestclient.getParsedTestDataConfig()
 | 
						|
        # Get Zone, Domain and templates
 | 
						|
        cls.domain = get_domain(cls.api_client)
 | 
						|
        cls.zone = get_zone(cls.api_client, cloudstacktestclient.getZoneForTests())
 | 
						|
        cls.services["mode"] = cls.zone.networktype
 | 
						|
        cls._cleanup = []
 | 
						|
        cls.unsupportedStorageType = False
 | 
						|
        cls.template = get_template(cls.api_client, cls.zone.id, cls.services["ostype"])
 | 
						|
        cls.services["virtual_machine"]["zoneid"] = cls.zone.id
 | 
						|
        cls.services["virtual_machine"]["template"] = cls.template.id
 | 
						|
        cls.services["volume"]["zoneid"] = cls.zone.id
 | 
						|
        try:
 | 
						|
            cls.service_offering = ServiceOffering.create(cls.api_client, cls.services["service_offering"])
 | 
						|
            cls.services["disk_offering"]["disksize"] = 2
 | 
						|
            cls.disk_offering = DiskOffering.create(cls.api_client, cls.services["disk_offering"])
 | 
						|
            cls._cleanup.append(cls.service_offering)
 | 
						|
            cls._cleanup.append(cls.disk_offering)
 | 
						|
        except Exception as e:
 | 
						|
            cls.tearDownClass()
 | 
						|
            raise unittest.SkipTest("Exception in setUpClass: %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @classmethod
 | 
						|
    def tearDownClass(cls):
 | 
						|
        try:
 | 
						|
            # 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.cleanup = []
 | 
						|
        return
 | 
						|
 | 
						|
    def tearDown(self):
 | 
						|
        try:
 | 
						|
            # Clean up, terminate the created instance, volumes and snapshots
 | 
						|
            cleanup_resources(self.apiclient, self.cleanup)
 | 
						|
            pass
 | 
						|
        except Exception as e:
 | 
						|
            raise Exception("Warning: Exception during cleanup : %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    def setupNormalAccount(self):
 | 
						|
        """Setup the account required for the test"""
 | 
						|
 | 
						|
        try:
 | 
						|
            self.domain = Domain.create(self.apiclient,
 | 
						|
                                        services=self.services["domain"],
 | 
						|
                                        parentdomainid=self.domain.id)
 | 
						|
 | 
						|
            self.account = Account.create(self.apiclient, self.services["account"],
 | 
						|
                                          domainid=self.domain.id, admin=False)
 | 
						|
            self.cleanup.append(self.account)
 | 
						|
            self.cleanup.append(self.domain)
 | 
						|
 | 
						|
            self.virtualMachine = VirtualMachine.create(self.api_client, self.services["virtual_machine"],
 | 
						|
                                                        accountid=self.account.name, domainid=self.account.domainid,
 | 
						|
                                                        diskofferingid=self.disk_offering.id,
 | 
						|
                                                        serviceofferingid=self.service_offering.id)
 | 
						|
 | 
						|
            accounts = Account.list(self.apiclient, id=self.account.id)
 | 
						|
 | 
						|
            self.assertEqual(validateList(accounts)[0], PASS,
 | 
						|
                             "accounts list validation failed")
 | 
						|
 | 
						|
            self.initialResourceCount = int(accounts[0].primarystoragetotal)
 | 
						|
 | 
						|
            primarystoragelimit = self.initialResourceCount
 | 
						|
            update_resource_limit(self.api_client, RESOURCE_PRIMARY_STORAGE, account=self.account.name, domainid=self.account.domainid, max=primarystoragelimit)
 | 
						|
 | 
						|
        except Exception as e:
 | 
						|
            return [FAIL, e]
 | 
						|
        return [PASS, None]
 | 
						|
 | 
						|
    # @data(USER_ACCOUNT)
 | 
						|
    @attr(tags=["advanced","basic"], required_hardware="true")
 | 
						|
    def test_attach_volume_exceeding_primary_limits(self):
 | 
						|
        """
 | 
						|
        # do
 | 
						|
        # 1. create a normal user account and update primary store limits to the current resource count
 | 
						|
        # 2. Upload a volume of any size
 | 
						|
        # 3. Verify that upload volume succeeds
 | 
						|
        # 4. Verify that primary storage count doesnt change
 | 
						|
        # 6. Try attaching volume to VM and verify that the attach fails (as the resource limits exceed)
 | 
						|
        # 7. Verify that primary storage count doesnt change
 | 
						|
        # done
 | 
						|
        """
 | 
						|
        # create an account, launch a vm with default template and custom disk offering, update the primary store limits to the current primary store resource count
 | 
						|
        response = self.setupNormalAccount()
 | 
						|
        self.assertEqual(response[0], PASS, response[1])
 | 
						|
 | 
						|
        # upload volume and verify that the volume is uploaded
 | 
						|
        volume = Volume.upload(self.apiclient, self.services["configurableData"]["upload_volume"],
 | 
						|
                               zoneid=self.zone.id, account=self.account.name,
 | 
						|
                               domainid=self.account.domainid)
 | 
						|
 | 
						|
        volume.wait_for_upload(self.apiclient)
 | 
						|
        volumes = Volume.list(self.apiclient, id=volume.id,
 | 
						|
                              zoneid=self.zone.id, listall=True)
 | 
						|
        validationresult = validateList(volumes)
 | 
						|
        assert validationresult[0] == PASS, "volumes list validation failed: %s" % validationresult[2]
 | 
						|
        assert str(volumes[0].state).lower() == "uploaded", "Volume state should be 'uploaded' but it is %s" % volumes[0].state
 | 
						|
 | 
						|
        # verify that the resource count didnt change due to upload volume
 | 
						|
        response = matchResourceCount(
 | 
						|
            self.apiclient, self.initialResourceCount,
 | 
						|
            RESOURCE_PRIMARY_STORAGE,
 | 
						|
            accountid=self.account.id)
 | 
						|
        self.assertEqual(response[0], PASS, response[1])
 | 
						|
 | 
						|
        # attach the above volume to the vm
 | 
						|
        try:
 | 
						|
            self.virtualMachine.attach_volume(self.apiclient, volume=volume)
 | 
						|
        except Exception as e:
 | 
						|
            if "Maximum number of resources of type \'primary_storage\' for account name="+self.account.name in e.message:
 | 
						|
                self.assertTrue(True, "there should be primary store resource limit reached exception")
 | 
						|
            else:
 | 
						|
                self.fail("only resource limit reached exception is expected. some other exception occurred. Failing the test case.")
 | 
						|
 | 
						|
        # resource count should match as the attach should fail due to reaching resource limits
 | 
						|
        response = matchResourceCount(
 | 
						|
                self.apiclient, self.initialResourceCount,
 | 
						|
                RESOURCE_PRIMARY_STORAGE,
 | 
						|
                accountid=self.account.id)
 | 
						|
        self.assertEqual(response[0], PASS, response[1])
 | 
						|
 | 
						|
        return
 | 
						|
 |