| # 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. |
| |
| from nose.plugins.attrib import attr |
| |
| # Import Local Modules |
| from marvin.cloudstackTestCase import cloudstackTestCase |
| from marvin.lib.base import (Domain, |
| Account, |
| ServiceOffering, |
| VirtualMachine, updateResourceCount) |
| from marvin.lib.common import (get_zone, |
| get_test_template) |
| from marvin.lib.utils import (cleanup_resources) |
| |
| |
| class Services: |
| """Test Account Services |
| """ |
| |
| def __init__(self): |
| self.services = { |
| "domain": { |
| "name": "Domain", |
| }, |
| "account": { |
| "email": "test@test.com", |
| "firstname": "Test", |
| "lastname": "User", |
| "username": "test", |
| # Random characters are appended for unique |
| # username |
| "password": "fr3sca", |
| }, |
| "user": { |
| "email": "user@test.com", |
| "firstname": "User", |
| "lastname": "User", |
| "username": "User", |
| # Random characters are appended for unique |
| # username |
| "password": "fr3sca", |
| }, |
| "service_offering_it_1": { |
| "name": "InstanceType-1", |
| "displaytext": "Tiny Instance", |
| "cpunumber": 1, |
| "cpuspeed": 100, |
| "memory": 128, |
| }, |
| "service_offering_it_2": { |
| "name": "InstanceType-2", |
| "displaytext": "Tiny Instance", |
| "cpunumber": 2, |
| "cpuspeed": 100, |
| "memory": 512, |
| }, |
| "virtual_machine_1": { |
| "displayname": "Test VM1", |
| "username": "root", |
| "password": "password", |
| "ssh_port": 22, |
| "hypervisor": 'XenServer', |
| "privateport": 22, |
| "publicport": 22, |
| "protocol": 'TCP', |
| }, |
| "virtual_machine_2": { |
| "displayname": "Test VM2", |
| "username": "root", |
| "password": "password", |
| "ssh_port": 22, |
| "hypervisor": 'XenServer', |
| "privateport": 22, |
| "publicport": 22, |
| "protocol": 'TCP', |
| }, |
| "template": { |
| "displaytext": "Public Template", |
| "name": "Public template", |
| "ostype": 'CentOS 5.6 (64-bit)', |
| "url": "", |
| "hypervisor": '', |
| "format": '', |
| "isfeatured": True, |
| "ispublic": True, |
| "isextractable": True, |
| "templatefilter": "self" |
| }, |
| "natrule": { |
| "publicport": 22, |
| "privateport": 22, |
| "protocol": 'TCP', |
| }, |
| "ostype": 'CentOS 5.6 (64-bit)', |
| "sleep": 60, |
| "timeout": 10, |
| } |
| |
| |
| class TestRAMCPUResourceAccounting(cloudstackTestCase): |
| |
| @classmethod |
| def setUpClass(cls): |
| cls.testClient = super( |
| TestRAMCPUResourceAccounting, |
| cls).getClsTestClient() |
| cls.api_client = cls.testClient.getApiClient() |
| |
| cls.services = Services().services |
| cls.zone = get_zone(cls.api_client, cls.testClient.getZoneForTests()) |
| cls.services['mode'] = cls.zone.networktype |
| cls.hypervisor = cls.testClient.getHypervisorInfo() |
| # Create an account, domain etc |
| cls.domain = Domain.create( |
| cls.api_client, |
| cls.services["domain"], |
| ) |
| cls.account = Account.create( |
| cls.api_client, |
| cls.services["account"], |
| admin=True, |
| domainid=cls.domain.id |
| ) |
| |
| cls.template = get_test_template( |
| cls.api_client, |
| cls.zone.id, |
| cls.hypervisor) |
| |
| cls.services["virtual_machine_1"]["zoneid"] = cls.zone.id |
| cls.services["virtual_machine_1"]["template"] = cls.template.id |
| |
| cls.services["virtual_machine_2"]["zoneid"] = cls.zone.id |
| cls.services["virtual_machine_2"]["template"] = cls.template.id |
| |
| cls._cleanup = [ |
| cls.account, |
| cls.domain |
| ] |
| |
| @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) |
| |
| def setUp(self): |
| self.apiclient = self.testClient.getApiClient() |
| self.dbclient = self.testClient.getDbConnection() |
| self.cleanup = [] |
| |
| def tearDown(self): |
| try: |
| # Clean up, terminate the created accounts |
| cleanup_resources(self.apiclient, self.cleanup) |
| except Exception as e: |
| raise Exception("Warning: Exception during cleanup : %s" % e) |
| |
| def get_resource_amount(self, resource_type): |
| cmd = updateResourceCount.updateResourceCountCmd() |
| cmd.account = self.account.name |
| cmd.domainid = self.domain.id |
| cmd.resourcetype = resource_type |
| response = self.apiclient.updateResourceCount(cmd) |
| amount = response[0].resourcecount |
| return amount |
| |
| @attr(tags=["advanced", "basic"], required_hardware="false") |
| def test_01_so_removal_resource_update(self): |
| |
| self.service_offering_it_1 = ServiceOffering.create( |
| self.api_client, |
| self.services["service_offering_it_1"], |
| domainid=self.domain.id |
| ) |
| |
| self.cleanup.append(self.service_offering_it_1) |
| |
| self.service_offering_it_2 = ServiceOffering.create( |
| self.api_client, |
| self.services["service_offering_it_2"], |
| domainid=self.domain.id |
| ) |
| |
| self.cleanup.append(self.service_offering_it_2) |
| |
| vm_1 = VirtualMachine.create( |
| self.apiclient, |
| self.services["virtual_machine_1"], |
| accountid=self.account.name, |
| domainid=self.account.domainid, |
| serviceofferingid=self.service_offering_it_1.id |
| ) |
| |
| self.debug("Deployed VM in account: %s, ID: %s" % (self.account.name, vm_1.id)) |
| self.cleanup.append(vm_1) |
| |
| vm_2 = VirtualMachine.create( |
| self.apiclient, |
| self.services["virtual_machine_2"], |
| accountid=self.account.name, |
| domainid=self.account.domainid, |
| serviceofferingid=self.service_offering_it_2.id |
| ) |
| |
| self.debug("Deployed VM in account: %s, ID: %s" % (self.account.name, vm_2.id)) |
| self.cleanup.append(vm_2) |
| |
| CPU_RESOURCE_ID = 8 |
| RAM_RESOURCE_ID = 9 |
| |
| cores = int(self.get_resource_amount(CPU_RESOURCE_ID)) |
| ram = int(self.get_resource_amount(RAM_RESOURCE_ID)) |
| |
| self.assertEqual(cores, self.services['service_offering_it_1']['cpunumber'] + self.services['service_offering_it_2']['cpunumber']) |
| self.assertEqual(ram, self.services['service_offering_it_1']['memory'] + self.services['service_offering_it_2']['memory']) |
| |
| self.service_offering_it_2.delete(self.apiclient) |
| |
| self.cleanup = self.cleanup[0:-1] |
| |
| cores = int(self.get_resource_amount(CPU_RESOURCE_ID)) |
| ram = int(self.get_resource_amount(RAM_RESOURCE_ID)) |
| |
| self.assertEqual(cores, self.services['service_offering_it_1']['cpunumber'] + self.services['service_offering_it_2']['cpunumber']) |
| self.assertEqual(ram, self.services['service_offering_it_1']['memory'] + self.services['service_offering_it_2']['memory']) |
