| #!/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. |
| import unittest2 as unittest |
| except ImportError: |
| import unittest |
| |
| import random |
| import hashlib |
| from cloudstackTestCase import * |
| |
| class Provision(cloudstackTestCase): |
| ''' |
| ''' |
| def setUp(self): |
| pass |
| |
| |
| def tearDown(self): |
| pass |
| |
| |
| def test_1_createAccounts(self, numberOfAccounts=5): |
| ''' |
| Create a bunch of user accounts |
| ''' |
| mdf = hashlib.md5() |
| mdf.update('password') |
| mdf_pass = mdf.hexdigest() |
| api = self.testClient.getApiClient() |
| for i in range(1, numberOfAccounts + 1): |
| acct = createAccount.createAccountCmd() |
| acct.accounttype = 0 |
| acct.firstname = 'user' + str(i) |
| acct.lastname = 'user' + str(i) |
| acct.password = mdf_pass |
| acct.username = 'user' + str(i) |
| acct.email = 'user@example.com' |
| acct.account = 'user' + str(i) |
| acct.domainid = 1 |
| acctResponse = api.createAccount(acct) |
| self.debug("successfully created account: %s, user: %s, id: %s"%(acctResponse.account, acctResponse.username, acctResponse.id)) |
| |
| |
| def test_2_createServiceOffering(self): |
| apiClient = self.testClient.getApiClient() |
| createSOcmd=createServiceOffering.createServiceOfferingCmd() |
| createSOcmd.name='Sample SO' |
| createSOcmd.displaytext='Sample SO' |
| createSOcmd.storagetype='shared' |
| createSOcmd.cpunumber=1 |
| createSOcmd.cpuspeed=100 |
| createSOcmd.memory=128 |
| createSOcmd.offerha='false' |
| createSOresponse = apiClient.createServiceOffering(createSOcmd) |
| return createSOresponse.id |
| |
| def deployCmd(self, account, service): |
| deployVmCmd = deployVirtualMachine.deployVirtualMachineCmd() |
| deployVmCmd.zoneid = 1 |
| deployVmCmd.hypervisor='Simulator' |
| deployVmCmd.account=account |
| deployVmCmd.domainid=1 |
| deployVmCmd.templateid=10 |
| deployVmCmd.serviceofferingid=service |
| return deployVmCmd |
| |
| def listVmsInAccountCmd(self, acct): |
| api = self.testClient.getApiClient() |
| listVmCmd = listVirtualMachines.listVirtualMachinesCmd() |
| listVmCmd.account = acct |
| listVmCmd.zoneid = 1 |
| listVmCmd.domainid = 1 |
| listVmResponse = api.listVirtualMachines(listVmCmd) |
| return listVmResponse |
| |
| |
| def destroyVmCmd(self, key): |
| api = self.testClient.getApiClient() |
| destroyVmCmd = destroyVirtualMachine.destroyVirtualMachineCmd() |
| destroyVmCmd.id = key |
| api.destroyVirtualMachine(destroyVmCmd) |
| |
| |
| def test_3_stressDeploy(self): |
| ''' |
| Deploy 20 Vms in each account |
| ''' |
| service_id = self.test_2_createServiceOffering() |
| api = self.testClient.getApiClient() |
| for acct in range(1, 5): |
| [api.deployVirtualMachine(self.deployCmd('user'+str(acct), service_id)) for x in range(0,20)] |
| |
| def test_4_stressDestroy(self): |
| ''' |
| Cleanup all Vms in every account |
| ''' |
| api = self.testClient.getApiClient() |
| for acct in range(1, 6): |
| for vm in self.listVmsInAccountCmd('user'+str(acct)): |
| if vm is not None: |
| self.destroyVmCmd(vm.id) |
| |
| def test_5_combineStress(self): |
| for i in range(0, 5): |
| self.test_3_stressDeploy() |
| self.test_4_stressDestroy() |
| |
| def deployN(self,nargs=300,batchsize=0): |
| ''' |
| Deploy Nargs number of VMs concurrently in batches of size {batchsize}. |
| When batchsize is 0 all Vms are deployed in one batch |
| VMs will be deployed in 5:2:6 ratio |
| ''' |
| cmds = [] |
| |
| if batchsize == 0: |
| self.testClient.submitCmdsAndWait(cmds) |
| else: |
| while len(z) > 0: |
| try: |
| newbatch = [cmds.pop() for b in range(batchsize)] #pop batchsize items |
| self.testClient.submitCmdsAndWait(newbatch) |
| except IndexError: |
| break |
