| #!/usr/bin/env python |
| |
| import marvin |
| from marvin import cloudstackTestCase |
| from marvin.cloudstackTestCase import * |
| from marvin.integration.lib.base import * |
| |
| import unittest |
| import hashlib |
| import random |
| import os |
| import string |
| |
| class TestDeployVmWithUserData(cloudstackTestCase): |
| """ |
| This test deploys a virtual machine into a user account |
| using the small service offering and builtin template |
| """ |
| def setUp(self): |
| password = "password" |
| |
| self.apiClient = self.testClient.getApiClient() #Get ourselves an API client |
| |
| self.acct = createAccount.createAccountCmd() #The createAccount command |
| self.acct.accounttype = 0 #We need a regular user. admins have accounttype=1 |
| self.acct.firstname = 'firstname' |
| self.acct.lastname = 'lastname' |
| self.acct.password = password |
| self.acct.username = 'user1' |
| self.acct.email = 'user1@user.com' |
| self.acct.account = 'user1' |
| self.acct.domainid = 1 #The default ROOT domain |
| self.acctResponse = self.apiClient.createAccount(self.acct) |
| |
| self.debug("Successfully created account: %s, user: %s, id: \ |
| %s"%(self.acctResponse.account.account, \ |
| self.acctResponse.account.username, \ |
| self.acctResponse.account.id)) |
| |
| # Generate userdata of 2500 bytes. This is larger than the 2048 bytes limit. |
| # CS however allows for upto 4K bytes in the code. So this must succeed. |
| # Overall, the query length must not exceed 4K, for then the json decoder |
| # will fail this operation at the marvin client side itself. |
| user_data = ''.join(random.choice(string.ascii_uppercase + string.digits) for x in range(2500)) |
| |
| self.virtual_machine = { |
| "displayname": "Test VM", |
| "username": "root", |
| "password": "password", |
| "ssh_port": 22, |
| "hypervisor": 'VMware', |
| "privateport": 22, |
| "publicport": 22, |
| "protocol": 'TCP', |
| } |
| #self.virtual_machine["userdata"] = base64.b64encode(user_data) |
| self.virtual_machine["userdata"] = user_data |
| |
| def test_DeployVm(self): |
| """ |
| Let's start by defining the attributes of our VM that we will be |
| deploying on CloudStack. We will be assuming a single zone is available |
| and is configured and all templates are Ready |
| """ |
| deployVmCmd = deployVirtualMachine.deployVirtualMachineCmd() |
| deployVmCmd.zoneid = 1 |
| deployVmCmd.account = self.acct.account |
| deployVmCmd.domainid = self.acct.domainid |
| deployVmCmd.templateid = 7 |
| deployVmCmd.serviceofferingid = 1 |
| |
| # Userdata is passed in the virtual_machine dictionary. |
| deployVmResponse = VirtualMachine.create( |
| self.apiClient, |
| self.virtual_machine, |
| accountid=self.acct.account, |
| domainid=self.acct.domainid, |
| serviceofferingid=deployVmCmd.serviceofferingid, |
| templateid=deployVmCmd.templateid, |
| zoneid=deployVmCmd.zoneid |
| ) |
| |
| # At this point our VM is expected to be Running. Let's find out what |
| # listVirtualMachines tells us about VMs in this account |
| |
| listVmCmd = listVirtualMachines.listVirtualMachinesCmd() |
| listVmCmd.id = deployVmResponse.id |
| listVmResponse = self.apiClient.listVirtualMachines(listVmCmd) |
| |
| self.assertNotEqual(len(listVmResponse), 0, "Check if the list API \ |
| returns a non-empty response") |
| |
| vm = listVmResponse[0] |
| |
| self.assertEqual(vm.id, deployVmResponse.id, "Check if the VM returned \ |
| is the same as the one we deployed") |
| |
| |
| self.assertEqual(vm.state, "Running", "Check if VM has reached \ |
| a state of running") |
| |
| def tearDown(self): |
| """ |
| Delete the account created. This will clear the VM belonging to that account as well. |
| """ |
| deleteAcct = deleteAccount.deleteAccountCmd() |
| deleteAcct.id = self.acctResponse.account.id |
| self.apiClient.deleteAccount(deleteAcct) |
| |
| |
