#!/usr/bin/env python
try:
    import unittest2 as unittest
except ImportError:
    import unittest
import random
from cloudstackAPI import *
from cloudstackTestCase import *

class Provision(cloudstackTestCase):
    '''
    This should test basic provisioning of virtual machines
    1. Deploy a no-frills VM
    2. Deploy with tags - hosttags
    3. Deploy with/without security groups
    '''

    solist = {}
    sgid = 1 

    def setUp(self):
        pass


    def tearDown(self):
        pass


    def test_0_createServiceOfferings(self):
        for tag in ['TAG1', 'TAG2', 'TAG3']:
            csocmd = createServiceOffering.createServiceOfferingCmd()
            csocmd.cpunumber = 1
            csocmd.cpuspeed = 2048
            csocmd.displaytext =  tag
            csocmd.memory = 7168
            csocmd.name = tag
            csocmd.hosttags = tag
            csocmd.storagetype = 'local'
            csoresponse = self.testClient.getApiClient().createServiceOffering(csocmd)
            self.debug("Created Service Offering: %s" %tag)
            self.solist[tag]=csoresponse.id


    def test_1_createSecurityGroupAndRules(self):
        apiClient = self.testClient.getApiClient()
        sgCmd=createSecurityGroup.createSecurityGroupCmd()
        sgCmd.description='default-zzzz'
        sgCmd.name='default-zzzz'
        sgCmd.account='admin'
        sgCmd.domainid='1'
        sgRes = apiClient.createSecurityGroup(sgCmd)

        ruleCmd=authorizeSecurityGroupIngress.authorizeSecurityGroupIngressCmd()
        ruleCmd.cidrlist = '172.16.0.0/12'
        ruleCmd.startport = '1'
        ruleCmd.endport = '65535'
        ruleCmd.protocol = 'TCP'
        ruleCmd.securitygroupid = sgRes.id
        ruleCmd.account='admin'
        ruleCmd.domainid='1'
        sgIngressresponse=apiClient.authorizeSecurityGroupIngress(ruleCmd)

        ruleCmd=authorizeSecurityGroupIngress.authorizeSecurityGroupIngressCmd()
        ruleCmd.cidrlist = '10.0.0.0/8'
        ruleCmd.startport = '22'
        ruleCmd.endport = '22'
        ruleCmd.protocol = 'TCP'
        ruleCmd.securitygroupid = sgRes.id
        ruleCmd.account='admin'
        ruleCmd.domainid='1'
        sgIngressresponse=apiClient.authorizeSecurityGroupIngress(ruleCmd)

        ruleCmd=authorizeSecurityGroupIngress.authorizeSecurityGroupIngressCmd()
        ruleCmd.cidrlist = '10.0.0.0/8'
        ruleCmd.startport = '80'
        ruleCmd.endport = '80'
        ruleCmd.protocol = 'TCP'
        ruleCmd.securitygroupid = sgRes.id
        ruleCmd.account='admin'
        ruleCmd.domainid='1'
        sgIngressresponse=apiClient.authorizeSecurityGroupIngress(ruleCmd)

        self.sgid = sgRes.id



    def test_2_DeployVMWithHostTags(self):
        '''
        Deploy 3 virtual machines one with each hosttag
        '''
        for k,v in self.solist:
            deployVmCmd = deployVirtualMachine.deployVirtualMachineCmd()
            deployVmCmd.zoneid = 1
            deployVmCmd.hypervisor='Simulator'
            deployVmCmd.serviceofferingid=v
            deployVmCmd.account='admin'
            deployVmCmd.domainid=1
            deployVmCmd.templateid=10
            deployVmResponse = self.testClient.getApiClient().deployVirtualMachine(deployVmCmd)
            self.debug("Deployed VM :%d in job: %d"%(deployVmResponse[0].id, deployVmResponse[0].jobid))


    def deployCmd(self, tag):
        deployVmCmd = deployVirtualMachine.deployVirtualMachineCmd()
        deployVmCmd.zoneid = 1
        deployVmCmd.hypervisor='Simulator'
        deployVmCmd.serviceofferingid=self.solist[tag]
        deployVmCmd.account='admin'
        deployVmCmd.domainid=1
        deployVmCmd.templateid=10
        deployVmCmd.securitygroupids=self.sgid
        return deployVmCmd

    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
        '''
        tag1=nargs*5/13
        tag2=nargs*2/13
        tag3=nargs-tag1-tag2

        cmds = []
        [cmds.append(self.deployCmd('TAG1')) for i in range(tag1)]
        [cmds.append(self.deployCmd('TAG2')) for i in range(tag2)]
        [cmds.append(self.deployCmd('TAG3')) for i in range(tag3)]
        random.shuffle(cmds) #with mix-and-match of Tags

        if batchsize == 0:
            self.testClient.submitCmdsAndWait(cmds)
        else:
            while len(cmds) > 0:
                try:
                    newbatch = [cmds.pop() for b in range(batchsize)] #pop batchsize items
                    self.testClient.submitCmdsAndWait(newbatch)
                except IndexError:
                    break

    def test_3_bulkDeploy(self):
        self.deployN(130,0)
        self.deployN(nargs=9000,batchsize=100)