| # 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 doesn't change | |
| # 6. Try attaching volume to VM and verify that the attach fails (as the resource limits exceed) | |
| # 7. Verify that primary storage count doesn't 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 |