# 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'])