Return-Path: X-Original-To: apmail-cloudstack-commits-archive@www.apache.org Delivered-To: apmail-cloudstack-commits-archive@www.apache.org Received: from mail.apache.org (hermes.apache.org [140.211.11.3]) by minotaur.apache.org (Postfix) with SMTP id 08B4A10C9B for ; Fri, 28 Jun 2013 19:22:27 +0000 (UTC) Received: (qmail 98045 invoked by uid 500); 28 Jun 2013 19:22:23 -0000 Delivered-To: apmail-cloudstack-commits-archive@cloudstack.apache.org Received: (qmail 97982 invoked by uid 500); 28 Jun 2013 19:22:23 -0000 Mailing-List: contact commits-help@cloudstack.apache.org; run by ezmlm Precedence: bulk List-Help: List-Unsubscribe: List-Post: List-Id: Reply-To: dev@cloudstack.apache.org Delivered-To: mailing list commits@cloudstack.apache.org Received: (qmail 97826 invoked by uid 99); 28 Jun 2013 19:22:23 -0000 Received: from tyr.zones.apache.org (HELO tyr.zones.apache.org) (140.211.11.114) by apache.org (qpsmtpd/0.29) with ESMTP; Fri, 28 Jun 2013 19:22:23 +0000 Received: by tyr.zones.apache.org (Postfix, from userid 65534) id F25A43AB5C; Fri, 28 Jun 2013 19:22:22 +0000 (UTC) Content-Type: text/plain; charset="us-ascii" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit From: tsp@apache.org To: commits@cloudstack.apache.org Date: Fri, 28 Jun 2013 19:22:37 -0000 Message-Id: In-Reply-To: References: X-Mailer: ASF-Git Admin Mailer Subject: [16/29] git commit: updated refs/heads/master-6-17-stable to 93712c2 CLOUDSTACK-2254: Tests for Affinity and Anti Affinity rules TestCreateAffinityGroup TestListAffinityGroups TestDeleteAffinityGroups TestUpdateVMAffinityGroups TestDeployVMAffinityGroups TestAffinityGroupsAdminUser Signed-off-by: Prasanna Santhanam (cherry picked from commit 3c0c6cb0b3c21d08d88763b229c17f4b41c74145) Project: http://git-wip-us.apache.org/repos/asf/cloudstack/repo Commit: http://git-wip-us.apache.org/repos/asf/cloudstack/commit/7bad7327 Tree: http://git-wip-us.apache.org/repos/asf/cloudstack/tree/7bad7327 Diff: http://git-wip-us.apache.org/repos/asf/cloudstack/diff/7bad7327 Branch: refs/heads/master-6-17-stable Commit: 7bad7327236366b662518ed644aaad6fa4a3ac6b Parents: 2c87763 Author: Girish Shilamkar Authored: Tue May 21 13:23:10 2013 +0530 Committer: Prasanna Santhanam Committed: Sat Jun 29 00:52:08 2013 +0530 ---------------------------------------------------------------------- .../component/test_affinity_groups.py | 1547 ++++++++++++++++++ tools/marvin/marvin/integration/lib/base.py | 38 +- 2 files changed, 1572 insertions(+), 13 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/cloudstack/blob/7bad7327/test/integration/component/test_affinity_groups.py ---------------------------------------------------------------------- diff --git a/test/integration/component/test_affinity_groups.py b/test/integration/component/test_affinity_groups.py new file mode 100644 index 0000000..79e35e0 --- /dev/null +++ b/test/integration/component/test_affinity_groups.py @@ -0,0 +1,1547 @@ +#!/usr/bin/env 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. +from marvin.cloudstackTestCase import * +from marvin.cloudstackAPI import * +from marvin.integration.lib.utils import * +from marvin.integration.lib.base import * +from marvin.integration.lib.common import * +from marvin import remoteSSHClient +from nose.plugins.attrib import attr + +class Services: + """Test Account Services + """ + + def __init__(self): + self.services = { + "domain": { + "name": "Domain", + }, + "account": { + "email": "newtest@test.com", + "firstname": "Test", + "lastname": "User", + "username": "test", + # Random characters are appended for unique + # username + "password": "password", + }, + "service_offering": { + "name": "Tiny Instance", + "displaytext": "Tiny Instance", + "cpunumber": 1, + "cpuspeed": 100, + # in MHz + "memory": 64, + # In MBs + }, + "ostype": 'CentOS 5.3 (64-bit)', + "host_anti_affinity_0": { + "name": "TestAffGrp_HA_0", + "type": "host anti-affinity", + }, + "host_anti_affinity_1": { + "name": "TestAffGrp_HA_1", + "type": "host anti-affinity", + }, + "virtual_machine" : { + "hypervisor" : "KVM", + }, + "new_domain": { + "name": "New Domain", + }, + "new_account": { + "email": "domain@test.com", + "firstname": "Domain", + "lastname": "Admin", + "username": "do_admin", + # Random characters are appended for unique + # username + "password": "password", + }, + "new_account1": { + "email": "user@test.com", + "firstname": "User", + "lastname": "User", + "username": "user", + # Random characters are appended for unique + # username + "password": "password", + }, + + } + +class TestCreateAffinityGroup(cloudstackTestCase): + """ + Test various scenarios for Create Affinity Group API + """ + + @classmethod + def setUpClass(cls): + + cls.api_client = super(TestCreateAffinityGroup, cls).getClsTestClient().getApiClient() + cls.services = Services().services + # Get Zone, Domain and templates + cls.domain = get_domain(cls.api_client, cls.services) + cls.zone = get_zone(cls.api_client, cls.services) + cls.template = get_template( + cls.api_client, + cls.zone.id, + cls.services["ostype"] + ) + cls.services["virtual_machine"]["zoneid"] = cls.zone.id + + cls.services["template"] = cls.template.id + cls.services["zoneid"] = cls.zone.id + + cls.account = Account.create( + cls.api_client, + cls.services["account"], + domainid=cls.domain.id + ) + + cls.services["account"] = cls.account.name + + cls.service_offering = ServiceOffering.create( + cls.api_client, + cls.services["service_offering"] + ) + + cls._cleanup = [ + cls.service_offering, + cls.account, + ] + return + + def setUp(self): + self.apiclient = self.testClient.getApiClient() + self.dbclient = self.testClient.getDbConnection() + self.cleanup = [] + + def tearDown(self): + try: + # Clean up, terminate the created instance, volumes and snapshots + cleanup_resources(self.apiclient, self.cleanup) + except Exception as e: + raise Exception("Warning: Exception during cleanup : %s" % e) + return + + @classmethod + def tearDownClass(cls): + try: + cls.api_client = super(TestCreateAffinityGroup, cls).getClsTestClient().getApiClient() + #Clean up, terminate the created templates + cleanup_resources(cls.api_client, cls._cleanup) + except Exception as e: + raise Exception("Warning: Exception during cleanup : %s" % e) + + def create_aff_grp(self, api_client=None, aff_grp=None, + acc=None, domainid=None): + + if api_client == None: + api_client = self.api_client + if aff_grp == None: + self.services["host_anti_affinity_0"] + if acc == None: + acc = self.account.name + if domainid == None: + domainid = self.domain.id + + try: + self.aff_grp = AffinityGroup.create(api_client, aff_grp, acc, domainid) + except Exception as e: + raise Exception("Error: Creation of Affinity Group failed : %s" %e) + + @attr(tags=["simulator", "basic", "advanced"]) + def test_01_admin_create_aff_grp(self): + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + self.debug("Created Affinity Group: %s" %self.aff_grp.name) + + list_aff_grps = AffinityGroup.list(self.api_client) + AffinityGroup.delete(self.api_client, list_aff_grps[0].name) + self.debug("Deleted Affinity Group: %s" %list_aff_grps[0].name) + + @attr(tags=["simulator", "basic", "advanced"]) + def test_02_doadmin_create_aff_grp(self): + + self.new_domain = Domain.create(self.api_client, self.services["new_domain"]) + self.do_admin = Account.create(self.api_client, self.services["new_account"], + admin=True, domainid=self.new_domain.id) + self.cleanup.append(self.do_admin) + self.cleanup.append(self.new_domain) + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"], + acc=self.do_admin.name, domainid=self.new_domain.id) + + AffinityGroup.delete(self.api_client, name=self.aff_grp.name, + account=self.do_admin.name, domainid=self.new_domain.id) + self.debug("Deleted Affinity Group: %s" %self.aff_grp.name) + + + @attr(tags=["simulator", "basic", "advanced"]) + def test_03_user_create_aff_grp(self): + + self.user = Account.create(self.api_client, self.services["new_account"], + domainid=self.domain.id) + + self.cleanup.append(self.user) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"], + acc=self.user.name, domainid=self.domain.id) + + AffinityGroup.delete(self.api_client, name=self.aff_grp.name, + account=self.user.name, domainid=self.domain.id) + self.debug("Deleted Affinity Group: %s" %self.aff_grp.name) + + + @attr(tags=["simulator", "basic", "advanced"]) + def test_04_user_create_aff_grp_existing_name(self): + + self.user = Account.create(self.api_client, self.services["new_account"], + domainid=self.domain.id) + + self.cleanup.append(self.user) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"], + acc=self.user.name, domainid=self.domain.id) + with self.assertRaises(Exception): + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"], + acc=self.user.name, domainid=self.domain.id) + + AffinityGroup.delete(self.api_client, name=self.aff_grp.name, + account=self.user.name, domainid=self.domain.id) + self.debug("Deleted Affinity Group: %s" %self.aff_grp.name) + + @attr(tags=["simulator", "basic", "advanced"]) + def test_05_create_aff_grp_same_name_diff_acc(self): + + self.user = Account.create(self.api_client, self.services["new_account"], + domainid=self.domain.id) + + self.cleanup.append(self.user) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"], + acc=self.user.name, domainid=self.domain.id) + + try: + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + except Exception: + self.debug("Error: Creating affinity group with same name from different account failed.") + + AffinityGroup.delete(self.api_client, name=self.aff_grp.name, + account=self.user.name, domainid=self.domain.id) + self.debug("Deleted Affinity Group: %s" %self.aff_grp.name) + + @attr(tags=["simulator", "basic", "advanced"]) + def test_06_create_aff_grp_nonexisting_type(self): + + self.non_existing_aff_grp = { + "name": "TestAffGrp_HA", + "type": "Incorrect type", + } + with self.assertRaises(Exception): + self.create_aff_grp(aff_grp=self.non_existing_aff_grp) + +class TestListAffinityGroups(cloudstackTestCase): + + @classmethod + def setUpClass(cls): + + cls.api_client = super(TestListAffinityGroups, cls).getClsTestClient().getApiClient() + cls.services = Services().services + # Get Zone, Domain and templates + cls.domain = get_domain(cls.api_client, cls.services) + cls.zone = get_zone(cls.api_client, cls.services) + cls.template = get_template( + cls.api_client, + cls.zone.id, + cls.services["ostype"] + ) + cls.services["virtual_machine"]["zoneid"] = cls.zone.id + + cls.services["template"] = cls.template.id + cls.services["zoneid"] = cls.zone.id + + cls.account = Account.create( + cls.api_client, + cls.services["account"], + domainid=cls.domain.id + ) + + cls.services["account"] = cls.account.name + + cls.service_offering = ServiceOffering.create( + cls.api_client, + cls.services["service_offering"] + ) + + cls.__cleanup = [ + cls.service_offering, + cls.account, + ] + + # Create multiple Affinity Groups + return + + def setUp(self): + self.apiclient = self.testClient.getApiClient() + self.dbclient = self.testClient.getDbConnection() + self.aff_grp = [] + self.cleanup = [] + + def tearDown(self): + try: + cls.api_client = super(TestListAffinityGroups, cls).getClsTestClient().getApiClient() + #Clean up, terminate the created templates + cleanup_resources(cls.api_client, cls.cleanup) + except Exception as e: + raise Exception("Warning: Exception during cleanup : %s" % e) + + @classmethod + def tearDownClass(cls): + + try: + cls.api_client = super(TestListAffinityGroups, cls).getClsTestClient().getApiClient() + #Clean up, terminate the created templates + cleanup_resources(cls.api_client, cls.__cleanup) + except Exception as e: + raise Exception("Warning: Exception during cleanup : %s" % e) + + def create_aff_grp(self, api_client=None, aff_grp=None, + acc=None, domainid=None): + + if api_client == None: + api_client = self.api_client + if aff_grp == None: + self.services["host_anti_affinity_0"] + #if acc == None: + # acc = self.account.name + #if domainid == None: + # domainid = self.domain.id + + try: + self.aff_grp.append(AffinityGroup.create(api_client, + aff_grp, acc, domainid)) + except Exception as e: + raise Exception("Error: Creation of Affinity Group failed : %s" %e) + + def create_vm_in_aff_grps(self, ag_list): + #try: + self.debug('Creating VM in AffinityGroup=%s' % ag_list[0]) + vm = VirtualMachine.create( + self.api_client, + self.services["virtual_machine"], + templateid=self.template.id, + #accountid=self.account.name, + #domainid=self.account.domainid, + serviceofferingid=self.service_offering.id, + affinitygroupnames=ag_list + ) + self.debug('Created VM=%s in Affinity Group=%s' % + (vm.id, ag_list[0])) + #except Exception: + #self.debug('Unable to create VM in a Affinity Group=%s' + # % ag_list[0]) + + list_vm = list_virtual_machines(self.api_client, id=vm.id) + + self.assertEqual(isinstance(list_vm, list), True, + "Check list response returns a valid list") + self.assertNotEqual(len(list_vm),0, + "Check VM available in List Virtual Machines") + + vm_response = list_vm[0] + self.assertEqual(vm_response.state, 'Running', + msg="VM is not in Running state") + + return vm, vm_response.hostid + + def test_01_list_aff_grps_for_vm(self): + """ + List affinity group for a vm + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + list_aff_grps = AffinityGroup.list(self.api_client) + + vm, hostid = self.create_vm_in_aff_grps([self.aff_grp[0].name]) + list_aff_grps = AffinityGroup.list(self.api_client, + virtualmachineid=vm.id) + + self.assertEqual(list_aff_grps[0].name, self.aff_grp[0].name, + "Listing Affinity Group by VM id failed") + + vm.delete(self.api_client) + #Wait for expunge interval to cleanup VM + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + + AffinityGroup.delete(self.api_client, self.aff_grp[0].name) + + def test_02_list_multiple_aff_grps_for_vm(self): + """ + List multiple affinity groups associated with a vm + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_1"]) + + aff_grps_names = [self.aff_grp[0].name, self.aff_grp[1].name] + vm, hostid = self.create_vm_in_aff_grps(aff_grps_names) + list_aff_grps = AffinityGroup.list(self.api_client, + virtualmachineid=vm.id) + + list_aff_grps_names = [list_aff_grps[0].name, list_aff_grps[1].name] + + aff_grps_names.sort() + list_aff_grps_names.sort() + self.assertEqual(aff_grps_names, list_aff_grps_names, + "One of the Affinity Groups is missing %s" + %list_aff_grps_names) + + vm.delete(self.api_client) + #Wait for expunge interval to cleanup VM + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + + for i in aff_grps_names: + AffinityGroup.delete(self.api_client, i) + + def test_03_list_aff_grps_by_id(self): + """ + List affinity groups by id + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + print self.aff_grp[0].__dict__ + list_aff_grps = AffinityGroup.list(self.api_client) + list_aff_grps = AffinityGroup.list(self.api_client, id=list_aff_grps[0].id) + self.assertEqual(list_aff_grps[0].name, self.aff_grp[0].name, + "Listing Affinity Group by VM id failed") + + AffinityGroup.delete(self.api_client, self.aff_grp[0].name) + + def test_04_list_aff_grps_by_name(self): + """ + List Affinity Groups by name + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + list_aff_grps = AffinityGroup.list(self.api_client, + name=self.aff_grp[0].name) + self.assertEqual(list_aff_grps[0].name, self.aff_grp[0].name, + "Listing Affinity Group by name failed") + + AffinityGroup.delete(self.api_client, self.aff_grp[0].name) + + def test_05_list_aff_grps_by_non_existing_id(self): + """ + List Affinity Groups by non-existing id + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + list_aff_grps = AffinityGroup.list(self.api_client, + id=1234) + self.assertEqual(list_aff_grps, None, + "Listing Affinity Group by non-existing id succeeded.") + + AffinityGroup.delete(self.api_client, self.aff_grp[0].name) + + def test_06_list_aff_grps_by_non_existing_name(self): + """ + List Affinity Groups by non-existing name + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + list_aff_grps = AffinityGroup.list(self.api_client, + name="NonexistingName") + self.assertEqual(list_aff_grps, None, + "Listing Affinity Group by non-existing name succeeded.") + + AffinityGroup.delete(self.api_client, self.aff_grp[0].name) + +class TestDeleteAffinityGroups(cloudstackTestCase): + + @classmethod + def setUpClass(cls): + + cls.api_client = super(TestDeleteAffinityGroups, cls).getClsTestClient().getApiClient() + cls.services = Services().services + # Get Zone, Domain and templates + cls.domain = get_domain(cls.api_client, cls.services) + cls.zone = get_zone(cls.api_client, cls.services) + cls.template = get_template( + cls.api_client, + cls.zone.id, + cls.services["ostype"] + ) + cls.services["virtual_machine"]["zoneid"] = cls.zone.id + + cls.services["template"] = cls.template.id + cls.services["zoneid"] = cls.zone.id + + cls.account = Account.create( + cls.api_client, + cls.services["account"], + domainid=cls.domain.id + ) + + cls.services["account"] = cls.account.name + + cls.service_offering = ServiceOffering.create( + cls.api_client, + cls.services["service_offering"] + ) + + cls.__cleanup = [ + cls.service_offering, + cls.account, + ] + + # Create multiple Affinity Groups + return + + def setUp(self): + self.apiclient = self.testClient.getApiClient() + self.dbclient = self.testClient.getDbConnection() + self.aff_grp = [] + self.cleanup = [] + + def tearDown(self): + try: + self.api_client = super(TestDeleteAffinityGroups,self).getClsTestClient().getApiClient() + #Clean up, terminate the created templates + cleanup_resources(self.api_client, self.cleanup) + except Exception as e: + raise Exception("Warning: Exception during cleanup : %s" % e) + + @classmethod + def tearDownClass(cls): + + try: + cls.api_client = super(TestDeleteAffinityGroups, cls).getClsTestClient().getApiClient() + #Clean up, terminate the created templates + cleanup_resources(cls.api_client, cls.__cleanup) + except Exception as e: + raise Exception("Warning: Exception during cleanup : %s" % e) + + def create_aff_grp(self, api_client=None, aff_grp=None, + acc=None, domainid=None): + + if api_client == None: + api_client = self.api_client + if aff_grp == None: + self.services["host_anti_affinity_0"] + #if acc == None: + # acc = self.account.name + #if domainid == None: + # domainid = self.domain.id + + try: + self.aff_grp.append(AffinityGroup.create(api_client, + aff_grp, acc, domainid)) + except Exception as e: + raise Exception("Error: Creation of Affinity Group failed : %s" %e) + + def create_vm_in_aff_grps(self, ag_list): + #try: + self.debug('Creating VM in AffinityGroup=%s' % ag_list[0]) + vm = VirtualMachine.create( + self.api_client, + self.services["virtual_machine"], + templateid=self.template.id, + #accountid=self.account.name, + #domainid=self.account.domainid, + serviceofferingid=self.service_offering.id, + affinitygroupnames=ag_list + ) + self.debug('Created VM=%s in Affinity Group=%s' % + (vm.id, ag_list[0])) + #except Exception: + #self.debug('Unable to create VM in a Affinity Group=%s' + # % ag_list[0]) + + list_vm = list_virtual_machines(self.api_client, id=vm.id) + + self.assertEqual(isinstance(list_vm, list), True, + "Check list response returns a valid list") + self.assertNotEqual(len(list_vm),0, + "Check VM available in Delete Virtual Machines") + + vm_response = list_vm[0] + self.assertEqual(vm_response.state, 'Running', + msg="VM is not in Running state") + + return vm, vm_response.hostid + + def test_01_delete_aff_grp_by_id(self): + """ + Delete Afifnity Group by id. + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_1"]) + + list_aff_grps = AffinityGroup.list(self.api_client, + name=self.aff_grp[0].name) + + AffinityGroup.delete(self.api_client, id=list_aff_grps[0].id) + + AffinityGroup.delete(self.api_client, name=self.aff_grp[1].name) + + def test_02_delete_aff_grp_for_acc(self): + """ + Delete Afifnity Group for an account. + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"], + acc=self.account.name, domainid=self.domain.id) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_1"], + acc=self.account.name, domainid=self.domain.id) + + AffinityGroup.delete(self.api_client, account=self.account.name, + domainid=self.domain.id, name=self.aff_grp[0].name) + + with self.assertRaises(Exception): + vm, hostid = self.create_vm_in_aff_grps([self.aff_grp[0].name]) + + AffinityGroup.delete(self.api_client, account=self.account.name, + domainid=self.domain.id, name=self.aff_grp[1].name) + + def test_03_delete_aff_grp_with_vms(self): + """ + Delete Afifnity Group which has vms in it. + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"], + acc=self.account.name, domainid=self.domain.id) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_1"], + acc=self.account.name, domainid=self.domain.id) + + vm, hostid = self.create_vm_in_aff_grps([self.aff_grp[0].name, + self.aff_grp[1].name]) + + AffinityGroup.delete(self.api_client, account=self.account.name, + domainid=self.domain.id, name=self.aff_grp[0].name) + + vm_list = list_virtual_machines(self.apiclient, + id=self.virtual_machine.id) + + + vm.delete(self.api_client) + #Wait for expunge interval to cleanup VM + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + + AffinityGroup.delete(self.api_client, account=self.account.name, + domainid=self.domain.id, name=self.aff_grp[0].name) + AffinityGroup.delete(self.api_client, account=self.account.name, + domainid=self.domain.id, name=self.aff_grp[1].name) + + def test_04_delete_aff_grp_with_vms(self): + """ + Delete Affinity Group which has after updating affinity group for + vms in it. + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + + vm1, hostid1 = self.create_vm_in_aff_grps([self.aff_grp[0].name]) + vm2, hostid2 = self.create_vm_in_aff_grps([self.aff_grp[0].name]) + + with self.assertRaises(Exception): + AffinityGroup.delete(self.api_client, name=self.aff_grp[0].name) + + vm1.update_affinity_group(self.api_client, affinitygroupnames=[]) + + with self.assertRaises(Exception): + AffinityGroup.delete(self.api_client, name=self.aff_grp[0].name) + + vm2.update_affinity_group(self.api_client, affinitygroupnames=[]) + + AffinityGroup.delete(self.api_client, name=self.aff_grp[0].name) + + vm1.delete(self.api_client) + vm2.delete(self.api_client) + #Wait for expunge interval to cleanup VM + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + + AffinityGroup.delete(self.api_client, name=self.aff_grp[1].name) + + def test_05_delete_aff_grp_id(self): + """ + Delete Affinity Group with id which does not belong to this user + """ + + self.user1 = Account.create(self.api_client, + self.services["new_account"]) + + self.cleanup.append(self.user1) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"], + acc=self.user1.name, + domainid=self.domain.id) + + self.user2 = Account.create(self.apiclient, self.services["new_account1"]) + self.cleanup.append(self.user2) + + userapiclient = self.testClient.createUserApiClient( + UserName=self.user2.name, + DomainName=self.user2.domain, + acctType=0) + + self.create_aff_grp(api_client=userapiclient, + aff_grp=self.services["host_anti_affinity_1"]) + + list_aff_grps = AffinityGroup.list(self.api_client, + name=self.aff_grp[0].name) + + # Delete Affinity group belonging to different user by id + with self.assertRaises(Exception): + AffinityGroup.delete(userapiclient, name=list_aff_grps.id) + + #Cleanup + AffinityGroup.delete(self.api_client, name=self.aff_grp[0].name) + AffinityGroup.delete(userapiclient, name=self.aff_grp[1].name) + + def test_06_delete_aff_grp_name(self): + """ + Delete Affinity Group by name which does not belong to this user + """ + + self.user1 = Account.create(self.api_client, + self.services["new_account"]) + + self.cleanup.append(self.user1) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"], + acc=self.user1.name, + domainid=self.domain.id) + + self.user2 = Account.create(self.apiclient, self.services["new_account1"]) + self.cleanup.append(self.user2) + + userapiclient = self.testClient.createUserApiClient( + UserName=self.user2.name, + DomainName=self.user2.domain, + acctType=0) + + self.create_aff_grp(api_client=userapiclient, + aff_grp=self.services["host_anti_affinity_1"]) + + # Delete Affinity group belonging to different user by name + with self.assertRaises(Exception): + AffinityGroup.delete(userapiclient, name=self.aff_grp[0].name) + + #Cleanup + AffinityGroup.delete(self.api_client, name=self.aff_grp[0].name) + AffinityGroup.delete(userapiclient, name=self.aff_grp[1].name) + +class TestUpdateVMAffinityGroups(cloudstackTestCase): + + @classmethod + def setUpClass(cls): + + cls.api_client = super(TestUpdateVMAffinityGroups, cls).getClsTestClient().getApiClient() + cls.services = Services().services + # Get Zone, Domain and templates + cls.domain = get_domain(cls.api_client, cls.services) + cls.zone = get_zone(cls.api_client, cls.services) + cls.template = get_template( + cls.api_client, + cls.zone.id, + cls.services["ostype"] + ) + cls.services["virtual_machine"]["zoneid"] = cls.zone.id + + cls.services["template"] = cls.template.id + cls.services["zoneid"] = cls.zone.id + + cls.account = Account.create( + cls.api_client, + cls.services["account"], + domainid=cls.domain.id + ) + + cls.services["account"] = cls.account.name + + cls.service_offering = ServiceOffering.create( + cls.api_client, + cls.services["service_offering"] + ) + + cls.__cleanup = [ + cls.service_offering, + cls.account, + ] + + # Create multiple Affinity Groups + return + + def setUp(self): + self.apiclient = self.testClient.getApiClient() + self.dbclient = self.testClient.getDbConnection() + self.aff_grp = [] + self.cleanup = [] + + def tearDown(self): + try: + self.api_client = super(TestUpdateVMAffinityGroups,self).getClsTestClient().getApiClient() + #Clean up, terminate the created templates + cleanup_resources(self.api_client, self.cleanup) + except Exception as e: + raise Exception("Warning: Exception during cleanup : %s" % e) + + @classmethod + def tearDownClass(cls): + + try: + cls.api_client = super(TestUpdateVMAffinityGroups, cls).getClsTestClient().getApiClient() + #Clean up, terminate the created templates + cleanup_resources(cls.api_client, cls.__cleanup) + except Exception as e: + raise Exception("Warning: Exception during cleanup : %s" % e) + + def create_aff_grp(self, api_client=None, aff_grp=None, + acc=None, domainid=None): + + if api_client == None: + api_client = self.api_client + if aff_grp == None: + self.services["host_anti_affinity_0"] + #if acc == None: + # acc = self.account.name + #if domainid == None: + # domainid = self.domain.id + + try: + self.aff_grp.append(AffinityGroup.create(api_client, + aff_grp, acc, domainid)) + except Exception as e: + raise Exception("Error: Creation of Affinity Group failed : %s" %e) + + def create_vm_in_aff_grps(self, ag_list): + #try: + self.debug('Creating VM in AffinityGroup=%s' % ag_list[0]) + vm = VirtualMachine.create( + self.api_client, + self.services["virtual_machine"], + templateid=self.template.id, + #accountid=self.account.name, + #domainid=self.account.domainid, + serviceofferingid=self.service_offering.id, + affinitygroupnames=ag_list + ) + self.debug('Created VM=%s in Affinity Group=%s' % + (vm.id, ag_list[0])) + #except Exception: + #self.debug('Unable to create VM in a Affinity Group=%s' + # % ag_list[0]) + + list_vm = list_virtual_machines(self.api_client, id=vm.id) + + self.assertEqual(isinstance(list_vm, list), True, + "Check list response returns a valid list") + self.assertNotEqual(len(list_vm),0, + "Check VM available in Delete Virtual Machines") + + vm_response = list_vm[0] + self.assertEqual(vm_response.state, 'Running', + msg="VM is not in Running state") + + return vm, vm_response.hostid + + def test_01_update_aff_grp_by_ids(self): + """ + Update the list of affinityGroups by using affinity groupids + + """ + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_1"]) + + vm1, hostid1 = self.create_vm_in_aff_grps([self.aff_grp[0].name]) + vm2, hostid2 = self.create_vm_in_aff_grps([self.aff_grp[0].name]) + + vm1.stop(self.api_client) + + list_aff_grps = AffinityGroup.list(self.api_client) + + self.assertEqual(len(list_aff_grps), 2 , "2 affinity groups should be present") + + vm1.update_affinity_group(self.api_client, + affinitygroupids=[list_aff_grps[0].id, + list_aff_grps[1].id]) + + list_aff_grps = AffinityGroup.list(self.api_client, + virtualmachineid=vm1.id) + + list_aff_grps_names = [list_aff_grps[0].name, list_aff_grps[1].name] + + aff_grps_names = [self.aff_grp[0].name, self.aff_grp[1].name] + aff_grps_names.sort() + list_aff_grps_names.sort() + self.assertEqual(aff_grps_names, list_aff_grps_names, + "One of the Affinity Groups is missing %s" + %list_aff_grps_names) + + vm1.start(self.api_client) + + vm_status = VirtualMachine.list(self.api_client, id=vm1.id) + self.assertNotEqual(vm_status[0].hostid, hostid2, "The virtual machine " + "started on host %s violating the host anti-affinity" + "rule" %vm_status[0].hostid) + + vm1.delete(self.api_client) + vm2.delete(self.api_client) + #Wait for expunge interval to cleanup VM + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + for i in aff_grps_names: + AffinityGroup.delete(self.api_client, i) + + def test_02_update_aff_grp_by_names(self): + """ + Update the list of affinityGroups by using affinity groupnames + + """ + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_1"]) + vm1, hostid1 = self.create_vm_in_aff_grps([self.aff_grp[0].name]) + vm2, hostid2 = self.create_vm_in_aff_grps([self.aff_grp[0].name]) + + vm1.stop(self.api_client) + + vm1.update_affinity_group(self.api_client, + affinitygroupnames=[self.aff_grp[0].name, + self.aff_grp[1].name]) + + list_aff_grps = AffinityGroup.list(self.api_client, + virtualmachineid=vm1.id) + + list_aff_grps_names = [list_aff_grps[0].name, list_aff_grps[1].name] + + aff_grps_names = [self.aff_grp[0].name, self.aff_grp[1].name] + aff_grps_names.sort() + list_aff_grps_names.sort() + self.assertEqual(aff_grps_names, list_aff_grps_names, + "One of the Affinity Groups is missing %s" + %list_aff_grps_names) + + vm1.start(self.api_client) + + vm_status = VirtualMachine.list(self.api_client, id=vm1.id) + self.assertNotEqual(vm_status[0].hostid, hostid2, "The virtual machine " + "started on host %s violating the host anti-affinity" + "rule" %vm_status[0].hostid) + + vm1.delete(self.api_client) + vm2.delete(self.api_client) + #Wait for expunge interval to cleanup VM + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + for i in aff_grps_names: + AffinityGroup.delete(self.api_client, i) + + def test_03_update_aff_grp_for_vm_with_no_aff_grp(self): + """ + Update the list of affinityGroups for vm which is not associated + with any affinity groups. + + """ + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_1"]) + vm1, hostid1 = self.create_vm_in_aff_grps([]) + vm2, hostid2 = self.create_vm_in_aff_grps([self.aff_grp[0].name]) + + vm1.stop(self.api_client) + + vm1.update_affinity_group(self.api_client, + affinitygroupnames=[self.aff_grp[0].name]) + + vm1.start(self.api_client) + + vm_status = VirtualMachine.list(self.api_client, id=vm1.id) + self.assertNotEqual(vm_status[0].hostid, hostid2, "The virtual machine " + "started on host %s violating the host anti-affinity" + "rule" %vm_status[0].hostid) + + vm1.delete(self.api_client) + vm2.delete(self.api_client) + #Wait for expunge interval to cleanup VM + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + aff_grps_names = [self.aff_grp[0].name, self.aff_grp[1].name] + for i in aff_grps_names: + AffinityGroup.delete(self.api_client, i) + + def test_04_update_aff_grp_remove_all(self): + """ + Update the list of Affinity Groups to empty list + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_1"]) + vm1, hostid1 = self.create_vm_in_aff_grps([self.aff_grp[0].name]) + + aff_grps_names = [self.aff_grp[0].name, self.aff_grp[1].name] + vm1.stop(self.api_client) + + vm1.update_affinity_group(self.api_client, + affinitygroupnames=[]) + + vm1.start(self.api_client) + list_aff_grps = AffinityGroup.list(self.api_client, + virtualmachineid=vm.id) + self.assertEqual(list_aff_grps, [], "The affinity groups list is not empyty") + + vm1.delete(self.api_client) + #Wait for expunge interval to cleanup VM + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + for i in aff_grps_names: + AffinityGroup.delete(self.api_client, i) + + def test_05_update_aff_grp_on_running_vm(self): + """ + Update the list of Affinity Groups on running vm + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_1"]) + vm1, hostid1 = self.create_vm_in_aff_grps([self.aff_grp[0].name]) + + aff_grps_names = [self.aff_grp[0].name, self.aff_grp[1].name] + with self.assertRaises(Exception): + vm1.update_affinity_group(self.api_client, affinitygroupnames=[]) + + vm1.delete(self.api_client) + #Wait for expunge interval to cleanup VM + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + for i in aff_grps_names: + AffinityGroup.delete(self.api_client, i) + +class TestDeployVMAffinityGroups(cloudstackTestCase): + + @classmethod + def setUpClass(cls): + + cls.api_client = super(TestDeployVMAffinityGroups, cls).getClsTestClient().getApiClient() + cls.services = Services().services + # Get Zone, Domain and templates + cls.domain = get_domain(cls.api_client, cls.services) + cls.zone = get_zone(cls.api_client, cls.services) + cls.template = get_template( + cls.api_client, + cls.zone.id, + cls.services["ostype"] + ) + cls.services["virtual_machine"]["zoneid"] = cls.zone.id + + cls.services["template"] = cls.template.id + cls.services["zoneid"] = cls.zone.id + + cls.account = Account.create( + cls.api_client, + cls.services["account"], + domainid=cls.domain.id + ) + + cls.services["account"] = cls.account.name + + cls.service_offering = ServiceOffering.create( + cls.api_client, + cls.services["service_offering"] + ) + + cls.__cleanup = [ + cls.service_offering, + cls.account, + ] + + # Create multiple Affinity Groups + return + + def setUp(self): + self.apiclient = self.testClient.getApiClient() + self.dbclient = self.testClient.getDbConnection() + self.aff_grp = [] + self.cleanup = [] + + def tearDown(self): + try: + self.api_client = super(TestDeployVMAffinityGroups,self).getClsTestClient().getApiClient() + #Clean up, terminate the created templates + cleanup_resources(self.api_client, self.cleanup) + except Exception as e: + raise Exception("Warning: Exception during cleanup : %s" % e) + + @classmethod + def tearDownClass(cls): + + try: + cls.api_client = super(TestDeployVMAffinityGroups, cls).getClsTestClient().getApiClient() + #Clean up, terminate the created templates + cleanup_resources(cls.api_client, cls.__cleanup) + except Exception as e: + raise Exception("Warning: Exception during cleanup : %s" % e) + + def create_aff_grp(self, api_client=None, aff_grp=None, + acc=None, domainid=None): + + if api_client == None: + api_client = self.api_client + if aff_grp == None: + self.services["host_anti_affinity_0"] + #if acc == None: + # acc = self.account.name + #if domainid == None: + # domainid = self.domain.id + + try: + self.aff_grp.append(AffinityGroup.create(api_client, + aff_grp, acc, domainid)) + except Exception as e: + raise Exception("Error: Creation of Affinity Group failed : %s" %e) + + def create_vm_in_aff_grps(self, api_client=None, ag_list=None, ag_ids=None): + + if api_client == None: + api_client = self.api_client + self.debug('Creating VM in AffinityGroup=%s' % ag_list) + vm = VirtualMachine.create( + api_client, + self.services["virtual_machine"], + templateid=self.template.id, + #accountid=self.account.name, + #domainid=self.account.domainid, + serviceofferingid=self.service_offering.id, + affinitygroupnames=ag_list, + affinitygroupids=ag_ids + ) + self.debug('Created VM=%s in Affinity Group=%s' % + (vm.id, ag_list)) + + list_vm = list_virtual_machines(self.api_client, id=vm.id) + + self.assertEqual(isinstance(list_vm, list), True, + "Check list response returns a valid list") + self.assertNotEqual(len(list_vm),0, + "Check VM available in Delete Virtual Machines") + + vm_response = list_vm[0] + self.assertEqual(vm_response.state, 'Running', + msg="VM is not in Running state") + + return vm, vm_response.hostid + + @attr(tags=["simulator", "basic", "advanced", "multihost"]) + def test_01_deploy_vm_without_aff_grp(self): + """ + Deploy VM without affinity group + """ + + vm1, hostid1 = self.create_vm_in_aff_grps() + + vm1.delete(self.api_client) + #Wait for expunge interval to cleanup VM + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + + @attr(tags=["simulator", "basic", "advanced", "multihost"]) + def test_02_deploy_vm_by_aff_grp_name(self): + """ + Deploy VM by aff grp name + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + vm1, hostid1 = self.create_vm_in_aff_grps([self.aff_grp[0].name]) + + vm1.delete(self.api_client) + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + AffinityGroup.delete(self.api_client, self.aff_grp[0].name) + + @attr(tags=["simulator", "basic", "advanced", "multihost"]) + def test_03_deploy_vm_by_aff_grp_id(self): + """ + Deploy VM by aff grp id + """ + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + + list_aff_grps = AffinityGroup.list(self.api_client, + name=self.aff_grp[0].name) + + vm1, hostid1 = self.create_vm_in_aff_grps(ag_ids=[list_aff_grps[0].id]) + + vm1.delete(self.api_client) + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + AffinityGroup.delete(self.api_client, self.aff_grp[0].name) + + @attr(tags=["simulator", "basic", "advanced", "multihost"]) + def test_04_deploy_vm_anti_affinity_group(self): + """ + test DeployVM in anti-affinity groups + + deploy VM1 and VM2 in the same host-anti-affinity groups + Verify that the vms are deployed on separate hosts + """ + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + vm1, hostid1 = self.create_vm_in_aff_grps(ag_list=[self.aff_grp[0].name]) + vm2, hostid2 = self.create_vm_in_aff_grps(ag_list=[self.aff_grp[0].name]) + + self.assertNotEqual(hostid1, hostid2, + msg="Both VMs of affinity group %s are on the same host" + % self.aff_grp[0].name) + + vm1.delete(self.api_client) + vm2.delete(self.api_client) + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + AffinityGroup.delete(self.api_client, self.aff_grp[0].name) + + @attr(tags=["simulator", "basic", "advanced", "multihost"]) + def test_05_deploy_vm_by_id(self): + """ + Deploy vms by affinity group id + """ + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + + list_aff_grps = AffinityGroup.list(self.api_client, + name=self.aff_grp[0].name) + + vm1, hostid1 = self.create_vm_in_aff_grps(ag_ids=[list_aff_grps[0].id]) + vm2, hostid2 = self.create_vm_in_aff_grps(ag_ids=[list_aff_grps[0].id]) + + self.assertNotEqual(hostid1, hostid2, + msg="Both VMs of affinity group %s are on the same host" + % self.aff_grp[0].name) + + vm1.delete(self.api_client) + vm2.delete(self.api_client) + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + AffinityGroup.delete(self.api_client, self.aff_grp[0].name) + + @attr(tags=["simulator", "basic", "advanced", "multihost"]) + def test_06_deploy_vm_aff_grp_of_other_user_by_name(self): + """ + Deploy vm in affinity group of another user by name + """ + + self.user1 = Account.create(self.api_client, + self.services["new_account"]) + + self.cleanup.append(self.user1) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"], + acc=self.user1.name, + domainid=self.domain.id) + + self.user2 = Account.create(self.apiclient, self.services["new_account1"]) + self.cleanup.append(self.user2) + + userapiclient = self.testClient.createUserApiClient( + UserName=self.user2.name, + DomainName=self.user2.domain, + acctType=0) + + self.create_aff_grp(api_client=userapiclient, + aff_grp=self.services["host_anti_affinity_1"]) + + with self.assertRaises(Exception): + vm1, hostid1 = self.create_vm_in_aff_grps(api_client=userapiclient, + ag_list=[self.aff_grp[0].name]) + + + AffinityGroup.delete(self.api_client, self.aff_grp[0].name) + AffinityGroup.delete(userapiclient, self.aff_grp[1].name) + + @attr(tags=["simulator", "basic", "advanced", "multihost"]) + def test_07_deploy_vm_aff_grp_of_other_user_by_id(self): + """ + Deploy vm in affinity group of another user by id + """ + + self.user1 = Account.create(self.api_client, + self.services["new_account"]) + + self.cleanup.append(self.user1) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"], + acc=self.user1.name, + domainid=self.domain.id) + + self.user2 = Account.create(self.apiclient, self.services["new_account1"]) + self.cleanup.append(self.user2) + + userapiclient = self.testClient.createUserApiClient( + UserName=self.user2.name, + DomainName=self.user2.domain, + acctType=0) + + self.create_aff_grp(api_client=userapiclient, + aff_grp=self.services["host_anti_affinity_1"]) + + list_aff_grps = AffinityGroup.list(self.api_client, + name=self.aff_grp[0].name) + + # Deploy VM in Affinity group belonging to different user by id + with self.assertRaises(Exception): + vm1, hostid1 = self.create_vm_in_aff_grps(api_client=userapiclient, + ag_ids=[list_aff_grps[0].id]) + + AffinityGroup.delete(self.api_client, self.aff_grp[0].name) + AffinityGroup.delete(userapiclient, self.aff_grp[1].name) + + @attr(tags=["simulator", "basic", "advanced", "multihost"]) + def test_08_deploy_vm_multiple_aff_grps(self): + """ + Deploy vm in multiple affinity groups + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_1"]) + vm1, hostid1 = self.create_vm_in_aff_grps(ag_list=[self.aff_grp[0].name, + self.aff_grp[1].name]) + + list_aff_grps = AffinityGroup.list(self.api_client, + virtualmachineid=vm1.id) + + aff_grps_names = [self.aff_grp[0].name, self.aff_grp[1].name] + list_aff_grps_names = [list_aff_grps[0].name, list_aff_grps[1].name] + + aff_grps_names.sort() + list_aff_grps_names.sort() + self.assertEqual(aff_grps_names, list_aff_grps_names, + "One of the Affinity Groups is missing %s" + %list_aff_grps_names) + + vm1.delete(self.api_client) + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + AffinityGroup.delete(self.api_client, self.aff_grp[0].name) + AffinityGroup.delete(self.api_client, self.aff_grp[1].name) + + @attr(tags=["simulator", "basic", "advanced", "multihost"]) + def test_09_deploy_vm_multiple_aff_grps(self): + """ + Deploy multiple vms in multiple affinity groups + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_1"]) + vm1, hostid1 = self.create_vm_in_aff_grps(ag_list=[self.aff_grp[0].name, + self.aff_grp[1].name]) + vm2, hostid2 = self.create_vm_in_aff_grps(ag_list=[self.aff_grp[0].name, + self.aff_grp[1].name]) + + aff_grps_names = [self.aff_grp[0].name, self.aff_grp[1].name] + aff_grps_names.sort() + + for vm in [vm1, vm2]: + list_aff_grps = AffinityGroup.list(self.api_client, + virtualmachineid=vm.id) + + list_aff_grps_names = [list_aff_grps[0].name, list_aff_grps[1].name] + + list_aff_grps_names.sort() + self.assertEqual(aff_grps_names, list_aff_grps_names, + "One of the Affinity Groups is missing %s" + %list_aff_grps_names) + + vm1.delete(self.api_client) + vm2.delete(self.api_client) + wait_for_cleanup(self.apiclient, ["expunge.delay", "expunge.interval"]) + + AffinityGroup.delete(self.api_client, self.aff_grp[0].name) + AffinityGroup.delete(self.api_client, self.aff_grp[1].name) + + @attr(tags=["simulator", "basic", "advanced", "multihost"]) + def test_10_deploy_vm_by_aff_grp_name_and_id(self): + """ + Deploy VM by aff grp name and id + """ + + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"]) + list_aff_grps = AffinityGroup.list(self.api_client, + name=self.aff_grp[0].name) + + with self.assertRaises(Exception): + vm1, hostid1 = self.create_vm_in_aff_grps(ag_list=[self.aff_grp[0].name], + ag_ids=[list_aff_grps[0].id]) + + AffinityGroup.delete(self.api_client, self.aff_grp[0].name) + +class TestAffinityGroupsAdminUser(cloudstackTestCase): + + @classmethod + def setUpClass(cls): + + cls.api_client = super(TestAffinityGroupsAdminUser, cls).getClsTestClient().getApiClient() + cls.services = Services().services + # Get Zone, Domain and templates + cls.domain = get_domain(cls.api_client, cls.services) + cls.zone = get_zone(cls.api_client, cls.services) + cls.template = get_template( + cls.api_client, + cls.zone.id, + cls.services["ostype"] + ) + cls.services["virtual_machine"]["zoneid"] = cls.zone.id + + cls.services["template"] = cls.template.id + cls.services["zoneid"] = cls.zone.id + + cls.account = Account.create( + cls.api_client, + cls.services["account"], + domainid=cls.domain.id + ) + + cls.services["account"] = cls.account.name + + cls.service_offering = ServiceOffering.create( + cls.api_client, + cls.services["service_offering"] + ) + + cls.__cleanup = [ + cls.service_offering, + cls.account, + ] + + # Create multiple Affinity Groups + return + + def setUp(self): + self.apiclient = self.testClient.getApiClient() + self.dbclient = self.testClient.getDbConnection() + self.aff_grp = [] + self.cleanup = [] + + def tearDown(self): + try: + self.api_client = super(TestAffinityGroupsAdminUser,self).getClsTestClient().getApiClient() + #Clean up, terminate the created templates + cleanup_resources(self.api_client, self.cleanup) + except Exception as e: + raise Exception("Warning: Exception during cleanup : %s" % e) + + @classmethod + def tearDownClass(cls): + + try: + cls.api_client = super(TestAffinityGroupsAdminUser, cls).getClsTestClient().getApiClient() + #Clean up, terminate the created templates + cleanup_resources(cls.api_client, cls.__cleanup) + except Exception as e: + raise Exception("Warning: Exception during cleanup : %s" % e) + + def create_aff_grp(self, api_client=None, aff_grp=None, + acc=None, domainid=None): + + if api_client == None: + api_client = self.api_client + if aff_grp == None: + self.services["host_anti_affinity_0"] + #if acc == None: + # acc = self.account.name + #if domainid == None: + # domainid = self.domain.id + + try: + self.aff_grp.append(AffinityGroup.create(api_client, + aff_grp, acc, domainid)) + except Exception as e: + raise Exception("Error: Creation of Affinity Group failed : %s" %e) + + def create_vm_in_aff_grps(self, api_client=None, ag_list=None, ag_ids=None): + + if api_client == None: + api_client = self.api_client + self.debug('Creating VM in AffinityGroup=%s' % ag_list) + vm = VirtualMachine.create( + api_client, + self.services["virtual_machine"], + templateid=self.template.id, + #accountid=self.account.name, + #domainid=self.account.domainid, + serviceofferingid=self.service_offering.id, + affinitygroupnames=ag_list, + affinitygroupids=ag_ids + ) + self.debug('Created VM=%s in Affinity Group=%s' % + (vm.id, ag_list)) + + list_vm = list_virtual_machines(self.api_client, id=vm.id) + + self.assertEqual(isinstance(list_vm, list), True, + "Check list response returns a valid list") + self.assertNotEqual(len(list_vm),0, + "Check VM available in Delete Virtual Machines") + + vm_response = list_vm[0] + self.assertEqual(vm_response.state, 'Running', + msg="VM is not in Running state") + + return vm, vm_response.hostid + + @attr(tags=["simulator", "basic", "advanced", "multihost"]) + def test_01_deploy_vm_another_user(self): + """ + Deploy vm in Affinity Group belonging to regular user + """ + self.user1 = Account.create(self.api_client, + self.services["new_account"]) + + self.cleanup.append(self.user1) + userapiclient = self.testClient.createUserApiClient( + UserName=self.user1.name, + DomainName=self.user1.domain, + acctType=0) + + self.create_aff_grp(api_client=userapiclient, + aff_grp=self.services["host_anti_affinity_0"]) + + with self.assertRaises(Exception): + vm1, hostid1 = self.create_vm_in_aff_grps(ag_list=[self.aff_grp[0].name]) + + AffinityGroup.delete(userapiclient, self.aff_grp[0].name) + + @attr(tags=["simulator", "basic", "advanced", "multihost"]) + def test_02_create_aff_grp_user(self): + """ + Create Affinity Group forregular user + """ + + self.user = Account.create(self.api_client, self.services["new_account"], + domainid=self.domain.id) + + self.cleanup.append(self.user) + self.create_aff_grp(aff_grp=self.services["host_anti_affinity_0"], + acc=self.user.name, domainid=self.domain.id) + + AffinityGroup.delete(self.api_client, name=self.aff_grp[0].name, + account=self.user.name, domainid=self.domain.id) + self.debug("Deleted Affinity Group: %s" %self.aff_grp[0].name) + + @attr(tags=["simulator", "basic", "advanced", "multihost"]) + def test_03_list_aff_grp_all_users(self): + """ + List Affinity Groups for all the users + """ + + self.user1 = Account.create(self.api_client, + self.services["new_account"]) + + self.cleanup.append(self.user1) + userapiclient = self.testClient.createUserApiClient( + UserName=self.user1.name, + DomainName=self.user1.domain, + acctType=0) + + self.create_aff_grp(api_client=userapiclient, + aff_grp=self.services["host_anti_affinity_0"]) + + list_aff_grps = AffinityGroup.list(self.api_client) + print list_aff_grps + self.assertNotEqual(list_aff_grps, [], "Admin not able to list Affinity " + "Groups of users") + AffinityGroup.delete(userapiclient, self.aff_grp[0].name) http://git-wip-us.apache.org/repos/asf/cloudstack/blob/7bad7327/tools/marvin/marvin/integration/lib/base.py ---------------------------------------------------------------------- diff --git a/tools/marvin/marvin/integration/lib/base.py b/tools/marvin/marvin/integration/lib/base.py index 503ed64..710d9a9 100755 --- a/tools/marvin/marvin/integration/lib/base.py +++ b/tools/marvin/marvin/integration/lib/base.py @@ -2965,28 +2965,40 @@ class AffinityGroup: self.__dict__.update(items) @classmethod - def create(cls, apiclient, services, account=None, domainid=None): - agCmd = createAffinityGroup.createAffinityGroupCmd() - agCmd.name = services['name'] - agCmd.displayText = services['displaytext'] if 'displaytext' in services else services['name'] - agCmd.type = services['type'] - agCmd.account = services['account'] if 'account' in services else account - agCmd.domainid = services['domainid'] if 'domainid' in services else domainid - return AffinityGroup(apiclient.createAffinityGroup(agCmd).__dict__) + def create(cls, apiclient, aff_grp, account=None, domainid=None): + cmd = createAffinityGroup.createAffinityGroupCmd() + cmd.name = aff_grp['name'] + cmd.displayText = aff_grp['name'] + cmd.type = aff_grp['type'] + #if account is not None: + # cmd.account = account + #if domainid is not None: + # cmd.domainid = domainid + return AffinityGroup(apiclient.createAffinityGroup(cmd).__dict__) def update(self, apiclient): pass - def delete(self, apiclient): + @classmethod + def delete(cls, apiclient, name=None, id=None, account=None, + domainid=None): cmd = deleteAffinityGroup.deleteAffinityGroupCmd() - cmd.id = self.id - return apiclient.deleteVPC(cmd) + if id is not None: + cmd.id = id + if name is not None: + cmd.name = name + if account is not None: + cmd.account = account + if domainid is not None: + cmd.domaindid = domainid + + return apiclient.deleteAffinityGroup(cmd) @classmethod def list(cls, apiclient, **kwargs): cmd = listAffinityGroups.listAffinityGroupsCmd() [setattr(cmd, k, v) for k, v in kwargs.items()] - return(apiclient.listVPCs(cmd)) + return apiclient.listAffinityGroups(cmd) class StaticRoute: """Manage static route lifecycle""" @@ -3373,4 +3385,4 @@ class ApplicationLoadBalancer: """List all appln load balancers""" cmd = listLoadBalancers.listLoadBalancersCmd() [setattr(cmd, k, v) for k, v in kwargs.items()] - return(apiclient.listLoadBalancerRules(cmd)) \ No newline at end of file + return(apiclient.listLoadBalancerRules(cmd))