mirror of
				https://github.com/apache/cloudstack.git
				synced 2025-11-04 00:02:37 +01:00 
			
		
		
		
	
		
			
				
	
	
		
			2958 lines
		
	
	
		
			128 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			2958 lines
		
	
	
		
			128 KiB
		
	
	
	
		
			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.
 | 
						|
 | 
						|
""" P1 tests for netscaler load balancing
 | 
						|
"""
 | 
						|
#Import Local Modules
 | 
						|
import marvin
 | 
						|
from nose.plugins.attrib import attr
 | 
						|
from marvin.cloudstackTestCase import *
 | 
						|
from marvin.cloudstackAPI import *
 | 
						|
from marvin.lib.utils import *
 | 
						|
from marvin.lib.base import *
 | 
						|
from marvin.lib.common import *
 | 
						|
from marvin.sshClient import SshClient
 | 
						|
import datetime
 | 
						|
 | 
						|
 | 
						|
class Services:
 | 
						|
    """Test netscaler services
 | 
						|
    """
 | 
						|
 | 
						|
    def __init__(self):
 | 
						|
        self.services = {
 | 
						|
                         "account": {
 | 
						|
                                    "email": "test@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": 128,       # In MBs
 | 
						|
                         },
 | 
						|
                         "virtual_machine": {
 | 
						|
                                    "displayname": "TestVM",
 | 
						|
                                    "username": "root",
 | 
						|
                                    "password": "password",
 | 
						|
                                    "ssh_port": 22,
 | 
						|
                                    "hypervisor": 'XenServer',
 | 
						|
                                    "privateport": 22,
 | 
						|
                                    "publicport": 22,
 | 
						|
                                    "protocol": 'TCP',
 | 
						|
                                },
 | 
						|
                         "netscaler": {
 | 
						|
                                "ipaddress": '10.147.40.100',
 | 
						|
                                "username": 'nsroot',
 | 
						|
                                "password": 'nsroot',
 | 
						|
                                "networkdevicetype": 'NetscalerVPXLoadBalancer',
 | 
						|
                                "publicinterface": '1/1',
 | 
						|
                                "privateinterface": '1/1',
 | 
						|
                                "numretries": 2,
 | 
						|
                                "lbdevicededicated": False,
 | 
						|
                                "lbdevicecapacity": 50,
 | 
						|
                                "port": 22,
 | 
						|
                         },
 | 
						|
                         "network_offering_dedicated": {
 | 
						|
                                    "name": 'Netscaler',
 | 
						|
                                    "displaytext": 'Netscaler',
 | 
						|
                                    "guestiptype": 'Isolated',
 | 
						|
                                    "supportedservices": 'Dhcp,Dns,SourceNat,PortForwarding,Vpn,Firewall,Lb,UserData,StaticNat',
 | 
						|
                                    "traffictype": 'GUEST',
 | 
						|
                                    "availability": 'Optional',
 | 
						|
                                    "specifyVlan": False,
 | 
						|
                                    "specifyIpRanges": False,
 | 
						|
                                    "serviceProviderList": {
 | 
						|
                                            "Dhcp": 'VirtualRouter',
 | 
						|
                                            "Dns": 'VirtualRouter',
 | 
						|
                                            "SourceNat": 'VirtualRouter',
 | 
						|
                                            "PortForwarding": 'VirtualRouter',
 | 
						|
                                            "Vpn": 'VirtualRouter',
 | 
						|
                                            "Firewall": 'VirtualRouter',
 | 
						|
                                            "Lb": 'Netscaler',
 | 
						|
                                            "UserData": 'VirtualRouter',
 | 
						|
                                            "StaticNat": 'VirtualRouter',
 | 
						|
                                    },
 | 
						|
                                    "serviceCapabilityList": {
 | 
						|
                                        "SourceNat": {
 | 
						|
                                            "SupportedSourceNatTypes": "peraccount"
 | 
						|
                                        },
 | 
						|
                                        "lb": {
 | 
						|
                                               "SupportedLbIsolation": "dedicated"
 | 
						|
                                        },
 | 
						|
                                    },
 | 
						|
                         },
 | 
						|
                         "network_offering": {
 | 
						|
                                    "name": 'Netscaler',
 | 
						|
                                    "displaytext": 'Netscaler',
 | 
						|
                                    "guestiptype": 'Isolated',
 | 
						|
                                    "supportedservices": 'Dhcp,Dns,SourceNat,PortForwarding,Vpn,Firewall,Lb,UserData,StaticNat',
 | 
						|
                                    "traffictype": 'GUEST',
 | 
						|
                                    "availability": 'Optional',
 | 
						|
                                    "serviceProviderList": {
 | 
						|
                                            "Dhcp": 'VirtualRouter',
 | 
						|
                                            "Dns": 'VirtualRouter',
 | 
						|
                                            "SourceNat": 'VirtualRouter',
 | 
						|
                                            "PortForwarding": 'VirtualRouter',
 | 
						|
                                            "Vpn": 'VirtualRouter',
 | 
						|
                                            "Firewall": 'VirtualRouter',
 | 
						|
                                            "Lb": 'Netscaler',
 | 
						|
                                            "UserData": 'VirtualRouter',
 | 
						|
                                            "StaticNat": 'VirtualRouter',
 | 
						|
                                    },
 | 
						|
                         },
 | 
						|
                         "network": {
 | 
						|
                                  "name": "Netscaler",
 | 
						|
                                  "displaytext": "Netscaler",
 | 
						|
                         },
 | 
						|
                         "lbrule": {
 | 
						|
                                    "name": "SSH",
 | 
						|
                                    "alg": "roundrobin",
 | 
						|
                                    # Algorithm used for load balancing
 | 
						|
                                    "privateport": 22,
 | 
						|
                                    "publicport": 22,
 | 
						|
                                    "openfirewall": False,
 | 
						|
                         },
 | 
						|
                         "natrule": {
 | 
						|
                                    "privateport": 22,
 | 
						|
                                    "publicport": 22,
 | 
						|
                                    "protocol": "TCP"
 | 
						|
                         },
 | 
						|
                         "ostype": 'CentOS 5.3 (64-bit)',
 | 
						|
                         # Cent OS 5.3 (64 bit)
 | 
						|
                         "sleep": 60,
 | 
						|
                         "timeout": 10,
 | 
						|
                         "mode": 'advanced'
 | 
						|
                    }
 | 
						|
 | 
						|
class TestLbSourceNat(cloudstackTestCase):
 | 
						|
 | 
						|
    @classmethod
 | 
						|
    def setUpClass(cls):
 | 
						|
        cls._cleanup = []
 | 
						|
        cls.testClient = super(TestLbSourceNat, cls).getClsTestClient()
 | 
						|
        cls.api_client = cls.testClient.getApiClient()
 | 
						|
 | 
						|
        cls.services = Services().services
 | 
						|
        # Get Zone, Domain and templates
 | 
						|
        cls.domain = get_domain(cls.api_client)
 | 
						|
        cls.zone = get_zone(cls.api_client, cls.testClient.getZoneForTests())
 | 
						|
        cls.template = get_template(
 | 
						|
                            cls.api_client,
 | 
						|
                            cls.zone.id,
 | 
						|
                            cls.services["ostype"]
 | 
						|
                           )
 | 
						|
        try:
 | 
						|
           cls.netscaler = add_netscaler(cls.api_client, cls.zone.id, cls.services["netscaler"])
 | 
						|
           cls._cleanup.append(cls.netscaler)
 | 
						|
           cls.network_offering = NetworkOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["network_offering"],
 | 
						|
                                            conservemode=True
 | 
						|
                                            )
 | 
						|
           # Enable Network offering
 | 
						|
           cls.network_offering.update(cls.api_client, state='Enabled')
 | 
						|
           cls.services["virtual_machine"]["zoneid"] = cls.zone.id
 | 
						|
           cls.services["virtual_machine"]["template"] = cls.template.id
 | 
						|
           cls.service_offering = ServiceOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["service_offering"]
 | 
						|
                                            )
 | 
						|
        except Exception as e:
 | 
						|
           cls.tearDownClass()
 | 
						|
           raise Exception ("Warning: Exception in setUpClass: %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @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)
 | 
						|
        return
 | 
						|
 | 
						|
    def setUp(self):
 | 
						|
        self.apiclient = self.testClient.getApiClient()
 | 
						|
        self.dbclient = self.testClient.getDbConnection()
 | 
						|
        self.account = Account.create(
 | 
						|
                                     self.apiclient,
 | 
						|
                                     self.services["account"],
 | 
						|
                                     admin=True,
 | 
						|
                                     domainid=self.domain.id
 | 
						|
                                     )
 | 
						|
        self.cleanup = [self.account]
 | 
						|
        return
 | 
						|
 | 
						|
    def tearDown(self):
 | 
						|
        try:
 | 
						|
            self.debug("Cleaning up the resources")
 | 
						|
            #Clean up, terminate the created network offerings
 | 
						|
            cleanup_resources(self.apiclient, self.cleanup)
 | 
						|
            interval = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='network.gc.interval'
 | 
						|
                                    )
 | 
						|
            wait = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='network.gc.wait'
 | 
						|
                                    )
 | 
						|
            # Sleep to ensure that all resources are deleted
 | 
						|
            time.sleep(int(interval[0].value) + int(wait[0].value))
 | 
						|
            self.debug("Cleanup complete!")
 | 
						|
        except Exception as e:
 | 
						|
            raise Exception("Warning: Exception during cleanup : %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_add_lb_on_source_nat(self):
 | 
						|
        """Test Create LB rule for sourceNat IP address
 | 
						|
        """
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Deploy the first VM using a network from the above created
 | 
						|
        #    Network offering.
 | 
						|
        # 2. Deploy another VM.
 | 
						|
        # 3. Create LB rule for the sourceNat IP address. User should NOT be
 | 
						|
        #    allowed to create an LB rule on source NAT
 | 
						|
 | 
						|
        # Creating network using the network offering created
 | 
						|
        self.debug("Creating network with network offering: %s" %
 | 
						|
                                                    self.network_offering.id)
 | 
						|
        self.network = Network.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["network"],
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    domainid=self.account.domainid,
 | 
						|
                                    networkofferingid=self.network_offering.id,
 | 
						|
                                    zoneid=self.zone.id
 | 
						|
                                    )
 | 
						|
        self.debug("Created network with ID: %s" % self.network.id)
 | 
						|
 | 
						|
        self.debug("Deploying VM in account: %s" % self.account.name)
 | 
						|
 | 
						|
        # Spawn an instance in that network
 | 
						|
        virtual_machine_1 = VirtualMachine.create(
 | 
						|
                                  self.apiclient,
 | 
						|
                                  self.services["virtual_machine"],
 | 
						|
                                  accountid=self.account.name,
 | 
						|
                                  domainid=self.account.domainid,
 | 
						|
                                  serviceofferingid=self.service_offering.id,
 | 
						|
                                  networkids=[str(self.network.id)]
 | 
						|
                                  )
 | 
						|
        self.debug("Deployed VM in network: %s" % self.network.id)
 | 
						|
        list_vm_response = VirtualMachine.list(
 | 
						|
                                        self.apiclient,
 | 
						|
                                        id=virtual_machine_1.id
 | 
						|
                                        )
 | 
						|
 | 
						|
        self.debug(
 | 
						|
                "Verify listVirtualMachines response for virtual machine: %s" \
 | 
						|
                % virtual_machine_1.id
 | 
						|
            )
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            isinstance(list_vm_response, list),
 | 
						|
                            True,
 | 
						|
                            "Check list response returns a valid list"
 | 
						|
                        )
 | 
						|
        vm_response = list_vm_response[0]
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            vm_response.state,
 | 
						|
                            "Running",
 | 
						|
                            "VM state should be running after deployment"
 | 
						|
                        )
 | 
						|
 | 
						|
        self.debug("Deploying another VM in account: %s" %
 | 
						|
                                            self.account.name)
 | 
						|
 | 
						|
        # Spawn an instance in that network
 | 
						|
        virtual_machine_2 = VirtualMachine.create(
 | 
						|
                                  self.apiclient,
 | 
						|
                                  self.services["virtual_machine"],
 | 
						|
                                  accountid=self.account.name,
 | 
						|
                                  domainid=self.account.domainid,
 | 
						|
                                  serviceofferingid=self.service_offering.id,
 | 
						|
                                  networkids=[str(self.network.id)]
 | 
						|
                                  )
 | 
						|
        self.debug("Deployed VM in network: %s" % self.network.id)
 | 
						|
        list_vm_response = VirtualMachine.list(
 | 
						|
                                        self.apiclient,
 | 
						|
                                        id=virtual_machine_2.id
 | 
						|
                                        )
 | 
						|
 | 
						|
        self.debug(
 | 
						|
                "Verify listVirtualMachines response for virtual machine: %s" \
 | 
						|
                % virtual_machine_2.id
 | 
						|
            )
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            isinstance(list_vm_response, list),
 | 
						|
                            True,
 | 
						|
                            "Check list response returns a valid list"
 | 
						|
                        )
 | 
						|
        vm_response = list_vm_response[0]
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            vm_response.state,
 | 
						|
                            "Running",
 | 
						|
                            "VM state should be running after deployment"
 | 
						|
                        )
 | 
						|
        src_nat_list = PublicIPAddress.list(
 | 
						|
                                        self.apiclient,
 | 
						|
                                        associatednetworkid=self.network.id,
 | 
						|
                                        account=self.account.name,
 | 
						|
                                        domainid=self.account.domainid,
 | 
						|
                                        listall=True,
 | 
						|
                                        issourcenat=True,
 | 
						|
                                        )
 | 
						|
        self.assertEqual(
 | 
						|
                         isinstance(src_nat_list, list),
 | 
						|
                         True,
 | 
						|
                         "List Public IP should return a valid source NAT"
 | 
						|
                         )
 | 
						|
        self.assertNotEqual(
 | 
						|
                    len(src_nat_list),
 | 
						|
                    0,
 | 
						|
                    "Length of response from listPublicIp should not be 0"
 | 
						|
                    )
 | 
						|
 | 
						|
        src_nat = src_nat_list[0]
 | 
						|
 | 
						|
        self.debug("Trying to create LB rule on source NAT IP: %s" %
 | 
						|
                                                        src_nat.ipaddress)
 | 
						|
        # Create Load Balancer rule with source NAT
 | 
						|
        with self.assertRaises(Exception):
 | 
						|
            LoadBalancerRule.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["lbrule"],
 | 
						|
                                    ipaddressid=src_nat.id,
 | 
						|
                                    accountid=self.account.name
 | 
						|
                                )
 | 
						|
        return
 | 
						|
 | 
						|
class TestLbOnIpWithPf(cloudstackTestCase):
 | 
						|
 | 
						|
    @classmethod
 | 
						|
    def setUpClass(cls):
 | 
						|
        cls._cleanup = []
 | 
						|
        cls.testClient = super(TestLbOnIpWithPf, cls).getClsTestClient()
 | 
						|
        cls.api_client = cls.testClient.getApiClient()
 | 
						|
 | 
						|
        cls.services = Services().services
 | 
						|
        # Get Zone, Domain and templates
 | 
						|
        cls.domain = get_domain(cls.api_client)
 | 
						|
        cls.zone = get_zone(cls.api_client, cls.testClient.getZoneForTests())
 | 
						|
        cls.template = get_template(
 | 
						|
                            cls.api_client,
 | 
						|
                            cls.zone.id,
 | 
						|
                            cls.services["ostype"]
 | 
						|
                            )
 | 
						|
        try:
 | 
						|
           cls.netscaler = add_netscaler(cls.api_client, cls.zone.id, cls.services["netscaler"])
 | 
						|
           cls._cleanup.append(cls.netscaler)
 | 
						|
           cls.network_offering = NetworkOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["network_offering"],
 | 
						|
                                            conservemode=True
 | 
						|
                                            )
 | 
						|
           # Enable Network offering
 | 
						|
           cls.network_offering.update(cls.api_client, state='Enabled')
 | 
						|
           cls.services["virtual_machine"]["zoneid"] = cls.zone.id
 | 
						|
           cls.services["virtual_machine"]["template"] = cls.template.id
 | 
						|
 | 
						|
           cls.service_offering = ServiceOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["service_offering"]
 | 
						|
                                            )
 | 
						|
        except Exception as e:
 | 
						|
           cls.tearDownClass()
 | 
						|
           raise Exception ("Warning: Exception in setUpClass: %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @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)
 | 
						|
        return
 | 
						|
 | 
						|
    def setUp(self):
 | 
						|
        self.apiclient = self.testClient.getApiClient()
 | 
						|
        self.dbclient = self.testClient.getDbConnection()
 | 
						|
        self.account = Account.create(
 | 
						|
                                     self.apiclient,
 | 
						|
                                     self.services["account"],
 | 
						|
                                     admin=True,
 | 
						|
                                     domainid=self.domain.id
 | 
						|
                                     )
 | 
						|
        self.cleanup = [self.account]
 | 
						|
        return
 | 
						|
 | 
						|
    def tearDown(self):
 | 
						|
        try:
 | 
						|
            self.debug("Cleaning up the resources")
 | 
						|
            #Clean up, terminate the created network offerings
 | 
						|
            cleanup_resources(self.apiclient, self.cleanup)
 | 
						|
            interval = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='network.gc.interval'
 | 
						|
                                    )
 | 
						|
            wait = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='network.gc.wait'
 | 
						|
                                    )
 | 
						|
            # Sleep to ensure that all resources are deleted
 | 
						|
            time.sleep(int(interval[0].value) + int(wait[0].value))
 | 
						|
            self.debug("Cleanup complete!")
 | 
						|
        except Exception as e:
 | 
						|
            raise Exception("Warning: Exception during cleanup : %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_lb_on_ip_with_pf(self):
 | 
						|
        """Test Create LB rule for sourceNat IP address
 | 
						|
        """
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Deploy the first VM using a network from the above created
 | 
						|
        #    Network offering.
 | 
						|
        # 2. Deploy another VM.
 | 
						|
        # 3. Create LB rule on Ip with PF rule. User should NOT be
 | 
						|
        #    allowed to create an LB rule on Ip with PF
 | 
						|
 | 
						|
        # Creating network using the network offering created
 | 
						|
        self.debug("Creating network with network offering: %s" %
 | 
						|
                                                    self.network_offering.id)
 | 
						|
        self.network = Network.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["network"],
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    domainid=self.account.domainid,
 | 
						|
                                    networkofferingid=self.network_offering.id,
 | 
						|
                                    zoneid=self.zone.id
 | 
						|
                                    )
 | 
						|
        self.debug("Created network with ID: %s" % self.network.id)
 | 
						|
 | 
						|
        self.debug("Deploying VM in account: %s" % self.account.name)
 | 
						|
 | 
						|
        # Spawn an instance in that network
 | 
						|
        virtual_machine_1 = VirtualMachine.create(
 | 
						|
                                  self.apiclient,
 | 
						|
                                  self.services["virtual_machine"],
 | 
						|
                                  accountid=self.account.name,
 | 
						|
                                  domainid=self.account.domainid,
 | 
						|
                                  serviceofferingid=self.service_offering.id,
 | 
						|
                                  networkids=[str(self.network.id)]
 | 
						|
                                  )
 | 
						|
        self.debug("Deployed VM in network: %s" % self.network.id)
 | 
						|
        list_vm_response = VirtualMachine.list(
 | 
						|
                                        self.apiclient,
 | 
						|
                                        id=virtual_machine_1.id
 | 
						|
                                        )
 | 
						|
 | 
						|
        self.debug(
 | 
						|
                "Verify listVirtualMachines response for virtual machine: %s" \
 | 
						|
                % virtual_machine_1.id
 | 
						|
            )
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            isinstance(list_vm_response, list),
 | 
						|
                            True,
 | 
						|
                            "Check list response returns a valid list"
 | 
						|
                        )
 | 
						|
        vm_response = list_vm_response[0]
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            vm_response.state,
 | 
						|
                            "Running",
 | 
						|
                            "VM state should be running after deployment"
 | 
						|
                        )
 | 
						|
 | 
						|
        self.debug("Deploying another VM in account: %s" %
 | 
						|
                                            self.account.name)
 | 
						|
 | 
						|
        # Spawn an instance in that network
 | 
						|
        virtual_machine_2 = VirtualMachine.create(
 | 
						|
                                  self.apiclient,
 | 
						|
                                  self.services["virtual_machine"],
 | 
						|
                                  accountid=self.account.name,
 | 
						|
                                  domainid=self.account.domainid,
 | 
						|
                                  serviceofferingid=self.service_offering.id,
 | 
						|
                                  networkids=[str(self.network.id)]
 | 
						|
                                  )
 | 
						|
        self.debug("Deployed VM in network: %s" % self.network.id)
 | 
						|
        list_vm_response = VirtualMachine.list(
 | 
						|
                                        self.apiclient,
 | 
						|
                                        id=virtual_machine_2.id
 | 
						|
                                        )
 | 
						|
 | 
						|
        self.debug(
 | 
						|
                "Verify listVirtualMachines response for virtual machine: %s" \
 | 
						|
                % virtual_machine_2.id
 | 
						|
            )
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            isinstance(list_vm_response, list),
 | 
						|
                            True,
 | 
						|
                            "Check list response returns a valid list"
 | 
						|
                        )
 | 
						|
        vm_response = list_vm_response[0]
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            vm_response.state,
 | 
						|
                            "Running",
 | 
						|
                            "VM state should be running after deployment"
 | 
						|
                        )
 | 
						|
        self.debug("Associating public IP for network: %s" % self.network.id)
 | 
						|
        ip_with_nat_rule = PublicIPAddress.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    zoneid=self.zone.id,
 | 
						|
                                    domainid=self.account.domainid,
 | 
						|
                                    networkid=self.network.id
 | 
						|
                                    )
 | 
						|
 | 
						|
        self.debug("Associated %s with network %s" % (
 | 
						|
                                        ip_with_nat_rule.ipaddress.ipaddress,
 | 
						|
                                        self.network.id
 | 
						|
                                        ))
 | 
						|
        self.debug("Creating PF rule for IP address: %s" %
 | 
						|
                                        ip_with_nat_rule.ipaddress.ipaddress)
 | 
						|
        NATRule.create(
 | 
						|
                         self.apiclient,
 | 
						|
                         virtual_machine_1,
 | 
						|
                         self.services["natrule"],
 | 
						|
                         ipaddressid=ip_with_nat_rule.ipaddress.id
 | 
						|
                      )
 | 
						|
 | 
						|
        self.debug("Trying to create LB rule on IP with NAT: %s" %
 | 
						|
                                    ip_with_nat_rule.ipaddress.ipaddress)
 | 
						|
 | 
						|
        # Create Load Balancer rule on IP already having NAT rule
 | 
						|
        with self.assertRaises(Exception):
 | 
						|
                LoadBalancerRule.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["lbrule"],
 | 
						|
                                    ipaddressid=ip_with_nat_rule.ipaddress.id,
 | 
						|
                                    accountid=self.account.name
 | 
						|
                                    )
 | 
						|
        return
 | 
						|
 | 
						|
 | 
						|
class TestPfOnIpWithLb(cloudstackTestCase):
 | 
						|
 | 
						|
    @classmethod
 | 
						|
    def setUpClass(cls):
 | 
						|
        cls._cleanup = []
 | 
						|
        cls.testClient = super(TestPfOnIpWithLb, cls).getClsTestClient()
 | 
						|
        cls.api_client = cls.testClient.getApiClient()
 | 
						|
 | 
						|
        cls.services = Services().services
 | 
						|
        # Get Zone, Domain and templates
 | 
						|
        cls.domain = get_domain(cls.api_client)
 | 
						|
        cls.zone = get_zone(cls.api_client, cls.testClient.getZoneForTests())
 | 
						|
        cls.template = get_template(
 | 
						|
                            cls.api_client,
 | 
						|
                            cls.zone.id,
 | 
						|
                            cls.services["ostype"]
 | 
						|
                            )
 | 
						|
        try:
 | 
						|
           cls.netscaler = add_netscaler(cls.api_client, cls.zone.id, cls.services["netscaler"])
 | 
						|
           cls._cleanup.append(cls.netscaler)
 | 
						|
           cls.network_offering = NetworkOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["network_offering"],
 | 
						|
                                            conservemode=True
 | 
						|
                                            )
 | 
						|
           # Enable Network offering
 | 
						|
           cls.network_offering.update(cls.api_client, state='Enabled')
 | 
						|
           cls.services["virtual_machine"]["zoneid"] = cls.zone.id
 | 
						|
           cls.services["virtual_machine"]["template"] = cls.template.id
 | 
						|
 | 
						|
           cls.service_offering = ServiceOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["service_offering"]
 | 
						|
                                            )
 | 
						|
        except Exception as e:
 | 
						|
           cls.tearDownClass()
 | 
						|
           raise Exception ("Warning: Exception in setUpClass: %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @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)
 | 
						|
        return
 | 
						|
 | 
						|
    def setUp(self):
 | 
						|
        self.apiclient = self.testClient.getApiClient()
 | 
						|
        self.dbclient = self.testClient.getDbConnection()
 | 
						|
        self.account = Account.create(
 | 
						|
                                     self.apiclient,
 | 
						|
                                     self.services["account"],
 | 
						|
                                     admin=True,
 | 
						|
                                     domainid=self.domain.id
 | 
						|
                                     )
 | 
						|
        self.cleanup = [self.account]
 | 
						|
        return
 | 
						|
 | 
						|
    def tearDown(self):
 | 
						|
        try:
 | 
						|
            self.debug("Cleaning up the resources")
 | 
						|
            #Clean up, terminate the created network offerings
 | 
						|
            cleanup_resources(self.apiclient, self.cleanup)
 | 
						|
            interval = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='network.gc.interval'
 | 
						|
                                    )
 | 
						|
            wait = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='network.gc.wait'
 | 
						|
                                    )
 | 
						|
            # Sleep to ensure that all resources are deleted
 | 
						|
            time.sleep(int(interval[0].value) + int(wait[0].value))
 | 
						|
            self.debug("Cleanup complete!")
 | 
						|
        except Exception as e:
 | 
						|
            raise Exception("Warning: Exception during cleanup : %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_pf_on_ip_with_lb(self):
 | 
						|
        """Test Create a port forwarding rule  on an Ip address that already has a LB rule.
 | 
						|
        """
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Deploy the first VM using a network from the above created
 | 
						|
        #    Network offering.
 | 
						|
        # 2. Deploy another VM.
 | 
						|
        # 3. Create PF rule on Ip with LB rule. User should NOT be
 | 
						|
        #    allowed to create an LB rule on Ip with LB
 | 
						|
 | 
						|
        # Creating network using the network offering created
 | 
						|
        self.debug("Creating network with network offering: %s" %
 | 
						|
                                                    self.network_offering.id)
 | 
						|
        self.network = Network.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["network"],
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    domainid=self.account.domainid,
 | 
						|
                                    networkofferingid=self.network_offering.id,
 | 
						|
                                    zoneid=self.zone.id
 | 
						|
                                    )
 | 
						|
        self.debug("Created network with ID: %s" % self.network.id)
 | 
						|
 | 
						|
        self.debug("Deploying VM in account: %s" % self.account.name)
 | 
						|
 | 
						|
        # Spawn an instance in that network
 | 
						|
        virtual_machine_1 = VirtualMachine.create(
 | 
						|
                                  self.apiclient,
 | 
						|
                                  self.services["virtual_machine"],
 | 
						|
                                  accountid=self.account.name,
 | 
						|
                                  domainid=self.account.domainid,
 | 
						|
                                  serviceofferingid=self.service_offering.id,
 | 
						|
                                  networkids=[str(self.network.id)]
 | 
						|
                                  )
 | 
						|
        self.debug("Deployed VM in network: %s" % self.network.id)
 | 
						|
        list_vm_response = VirtualMachine.list(
 | 
						|
                                        self.apiclient,
 | 
						|
                                        id=virtual_machine_1.id
 | 
						|
                                        )
 | 
						|
 | 
						|
        self.debug(
 | 
						|
                "Verify listVirtualMachines response for virtual machine: %s" \
 | 
						|
                % virtual_machine_1.id
 | 
						|
            )
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            isinstance(list_vm_response, list),
 | 
						|
                            True,
 | 
						|
                            "Check list response returns a valid list"
 | 
						|
                        )
 | 
						|
        vm_response = list_vm_response[0]
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            vm_response.state,
 | 
						|
                            "Running",
 | 
						|
                            "VM state should be running after deployment"
 | 
						|
                        )
 | 
						|
 | 
						|
        self.debug("Deploying another VM in account: %s" %
 | 
						|
                                            self.account.name)
 | 
						|
 | 
						|
        # Spawn an instance in that network
 | 
						|
        virtual_machine_2 = VirtualMachine.create(
 | 
						|
                                  self.apiclient,
 | 
						|
                                  self.services["virtual_machine"],
 | 
						|
                                  accountid=self.account.name,
 | 
						|
                                  domainid=self.account.domainid,
 | 
						|
                                  serviceofferingid=self.service_offering.id,
 | 
						|
                                  networkids=[str(self.network.id)]
 | 
						|
                                  )
 | 
						|
        self.debug("Deployed VM in network: %s" % self.network.id)
 | 
						|
        list_vm_response = VirtualMachine.list(
 | 
						|
                                        self.apiclient,
 | 
						|
                                        id=virtual_machine_2.id
 | 
						|
                                        )
 | 
						|
 | 
						|
        self.debug(
 | 
						|
                "Verify listVirtualMachines response for virtual machine: %s" \
 | 
						|
                % virtual_machine_2.id
 | 
						|
            )
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            isinstance(list_vm_response, list),
 | 
						|
                            True,
 | 
						|
                            "Check list response returns a valid list"
 | 
						|
                        )
 | 
						|
        vm_response = list_vm_response[0]
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            vm_response.state,
 | 
						|
                            "Running",
 | 
						|
                            "VM state should be running after deployment"
 | 
						|
                        )
 | 
						|
        self.debug("Associating public IP for network: %s" % self.network.id)
 | 
						|
 | 
						|
        ip_with_lb_rule = PublicIPAddress.create(
 | 
						|
                                self.apiclient,
 | 
						|
                                accountid=self.account.name,
 | 
						|
                                zoneid=self.zone.id,
 | 
						|
                                domainid=self.account.domainid,
 | 
						|
                                networkid=self.network.id
 | 
						|
                                )
 | 
						|
        self.debug("Associated %s with network %s" % (
 | 
						|
                                        ip_with_lb_rule.ipaddress.ipaddress,
 | 
						|
                                        self.network.id
 | 
						|
                                        ))
 | 
						|
        self.debug("Creating LB rule for IP address: %s" %
 | 
						|
                                        ip_with_lb_rule.ipaddress.ipaddress)
 | 
						|
 | 
						|
        LoadBalancerRule.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["lbrule"],
 | 
						|
                                    ipaddressid=ip_with_lb_rule.ipaddress.id,
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    networkid=self.network.id
 | 
						|
                                )
 | 
						|
 | 
						|
        self.debug("Trying to create PF rule on IP with LB rule: %s" %
 | 
						|
                                        ip_with_lb_rule.ipaddress.ipaddress)
 | 
						|
 | 
						|
        with self.assertRaises(Exception):
 | 
						|
            NATRule.create(
 | 
						|
                         self.apiclient,
 | 
						|
                         virtual_machine,
 | 
						|
                         self.services["natrule"],
 | 
						|
                         ipaddressid=ip_with_lb_rule.ipaddress.id
 | 
						|
                      )
 | 
						|
        return
 | 
						|
 | 
						|
 | 
						|
class TestLbOnNonSourceNat(cloudstackTestCase):
 | 
						|
 | 
						|
    @classmethod
 | 
						|
    def setUpClass(cls):
 | 
						|
        cls._cleanup = []
 | 
						|
        cls.testClient = super(TestLbOnNonSourceNat, cls).getClsTestClient()
 | 
						|
        cls.api_client = cls.testClient.getApiClient()
 | 
						|
 | 
						|
        cls.services = Services().services
 | 
						|
        # Get Zone, Domain and templates
 | 
						|
        cls.domain = get_domain(cls.api_client)
 | 
						|
        cls.zone = get_zone(cls.api_client, cls.testClient.getZoneForTests())
 | 
						|
        cls.template = get_template(
 | 
						|
                            cls.api_client,
 | 
						|
                            cls.zone.id,
 | 
						|
                            cls.services["ostype"]
 | 
						|
                            )
 | 
						|
        try:
 | 
						|
           cls.netscaler = add_netscaler(cls.api_client, cls.zone.id, cls.services["netscaler"])
 | 
						|
           cls._cleanup.append(cls.netscaler)
 | 
						|
           cls.network_offering = NetworkOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["network_offering"],
 | 
						|
                                            conservemode=False
 | 
						|
                                            )
 | 
						|
           # Enable Network offering
 | 
						|
           cls.network_offering.update(cls.api_client, state='Enabled')
 | 
						|
           cls.services["virtual_machine"]["zoneid"] = cls.zone.id
 | 
						|
           cls.services["virtual_machine"]["template"] = cls.template.id
 | 
						|
 | 
						|
           cls.service_offering = ServiceOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["service_offering"]
 | 
						|
                                            )
 | 
						|
        except Exception as e:
 | 
						|
           cls.tearDownClass()
 | 
						|
           raise Exception ("Warning: Exception in setUpClass: %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @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)
 | 
						|
        return
 | 
						|
 | 
						|
    def setUp(self):
 | 
						|
        self.apiclient = self.testClient.getApiClient()
 | 
						|
        self.dbclient = self.testClient.getDbConnection()
 | 
						|
        self.account = Account.create(
 | 
						|
                                     self.apiclient,
 | 
						|
                                     self.services["account"],
 | 
						|
                                     admin=True,
 | 
						|
                                     domainid=self.domain.id
 | 
						|
                                     )
 | 
						|
        self.cleanup = [self.account]
 | 
						|
        return
 | 
						|
 | 
						|
    def tearDown(self):
 | 
						|
        try:
 | 
						|
            self.debug("Cleaning up the resources")
 | 
						|
            #Clean up, terminate the created network offerings
 | 
						|
            cleanup_resources(self.apiclient, self.cleanup)
 | 
						|
            interval = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='network.gc.interval'
 | 
						|
                                    )
 | 
						|
            wait = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='network.gc.wait'
 | 
						|
                                    )
 | 
						|
            # Sleep to ensure that all resources are deleted
 | 
						|
            time.sleep(int(interval[0].value) + int(wait[0].value))
 | 
						|
            self.debug("Cleanup complete!")
 | 
						|
        except Exception as e:
 | 
						|
            raise Exception("Warning: Exception during cleanup : %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_lb_on_non_source_nat(self):
 | 
						|
        """Test Create LB rule for non-sourceNat IP address
 | 
						|
        """
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Deploy the first VM using a network from the above created
 | 
						|
        #    Network offering.
 | 
						|
        # 2. Deploy another VM.
 | 
						|
        # 3. Acquire Ip address and create LB rule on it. LB rule should be
 | 
						|
        #    created successfully
 | 
						|
 | 
						|
        # Creating network using the network offering created
 | 
						|
        self.debug("Creating network with network offering: %s" %
 | 
						|
                                                    self.network_offering.id)
 | 
						|
        self.network = Network.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["network"],
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    domainid=self.account.domainid,
 | 
						|
                                    networkofferingid=self.network_offering.id,
 | 
						|
                                    zoneid=self.zone.id
 | 
						|
                                    )
 | 
						|
        self.debug("Created network with ID: %s" % self.network.id)
 | 
						|
 | 
						|
        self.debug("Deploying VM in account: %s" % self.account.name)
 | 
						|
 | 
						|
        # Spawn an instance in that network
 | 
						|
        virtual_machine_1 = VirtualMachine.create(
 | 
						|
                                  self.apiclient,
 | 
						|
                                  self.services["virtual_machine"],
 | 
						|
                                  accountid=self.account.name,
 | 
						|
                                  domainid=self.account.domainid,
 | 
						|
                                  serviceofferingid=self.service_offering.id,
 | 
						|
                                  networkids=[str(self.network.id)]
 | 
						|
                                  )
 | 
						|
        self.debug("Deployed VM in network: %s" % self.network.id)
 | 
						|
        list_vm_response = VirtualMachine.list(
 | 
						|
                                        self.apiclient,
 | 
						|
                                        id=virtual_machine_1.id
 | 
						|
                                        )
 | 
						|
 | 
						|
        self.debug(
 | 
						|
                "Verify listVirtualMachines response for virtual machine: %s" \
 | 
						|
                % virtual_machine_1.id
 | 
						|
            )
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            isinstance(list_vm_response, list),
 | 
						|
                            True,
 | 
						|
                            "Check list response returns a valid list"
 | 
						|
                        )
 | 
						|
        vm_response = list_vm_response[0]
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            vm_response.state,
 | 
						|
                            "Running",
 | 
						|
                            "VM state should be running after deployment"
 | 
						|
                        )
 | 
						|
 | 
						|
        self.debug("Deploying another VM in account: %s" %
 | 
						|
                                            self.account.name)
 | 
						|
 | 
						|
        # Spawn an instance in that network
 | 
						|
        virtual_machine_2 = VirtualMachine.create(
 | 
						|
                                  self.apiclient,
 | 
						|
                                  self.services["virtual_machine"],
 | 
						|
                                  accountid=self.account.name,
 | 
						|
                                  domainid=self.account.domainid,
 | 
						|
                                  serviceofferingid=self.service_offering.id,
 | 
						|
                                  networkids=[str(self.network.id)]
 | 
						|
                                  )
 | 
						|
        self.debug("Deployed VM in network: %s" % self.network.id)
 | 
						|
        list_vm_response = VirtualMachine.list(
 | 
						|
                                        self.apiclient,
 | 
						|
                                        id=virtual_machine_2.id
 | 
						|
                                        )
 | 
						|
 | 
						|
        self.debug(
 | 
						|
                "Verify listVirtualMachines response for virtual machine: %s" \
 | 
						|
                % virtual_machine_2.id
 | 
						|
            )
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            isinstance(list_vm_response, list),
 | 
						|
                            True,
 | 
						|
                            "Check list response returns a valid list"
 | 
						|
                        )
 | 
						|
        vm_response = list_vm_response[0]
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            vm_response.state,
 | 
						|
                            "Running",
 | 
						|
                            "VM state should be running after deployment"
 | 
						|
                        )
 | 
						|
        self.debug("Associating public IP for network: %s" % self.network.id)
 | 
						|
 | 
						|
        ip_with_lb_rule = PublicIPAddress.create(
 | 
						|
                                self.apiclient,
 | 
						|
                                accountid=self.account.name,
 | 
						|
                                zoneid=self.zone.id,
 | 
						|
                                domainid=self.account.domainid,
 | 
						|
                                networkid=self.network.id
 | 
						|
                                )
 | 
						|
        self.debug("Associated %s with network %s" % (
 | 
						|
                                        ip_with_lb_rule.ipaddress.ipaddress,
 | 
						|
                                        self.network.id
 | 
						|
                                        ))
 | 
						|
        self.debug("Creating LB rule for IP address: %s" %
 | 
						|
                                        ip_with_lb_rule.ipaddress.ipaddress)
 | 
						|
 | 
						|
        lb_rule = LoadBalancerRule.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["lbrule"],
 | 
						|
                                    ipaddressid=ip_with_lb_rule.ipaddress.id,
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    networkid=self.network.id
 | 
						|
                                )
 | 
						|
 | 
						|
        self.debug("Trying to create PF rule on IP with LB rule: %s" %
 | 
						|
                                        ip_with_lb_rule.ipaddress.ipaddress)
 | 
						|
 | 
						|
        lb_rules = LoadBalancerRule.list(
 | 
						|
                                         self.apiclient,
 | 
						|
                                         id=lb_rule.id,
 | 
						|
                                         listall=True
 | 
						|
                                         )
 | 
						|
        self.assertEqual(
 | 
						|
                         isinstance(lb_rules, list),
 | 
						|
                         True,
 | 
						|
                         "List LB rules should return a newly created LB rule"
 | 
						|
                         )
 | 
						|
        return
 | 
						|
 | 
						|
 | 
						|
class TestAddMultipleVmsLb(cloudstackTestCase):
 | 
						|
 | 
						|
    @classmethod
 | 
						|
    def setUpClass(cls):
 | 
						|
        cls._cleanup = []
 | 
						|
        cls.testClient = super(TestAddMultipleVmsLb, cls).getClsTestClient()
 | 
						|
        cls.api_client = cls.testClient.getApiClient()
 | 
						|
 | 
						|
        cls.services = Services().services
 | 
						|
        # Get Zone, Domain and templates
 | 
						|
        cls.domain = get_domain(cls.api_client)
 | 
						|
        cls.zone = get_zone(cls.api_client, cls.testClient.getZoneForTests())
 | 
						|
        cls.template = get_template(
 | 
						|
                            cls.api_client,
 | 
						|
                            cls.zone.id,
 | 
						|
                            cls.services["ostype"]
 | 
						|
                            )
 | 
						|
        try:
 | 
						|
           cls.netscaler = add_netscaler(cls.api_client, cls.zone.id, cls.services["netscaler"])
 | 
						|
           cls._cleanup.append(cls.netscaler)
 | 
						|
           cls.network_offering = NetworkOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["network_offering"],
 | 
						|
                                            conservemode=True
 | 
						|
                                            )
 | 
						|
           # Enable Network offering
 | 
						|
           cls.network_offering.update(cls.api_client, state='Enabled')
 | 
						|
           cls.services["virtual_machine"]["zoneid"] = cls.zone.id
 | 
						|
           cls.services["virtual_machine"]["template"] = cls.template.id
 | 
						|
 | 
						|
           cls.service_offering = ServiceOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["service_offering"]
 | 
						|
                                            )
 | 
						|
        except Exception as e:
 | 
						|
           cls.tearDownClass()
 | 
						|
           raise Exception ("Warning: Exception in setUpClass: %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @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)
 | 
						|
        return
 | 
						|
 | 
						|
    def setUp(self):
 | 
						|
        self.apiclient = self.testClient.getApiClient()
 | 
						|
        self.dbclient = self.testClient.getDbConnection()
 | 
						|
        self.account = Account.create(
 | 
						|
                                     self.apiclient,
 | 
						|
                                     self.services["account"],
 | 
						|
                                     admin=True,
 | 
						|
                                     domainid=self.domain.id
 | 
						|
                                     )
 | 
						|
        self.cleanup = [self.account]
 | 
						|
        return
 | 
						|
 | 
						|
    def tearDown(self):
 | 
						|
        try:
 | 
						|
            self.debug("Cleaning up the resources")
 | 
						|
            #Clean up, terminate the created network offerings
 | 
						|
            cleanup_resources(self.apiclient, self.cleanup)
 | 
						|
            interval = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='network.gc.interval'
 | 
						|
                                    )
 | 
						|
            wait = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='network.gc.wait'
 | 
						|
                                    )
 | 
						|
            # Sleep to ensure that all resources are deleted
 | 
						|
            time.sleep(int(interval[0].value) + int(wait[0].value))
 | 
						|
            self.debug("Cleanup complete!")
 | 
						|
        except Exception as e:
 | 
						|
            raise Exception("Warning: Exception during cleanup : %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_add_multiple_vms_lb(self):
 | 
						|
        """Test Add multiple Vms to an existing LB rule.
 | 
						|
        """
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Deploy the first VM using a network from the above created
 | 
						|
        #    Network offering.
 | 
						|
        # 2. Deploy another VM.
 | 
						|
        # 3. Acquire Ip address and create LB rule on it. Add multiple VMs to
 | 
						|
        #    this rule. LB rule should be
 | 
						|
        # In Netscaler: For every Vm added to the LB rule:
 | 
						|
        # 1. A server and service instance is created using guest VMs IP and
 | 
						|
        #    port number on the Netscaler LB device,
 | 
						|
        # 2. This service is bound to lb virtual server corresponding to lb
 | 
						|
        #    rule.
 | 
						|
 | 
						|
        # Creating network using the network offering created
 | 
						|
        self.debug("Creating network with network offering: %s" %
 | 
						|
                                                    self.network_offering.id)
 | 
						|
        self.network = Network.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["network"],
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    domainid=self.account.domainid,
 | 
						|
                                    networkofferingid=self.network_offering.id,
 | 
						|
                                    zoneid=self.zone.id
 | 
						|
                                    )
 | 
						|
        self.debug("Created network with ID: %s" % self.network.id)
 | 
						|
 | 
						|
        self.debug("Deploying VM in account: %s" % self.account.name)
 | 
						|
 | 
						|
        # Spawn an instance in that network
 | 
						|
        virtual_machine_1 = VirtualMachine.create(
 | 
						|
                                  self.apiclient,
 | 
						|
                                  self.services["virtual_machine"],
 | 
						|
                                  accountid=self.account.name,
 | 
						|
                                  domainid=self.account.domainid,
 | 
						|
                                  serviceofferingid=self.service_offering.id,
 | 
						|
                                  networkids=[str(self.network.id)]
 | 
						|
                                  )
 | 
						|
        self.debug("Deployed VM in network: %s" % self.network.id)
 | 
						|
        list_vm_response = VirtualMachine.list(
 | 
						|
                                        self.apiclient,
 | 
						|
                                        id=virtual_machine_1.id
 | 
						|
                                        )
 | 
						|
 | 
						|
        self.debug(
 | 
						|
                "Verify listVirtualMachines response for virtual machine: %s" \
 | 
						|
                % virtual_machine_1.id
 | 
						|
            )
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            isinstance(list_vm_response, list),
 | 
						|
                            True,
 | 
						|
                            "Check list response returns a valid list"
 | 
						|
                        )
 | 
						|
        vm_response = list_vm_response[0]
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            vm_response.state,
 | 
						|
                            "Running",
 | 
						|
                            "VM state should be running after deployment"
 | 
						|
                        )
 | 
						|
 | 
						|
        self.debug("Deploying another VM in account: %s" %
 | 
						|
                                            self.account.name)
 | 
						|
 | 
						|
        # Spawn an instance in that network
 | 
						|
        virtual_machine_2 = VirtualMachine.create(
 | 
						|
                                  self.apiclient,
 | 
						|
                                  self.services["virtual_machine"],
 | 
						|
                                  accountid=self.account.name,
 | 
						|
                                  domainid=self.account.domainid,
 | 
						|
                                  serviceofferingid=self.service_offering.id,
 | 
						|
                                  networkids=[str(self.network.id)]
 | 
						|
                                  )
 | 
						|
        self.debug("Deployed VM in network: %s" % self.network.id)
 | 
						|
        list_vm_response = VirtualMachine.list(
 | 
						|
                                        self.apiclient,
 | 
						|
                                        id=virtual_machine_2.id
 | 
						|
                                        )
 | 
						|
 | 
						|
        self.debug(
 | 
						|
                "Verify listVirtualMachines response for virtual machine: %s" \
 | 
						|
                % virtual_machine_2.id
 | 
						|
            )
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            isinstance(list_vm_response, list),
 | 
						|
                            True,
 | 
						|
                            "Check list response returns a valid list"
 | 
						|
                        )
 | 
						|
        vm_response = list_vm_response[0]
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            vm_response.state,
 | 
						|
                            "Running",
 | 
						|
                            "VM state should be running after deployment"
 | 
						|
                        )
 | 
						|
        self.debug("Associating public IP for network: %s" % self.network.id)
 | 
						|
 | 
						|
        ip_with_lb_rule = PublicIPAddress.create(
 | 
						|
                                self.apiclient,
 | 
						|
                                accountid=self.account.name,
 | 
						|
                                zoneid=self.zone.id,
 | 
						|
                                domainid=self.account.domainid,
 | 
						|
                                networkid=self.network.id
 | 
						|
                                )
 | 
						|
        self.debug("Associated %s with network %s" % (
 | 
						|
                                        ip_with_lb_rule.ipaddress.ipaddress,
 | 
						|
                                        self.network.id
 | 
						|
                                        ))
 | 
						|
        self.debug("Creating LB rule for IP address: %s" %
 | 
						|
                                        ip_with_lb_rule.ipaddress.ipaddress)
 | 
						|
 | 
						|
        lb_rule = LoadBalancerRule.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["lbrule"],
 | 
						|
                                    ipaddressid=ip_with_lb_rule.ipaddress.id,
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    networkid=self.network.id
 | 
						|
                                )
 | 
						|
 | 
						|
        self.debug("Trying to create PF rule on IP with LB rule: %s" %
 | 
						|
                                        ip_with_lb_rule.ipaddress.ipaddress)
 | 
						|
 | 
						|
        lb_rules = LoadBalancerRule.list(
 | 
						|
                                         self.apiclient,
 | 
						|
                                         id=lb_rule.id,
 | 
						|
                                         listall=True
 | 
						|
                                         )
 | 
						|
        self.assertEqual(
 | 
						|
                         isinstance(lb_rules, list),
 | 
						|
                         True,
 | 
						|
                         "List LB rules should return a newly created LB rule"
 | 
						|
                         )
 | 
						|
        self.debug("Assigning virtual machines to LB rule")
 | 
						|
        lb_rule.assign(self.apiclient, [virtual_machine_1, virtual_machine_2])
 | 
						|
 | 
						|
        self.debug("SSH into netscaler: %s" %
 | 
						|
                                    self.services["netscaler"]["ipaddress"])
 | 
						|
        try:
 | 
						|
            ssh_client = SshClient(
 | 
						|
                                    self.services["netscaler"]["ipaddress"],
 | 
						|
                                    22,
 | 
						|
                                    self.services["netscaler"]["username"],
 | 
						|
                                    self.services["netscaler"]["password"],
 | 
						|
                                    )
 | 
						|
            self.debug("command: show server")
 | 
						|
            res = ssh_client.execute("show server")
 | 
						|
            result = str(res)
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
 | 
						|
            self.assertEqual(
 | 
						|
                    result.count(virtual_machine_1.ipaddress),
 | 
						|
                    2,
 | 
						|
                    "Server must be configured for virtual machines"
 | 
						|
                    )
 | 
						|
            self.assertEqual(
 | 
						|
                    result.count(virtual_machine_2.ipaddress),
 | 
						|
                    2,
 | 
						|
                    "Server must be configured for virtual machines"
 | 
						|
                    )
 | 
						|
 | 
						|
            self.debug("Command:show service")
 | 
						|
            res = ssh_client.execute("show service")
 | 
						|
 | 
						|
            result = str(res)
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
 | 
						|
            self.assertEqual(
 | 
						|
                    result.count(virtual_machine_1.ipaddress),
 | 
						|
                    3,
 | 
						|
                    "Service must be configured for virtual machines"
 | 
						|
                    )
 | 
						|
            self.assertEqual(
 | 
						|
                    result.count(virtual_machine_2.ipaddress),
 | 
						|
                    3,
 | 
						|
                    "Service must be configured for virtual machines"
 | 
						|
                    )
 | 
						|
            self.debug("Command:show lb vserver")
 | 
						|
            res = ssh_client.execute("show lb vserver")
 | 
						|
 | 
						|
            result = str(res)
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
 | 
						|
            self.assertEqual(
 | 
						|
                    result.count(ip_with_lb_rule.ipaddress.ipaddress),
 | 
						|
                    2,
 | 
						|
                    "virtual server must be configured for public IP address"
 | 
						|
                    )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("SSH Access failed for %s: %s" % \
 | 
						|
                      (self.services["netscaler"]["ipaddress"], e))
 | 
						|
        return
 | 
						|
 | 
						|
class TestMultipleLbRules(cloudstackTestCase):
 | 
						|
 | 
						|
    @classmethod
 | 
						|
    def setUpClass(cls):
 | 
						|
        cls._cleanup = []
 | 
						|
        cls.testClient = super(TestMultipleLbRules, cls).getClsTestClient()
 | 
						|
        cls.api_client = cls.testClient.getApiClient()
 | 
						|
 | 
						|
        cls.services = Services().services
 | 
						|
        # Get Zone, Domain and templates
 | 
						|
        cls.domain = get_domain(cls.api_client)
 | 
						|
        cls.zone = get_zone(cls.api_client, cls.testClient.getZoneForTests())
 | 
						|
        cls.template = get_template(
 | 
						|
                            cls.api_client,
 | 
						|
                            cls.zone.id,
 | 
						|
                            cls.services["ostype"]
 | 
						|
                            )
 | 
						|
        try:
 | 
						|
           cls.netscaler = add_netscaler(cls.api_client, cls.zone.id, cls.services["netscaler"])
 | 
						|
           cls._cleanup.append(cls.netscaler)
 | 
						|
           cls.network_offering = NetworkOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["network_offering"],
 | 
						|
                                            conservemode=True
 | 
						|
                                            )
 | 
						|
           # Enable Network offering
 | 
						|
           cls.network_offering.update(cls.api_client, state='Enabled')
 | 
						|
           cls.services["virtual_machine"]["zoneid"] = cls.zone.id
 | 
						|
           cls.services["virtual_machine"]["template"] = cls.template.id
 | 
						|
 | 
						|
           cls.service_offering = ServiceOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["service_offering"]
 | 
						|
                                            )
 | 
						|
        except Exception as e:
 | 
						|
           cls.tearDownClass()
 | 
						|
           raise Exception ("Warning: Exception in setUpClass: %s" % e)
 | 
						|
 | 
						|
        return
 | 
						|
 | 
						|
    @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)
 | 
						|
        return
 | 
						|
 | 
						|
    def setUp(self):
 | 
						|
        self.apiclient = self.testClient.getApiClient()
 | 
						|
        self.dbclient = self.testClient.getDbConnection()
 | 
						|
        self.account = Account.create(
 | 
						|
                                     self.apiclient,
 | 
						|
                                     self.services["account"],
 | 
						|
                                     admin=True,
 | 
						|
                                     domainid=self.domain.id
 | 
						|
                                     )
 | 
						|
        self.cleanup = [self.account]
 | 
						|
        return
 | 
						|
 | 
						|
    def tearDown(self):
 | 
						|
        try:
 | 
						|
            self.debug("Cleaning up the resources")
 | 
						|
            #Clean up, terminate the created network offerings
 | 
						|
            cleanup_resources(self.apiclient, self.cleanup)
 | 
						|
            interval = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='network.gc.interval'
 | 
						|
                                    )
 | 
						|
            wait = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='network.gc.wait'
 | 
						|
                                    )
 | 
						|
            # Sleep to ensure that all resources are deleted
 | 
						|
            time.sleep(int(interval[0].value) + int(wait[0].value))
 | 
						|
            self.debug("Cleanup complete!")
 | 
						|
        except Exception as e:
 | 
						|
            raise Exception("Warning: Exception during cleanup : %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_multiple_lb_publicip(self):
 | 
						|
        """Test Create multiple LB rules using different public Ips acquired
 | 
						|
        """
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Deploy the first VM using a network from the above created
 | 
						|
        #    Network offering.
 | 
						|
        # 2. Deploy few more VMs.
 | 
						|
        # 3. Acquire an Ipaddress and create an LB rule for multiple Vms.
 | 
						|
        #    Repeat step2 for few times Requests to all these LB rules should
 | 
						|
        #    be serviced correctly.
 | 
						|
 | 
						|
        # Creating network using the network offering created
 | 
						|
        self.debug("Creating network with network offering: %s" %
 | 
						|
                                                    self.network_offering.id)
 | 
						|
        self.network = Network.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["network"],
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    domainid=self.account.domainid,
 | 
						|
                                    networkofferingid=self.network_offering.id,
 | 
						|
                                    zoneid=self.zone.id
 | 
						|
                                    )
 | 
						|
        self.debug("Created network with ID: %s" % self.network.id)
 | 
						|
 | 
						|
        self.debug("Deploying VM in account: %s" % self.account.name)
 | 
						|
 | 
						|
        # Spawn an instance in that network
 | 
						|
        virtual_machine_1 = VirtualMachine.create(
 | 
						|
                                  self.apiclient,
 | 
						|
                                  self.services["virtual_machine"],
 | 
						|
                                  accountid=self.account.name,
 | 
						|
                                  domainid=self.account.domainid,
 | 
						|
                                  serviceofferingid=self.service_offering.id,
 | 
						|
                                  networkids=[str(self.network.id)]
 | 
						|
                                  )
 | 
						|
        self.debug("Deployed VM in network: %s" % self.network.id)
 | 
						|
        list_vm_response = VirtualMachine.list(
 | 
						|
                                        self.apiclient,
 | 
						|
                                        id=virtual_machine_1.id
 | 
						|
                                        )
 | 
						|
 | 
						|
        self.debug(
 | 
						|
                "Verify listVirtualMachines response for virtual machine: %s" \
 | 
						|
                % virtual_machine_1.id
 | 
						|
            )
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            isinstance(list_vm_response, list),
 | 
						|
                            True,
 | 
						|
                            "Check list response returns a valid list"
 | 
						|
                        )
 | 
						|
        vm_response = list_vm_response[0]
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            vm_response.state,
 | 
						|
                            "Running",
 | 
						|
                            "VM state should be running after deployment"
 | 
						|
                        )
 | 
						|
 | 
						|
        self.debug("Deploying another VM in account: %s" %
 | 
						|
                                            self.account.name)
 | 
						|
 | 
						|
        # Spawn an instance in that network
 | 
						|
        virtual_machine_2 = VirtualMachine.create(
 | 
						|
                                  self.apiclient,
 | 
						|
                                  self.services["virtual_machine"],
 | 
						|
                                  accountid=self.account.name,
 | 
						|
                                  domainid=self.account.domainid,
 | 
						|
                                  serviceofferingid=self.service_offering.id,
 | 
						|
                                  networkids=[str(self.network.id)]
 | 
						|
                                  )
 | 
						|
        self.debug("Deployed VM in network: %s" % self.network.id)
 | 
						|
        list_vm_response = VirtualMachine.list(
 | 
						|
                                        self.apiclient,
 | 
						|
                                        id=virtual_machine_2.id
 | 
						|
                                        )
 | 
						|
 | 
						|
        self.debug(
 | 
						|
                "Verify listVirtualMachines response for virtual machine: %s" \
 | 
						|
                % virtual_machine_2.id
 | 
						|
            )
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            isinstance(list_vm_response, list),
 | 
						|
                            True,
 | 
						|
                            "Check list response returns a valid list"
 | 
						|
                        )
 | 
						|
        vm_response = list_vm_response[0]
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            vm_response.state,
 | 
						|
                            "Running",
 | 
						|
                            "VM state should be running after deployment"
 | 
						|
                        )
 | 
						|
        self.debug(
 | 
						|
                "Associating first public IP for network: %s" %
 | 
						|
                                                            self.network.id)
 | 
						|
 | 
						|
        public_ip_1 = PublicIPAddress.create(
 | 
						|
                                self.apiclient,
 | 
						|
                                accountid=self.account.name,
 | 
						|
                                zoneid=self.zone.id,
 | 
						|
                                domainid=self.account.domainid,
 | 
						|
                                networkid=self.network.id
 | 
						|
                                )
 | 
						|
        self.debug("Associated %s with network %s" % (
 | 
						|
                                        public_ip_1.ipaddress.ipaddress,
 | 
						|
                                        self.network.id
 | 
						|
                                        ))
 | 
						|
 | 
						|
        self.debug(
 | 
						|
            "Creating LB rule for IP address: %s with round robin algo" %
 | 
						|
                                        public_ip_1.ipaddress.ipaddress)
 | 
						|
 | 
						|
        self.services["lbrule"]["alg"] = 'roundrobin'
 | 
						|
        lb_rule = LoadBalancerRule.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["lbrule"],
 | 
						|
                                    ipaddressid=public_ip_1.ipaddress.id,
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    networkid=self.network.id
 | 
						|
                                )
 | 
						|
 | 
						|
        lb_rules = LoadBalancerRule.list(
 | 
						|
                                         self.apiclient,
 | 
						|
                                         id=lb_rule.id,
 | 
						|
                                         listall=True
 | 
						|
                                         )
 | 
						|
        self.assertEqual(
 | 
						|
                         isinstance(lb_rules, list),
 | 
						|
                         True,
 | 
						|
                         "List LB rules should return a newly created LB rule"
 | 
						|
                         )
 | 
						|
        self.debug("Adding %s, %s to the LB rule %s" % (
 | 
						|
                                                    virtual_machine_1.name,
 | 
						|
                                                    virtual_machine_2.name,
 | 
						|
                                                    lb_rule.name
 | 
						|
                                                    ))
 | 
						|
        lb_rule.assign(self.apiclient, [virtual_machine_1, virtual_machine_2])
 | 
						|
 | 
						|
        self.debug(
 | 
						|
                "Associating second public IP for network: %s" %
 | 
						|
                                                            self.network.id)
 | 
						|
 | 
						|
        public_ip_2 = PublicIPAddress.create(
 | 
						|
                                self.apiclient,
 | 
						|
                                accountid=self.account.name,
 | 
						|
                                zoneid=self.zone.id,
 | 
						|
                                domainid=self.account.domainid,
 | 
						|
                                networkid=self.network.id
 | 
						|
                                )
 | 
						|
        self.debug("Associated %s with network %s" % (
 | 
						|
                                        public_ip_2.ipaddress.ipaddress,
 | 
						|
                                        self.network.id
 | 
						|
                                        ))
 | 
						|
 | 
						|
        self.debug(
 | 
						|
            "Creating LB rule for IP address: %s with round robin algo" %
 | 
						|
                                        public_ip_2.ipaddress.ipaddress)
 | 
						|
 | 
						|
        self.services["lbrule"]["alg"] = 'roundrobin'
 | 
						|
        lb_rule = LoadBalancerRule.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["lbrule"],
 | 
						|
                                    ipaddressid=public_ip_2.ipaddress.id,
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    networkid=self.network.id
 | 
						|
                                )
 | 
						|
 | 
						|
        self.debug("Trying to create PF rule on IP with LB rule: %s" %
 | 
						|
                                        public_ip_2.ipaddress.ipaddress)
 | 
						|
 | 
						|
        lb_rules = LoadBalancerRule.list(
 | 
						|
                                         self.apiclient,
 | 
						|
                                         id=lb_rule.id,
 | 
						|
                                         listall=True
 | 
						|
                                         )
 | 
						|
        self.assertEqual(
 | 
						|
                         isinstance(lb_rules, list),
 | 
						|
                         True,
 | 
						|
                         "List LB rules should return a newly created LB rule"
 | 
						|
                         )
 | 
						|
        self.debug("Adding %s, %s to the LB rule %s" % (
 | 
						|
                                                    virtual_machine_1.name,
 | 
						|
                                                    virtual_machine_2.name,
 | 
						|
                                                    lb_rule.name
 | 
						|
                                                    ))
 | 
						|
        lb_rule.assign(self.apiclient, [virtual_machine_1, virtual_machine_2])
 | 
						|
 | 
						|
        try:
 | 
						|
            self.debug(
 | 
						|
                    "Verifying VMs are accessible with different public Ips")
 | 
						|
            hostnames = []
 | 
						|
            ssh = virtual_machine_1.get_ssh_client(
 | 
						|
                                ipaddress=public_ip_1.ipaddress.ipaddress)
 | 
						|
            self.debug("Command: hostname")
 | 
						|
            result = ssh.execute("hostname")
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
            hostnames.append(result)
 | 
						|
 | 
						|
            res = str(result[0])
 | 
						|
            self.assertIn(
 | 
						|
                res,
 | 
						|
                [virtual_machine_1.name, virtual_machine_2.name],
 | 
						|
                "The hostname should match with atleast one of instance name"
 | 
						|
                )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("Exception occured during SSH: %s - %s" % (
 | 
						|
                                        public_ip_1.ipaddress.ipaddress,
 | 
						|
                                        e))
 | 
						|
        try:
 | 
						|
            ssh = virtual_machine_1.get_ssh_client(
 | 
						|
                                ipaddress=public_ip_2.ipaddress.ipaddress,
 | 
						|
                                reconnect=True
 | 
						|
                                )
 | 
						|
            self.debug("Command: hostname")
 | 
						|
            result = ssh.execute("hostname")
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
            hostnames.append(result)
 | 
						|
            self.debug("Hostnames: %s" % str(hostnames))
 | 
						|
 | 
						|
            res = str(result[0])
 | 
						|
            self.assertIn(
 | 
						|
                res,
 | 
						|
                [virtual_machine_1.name, virtual_machine_2.name],
 | 
						|
                "The hostname should match with atleast one of instance name"
 | 
						|
                )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("Exception occured during SSH: %s - %s" % (
 | 
						|
                                        public_ip_2.ipaddress.ipaddress,
 | 
						|
                                        e))
 | 
						|
        return
 | 
						|
 | 
						|
 | 
						|
class TestMultipleLbRulesSameIp(cloudstackTestCase):
 | 
						|
 | 
						|
    @classmethod
 | 
						|
    def setUpClass(cls):
 | 
						|
        cls._cleanup = []
 | 
						|
        cls.testClient = super(TestMultipleLbRulesSameIp, cls).getClsTestClient()
 | 
						|
        cls.api_client = cls.testClient.getApiClient()
 | 
						|
 | 
						|
        cls.services = Services().services
 | 
						|
        # Get Zone, Domain and templates
 | 
						|
        cls.domain = get_domain(cls.api_client)
 | 
						|
        cls.zone = get_zone(cls.api_client, cls.testClient.getZoneForTests())
 | 
						|
        cls.template = get_template(
 | 
						|
                            cls.api_client,
 | 
						|
                            cls.zone.id,
 | 
						|
                            cls.services["ostype"]
 | 
						|
                            )
 | 
						|
 | 
						|
        try:
 | 
						|
           cls.netscaler = add_netscaler(cls.api_client, cls.zone.id, cls.services["netscaler"])
 | 
						|
           cls._cleanup.append(cls.netscaler)
 | 
						|
           cls.network_offering = NetworkOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["network_offering"],
 | 
						|
                                            conservemode=True
 | 
						|
                                            )
 | 
						|
           # Enable Network offering
 | 
						|
           cls.network_offering.update(cls.api_client, state='Enabled')
 | 
						|
           cls.services["virtual_machine"]["zoneid"] = cls.zone.id
 | 
						|
           cls.services["virtual_machine"]["template"] = cls.template.id
 | 
						|
 | 
						|
           cls.service_offering = ServiceOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["service_offering"]
 | 
						|
                                            )
 | 
						|
        except Exception as e:
 | 
						|
           cls.tearDownClass()
 | 
						|
           raise Exception ("Warning: Exception in setUpClass: %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @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)
 | 
						|
        return
 | 
						|
 | 
						|
    def setUp(self):
 | 
						|
        self.apiclient = self.testClient.getApiClient()
 | 
						|
        self.dbclient = self.testClient.getDbConnection()
 | 
						|
        self.account = Account.create(
 | 
						|
                                     self.apiclient,
 | 
						|
                                     self.services["account"],
 | 
						|
                                     admin=True,
 | 
						|
                                     domainid=self.domain.id
 | 
						|
                                     )
 | 
						|
        self.cleanup = [self.account]
 | 
						|
        return
 | 
						|
 | 
						|
    def tearDown(self):
 | 
						|
        try:
 | 
						|
            self.debug("Cleaning up the resources")
 | 
						|
            #Clean up, terminate the created network offerings
 | 
						|
            cleanup_resources(self.apiclient, self.cleanup)
 | 
						|
            interval = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='network.gc.interval'
 | 
						|
                                    )
 | 
						|
            wait = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='network.gc.wait'
 | 
						|
                                    )
 | 
						|
            # Sleep to ensure that all resources are deleted
 | 
						|
            time.sleep(int(interval[0].value) + int(wait[0].value))
 | 
						|
            self.debug("Cleanup complete!")
 | 
						|
        except Exception as e:
 | 
						|
            raise Exception("Warning: Exception during cleanup : %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_multiple_lb_same_publicip(self):
 | 
						|
        """Test Create multiple LB rules using same public Ips on diff ports
 | 
						|
        """
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Deploy the first VM using a network from the above created
 | 
						|
        #    Network offering.
 | 
						|
        # 2. Deploy few more VMs.
 | 
						|
        # 3. Acquire an Ipaddress and create an LB rule for multiple Vms.
 | 
						|
        #    Create another Lb rule on the same Ipaddress pointing to
 | 
						|
        #    different public port. Requests to all these LB rules should be
 | 
						|
        #    serviced correctly.
 | 
						|
 | 
						|
        # Creating network using the network offering created
 | 
						|
        self.debug("Creating network with network offering: %s" %
 | 
						|
                                                    self.network_offering.id)
 | 
						|
        self.network = Network.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["network"],
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    domainid=self.account.domainid,
 | 
						|
                                    networkofferingid=self.network_offering.id,
 | 
						|
                                    zoneid=self.zone.id
 | 
						|
                                    )
 | 
						|
        self.debug("Created network with ID: %s" % self.network.id)
 | 
						|
 | 
						|
        self.debug("Deploying VM in account: %s" % self.account.name)
 | 
						|
 | 
						|
        # Spawn an instance in that network
 | 
						|
        virtual_machine_1 = VirtualMachine.create(
 | 
						|
                                  self.apiclient,
 | 
						|
                                  self.services["virtual_machine"],
 | 
						|
                                  accountid=self.account.name,
 | 
						|
                                  domainid=self.account.domainid,
 | 
						|
                                  serviceofferingid=self.service_offering.id,
 | 
						|
                                  networkids=[str(self.network.id)]
 | 
						|
                                  )
 | 
						|
        self.debug("Deployed VM in network: %s" % self.network.id)
 | 
						|
        list_vm_response = VirtualMachine.list(
 | 
						|
                                        self.apiclient,
 | 
						|
                                        id=virtual_machine_1.id
 | 
						|
                                        )
 | 
						|
 | 
						|
        self.debug(
 | 
						|
                "Verify listVirtualMachines response for virtual machine: %s" \
 | 
						|
                % virtual_machine_1.id
 | 
						|
            )
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            isinstance(list_vm_response, list),
 | 
						|
                            True,
 | 
						|
                            "Check list response returns a valid list"
 | 
						|
                        )
 | 
						|
        vm_response = list_vm_response[0]
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            vm_response.state,
 | 
						|
                            "Running",
 | 
						|
                            "VM state should be running after deployment"
 | 
						|
                        )
 | 
						|
 | 
						|
        self.debug("Deploying another VM in account: %s" %
 | 
						|
                                            self.account.name)
 | 
						|
 | 
						|
        # Spawn an instance in that network
 | 
						|
        virtual_machine_2 = VirtualMachine.create(
 | 
						|
                                  self.apiclient,
 | 
						|
                                  self.services["virtual_machine"],
 | 
						|
                                  accountid=self.account.name,
 | 
						|
                                  domainid=self.account.domainid,
 | 
						|
                                  serviceofferingid=self.service_offering.id,
 | 
						|
                                  networkids=[str(self.network.id)]
 | 
						|
                                  )
 | 
						|
        self.debug("Deployed VM in network: %s" % self.network.id)
 | 
						|
        list_vm_response = VirtualMachine.list(
 | 
						|
                                        self.apiclient,
 | 
						|
                                        id=virtual_machine_2.id
 | 
						|
                                        )
 | 
						|
 | 
						|
        self.debug(
 | 
						|
                "Verify listVirtualMachines response for virtual machine: %s" \
 | 
						|
                % virtual_machine_2.id
 | 
						|
            )
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            isinstance(list_vm_response, list),
 | 
						|
                            True,
 | 
						|
                            "Check list response returns a valid list"
 | 
						|
                        )
 | 
						|
        vm_response = list_vm_response[0]
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            vm_response.state,
 | 
						|
                            "Running",
 | 
						|
                            "VM state should be running after deployment"
 | 
						|
                        )
 | 
						|
        self.debug(
 | 
						|
                "Associating first public IP for network: %s" %
 | 
						|
                                                            self.network.id)
 | 
						|
 | 
						|
        public_ip = PublicIPAddress.create(
 | 
						|
                                self.apiclient,
 | 
						|
                                accountid=self.account.name,
 | 
						|
                                zoneid=self.zone.id,
 | 
						|
                                domainid=self.account.domainid,
 | 
						|
                                networkid=self.network.id
 | 
						|
                                )
 | 
						|
        self.debug("Associated %s with network %s" % (
 | 
						|
                                        public_ip.ipaddress.ipaddress,
 | 
						|
                                        self.network.id
 | 
						|
                                        ))
 | 
						|
 | 
						|
        self.debug(
 | 
						|
            "Creating LB rule for IP address: %s with round robin algo" %
 | 
						|
                                        public_ip.ipaddress.ipaddress)
 | 
						|
 | 
						|
        self.services["lbrule"]["alg"] = 'roundrobin'
 | 
						|
        lb_rule_1 = LoadBalancerRule.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["lbrule"],
 | 
						|
                                    ipaddressid=public_ip.ipaddress.id,
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    networkid=self.network.id
 | 
						|
                                )
 | 
						|
 | 
						|
        self.debug("Trying to create PF rule on IP with LB rule: %s" %
 | 
						|
                                        public_ip.ipaddress.ipaddress)
 | 
						|
 | 
						|
        lb_rules = LoadBalancerRule.list(
 | 
						|
                                         self.apiclient,
 | 
						|
                                         id=lb_rule_1.id,
 | 
						|
                                         listall=True
 | 
						|
                                         )
 | 
						|
        self.assertEqual(
 | 
						|
                         isinstance(lb_rules, list),
 | 
						|
                         True,
 | 
						|
                         "List LB rules should return a newly created LB rule"
 | 
						|
                         )
 | 
						|
        self.debug("Adding %s, %s to the LB rule %s" % (
 | 
						|
                                                    virtual_machine_1.name,
 | 
						|
                                                    virtual_machine_2.name,
 | 
						|
                                                    lb_rule_1.name
 | 
						|
                                                    ))
 | 
						|
        lb_rule_1.assign(self.apiclient, [
 | 
						|
                                          virtual_machine_1,
 | 
						|
                                          virtual_machine_2
 | 
						|
                                          ])
 | 
						|
        self.debug(
 | 
						|
            "Trying to create LB rule on IP: %s with on same ports" %
 | 
						|
                                                public_ip.ipaddress.ipaddress)
 | 
						|
        with self.assertRaises(Exception):
 | 
						|
            LoadBalancerRule.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["lbrule"],
 | 
						|
                                    ipaddressid=public_ip.ipaddress.id,
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    networkid=self.network.id
 | 
						|
                                )
 | 
						|
        self.debug("Create LB rule on same port failed!")
 | 
						|
        self.debug("Creating LB rule on IP: %s & public port: %s" % (
 | 
						|
                                                public_ip.ipaddress.ipaddress,
 | 
						|
                                                str(2222)))
 | 
						|
 | 
						|
        self.services["lbrule"]["alg"] = 'roundrobin'
 | 
						|
        self.services["lbrule"]["publicport"] = 2222
 | 
						|
        self.services["lbrule"]["name"] = 'SSH2'
 | 
						|
        lb_rule_2 = LoadBalancerRule.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["lbrule"],
 | 
						|
                                    ipaddressid=public_ip.ipaddress.id,
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    networkid=self.network.id
 | 
						|
                                )
 | 
						|
 | 
						|
        lb_rules = LoadBalancerRule.list(
 | 
						|
                                         self.apiclient,
 | 
						|
                                         id=lb_rule_2.id,
 | 
						|
                                         listall=True
 | 
						|
                                         )
 | 
						|
        self.assertEqual(
 | 
						|
                         isinstance(lb_rules, list),
 | 
						|
                         True,
 | 
						|
                         "List LB rules should return a newly created LB rule"
 | 
						|
                         )
 | 
						|
        self.debug("Adding %s, %s to the LB rule %s" % (
 | 
						|
                                                    virtual_machine_1.name,
 | 
						|
                                                    virtual_machine_2.name,
 | 
						|
                                                    lb_rule_2.name
 | 
						|
                                                    ))
 | 
						|
        lb_rule_2.assign(self.apiclient, [
 | 
						|
                                          virtual_machine_1,
 | 
						|
                                          virtual_machine_2
 | 
						|
                                          ])
 | 
						|
 | 
						|
        try:
 | 
						|
            self.debug("Verifying VMs are accessible on all LB rules")
 | 
						|
            hostnames = []
 | 
						|
            ssh = virtual_machine_1.get_ssh_client(
 | 
						|
                                    ipaddress=public_ip.ipaddress.ipaddress,
 | 
						|
                                    reconnect=True
 | 
						|
                                    )
 | 
						|
            self.debug("Command: hostname")
 | 
						|
            result = ssh.execute("hostname")
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
            hostnames.append(result)
 | 
						|
 | 
						|
            res = str(result[0])
 | 
						|
            self.assertIn(
 | 
						|
                res,
 | 
						|
                [virtual_machine_1.name, virtual_machine_2.name],
 | 
						|
                "The hostname should match with atleast one of instance name"
 | 
						|
                )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("Exception occured during SSH: %s - %s" % (
 | 
						|
                                        public_ip.ipaddress.ipaddress,
 | 
						|
                                        e))
 | 
						|
        try:
 | 
						|
            ssh = virtual_machine_1.get_ssh_client(
 | 
						|
                                ipaddress=public_ip.ipaddress.ipaddress,
 | 
						|
                                reconnect=True,
 | 
						|
                                port=self.services["lbrule"]["publicport"]
 | 
						|
                                )
 | 
						|
            self.debug("Command: hostname")
 | 
						|
            result = ssh.execute("hostname")
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
            hostnames.append(result)
 | 
						|
            self.debug("Hostnames: %s" % str(hostnames))
 | 
						|
 | 
						|
            res = str(result[0])
 | 
						|
            self.assertIn(
 | 
						|
                res,
 | 
						|
                [virtual_machine_1.name, virtual_machine_2.name],
 | 
						|
                "The hostname should match with atleast one of instance name"
 | 
						|
                )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("Exception occured during SSH: %s - %s" % (
 | 
						|
                                        public_ip.ipaddress.ipaddress,
 | 
						|
                                        e))
 | 
						|
        return
 | 
						|
 | 
						|
 | 
						|
class TestLoadBalancingRule(cloudstackTestCase):
 | 
						|
 | 
						|
    @classmethod
 | 
						|
    def setUpClass(cls):
 | 
						|
        cls._cleanup = []
 | 
						|
        cls.testClient = super(TestLoadBalancingRule, cls).getClsTestClient()
 | 
						|
        cls.api_client = cls.testClient.getApiClient()
 | 
						|
 | 
						|
        cls.services = Services().services
 | 
						|
        # Get Zone, Domain and templates
 | 
						|
        cls.domain = get_domain(cls.api_client)
 | 
						|
        cls.zone = get_zone(cls.api_client, cls.testClient.getZoneForTests())
 | 
						|
        cls.template = get_template(
 | 
						|
                            cls.api_client,
 | 
						|
                            cls.zone.id,
 | 
						|
                            cls.services["ostype"]
 | 
						|
                            )
 | 
						|
        try:
 | 
						|
           cls.netscaler = add_netscaler(cls.api_client, cls.zone.id, cls.services["netscaler"])
 | 
						|
           cls._cleanup.append(cls.netscaler)
 | 
						|
           cls.network_offering = NetworkOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["network_offering"],
 | 
						|
                                            conservemode=True
 | 
						|
                                            )
 | 
						|
           # Enable Network offering
 | 
						|
           cls.network_offering.update(cls.api_client, state='Enabled')
 | 
						|
           cls.services["virtual_machine"]["zoneid"] = cls.zone.id
 | 
						|
           cls.services["virtual_machine"]["template"] = cls.template.id
 | 
						|
 | 
						|
           cls.service_offering = ServiceOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["service_offering"]
 | 
						|
                                            )
 | 
						|
           cls.account = Account.create(
 | 
						|
                                     cls.api_client,
 | 
						|
                                     cls.services["account"],
 | 
						|
                                     admin=True,
 | 
						|
                                     domainid=cls.domain.id
 | 
						|
                                     )
 | 
						|
           cls._cleanup.insert(0,cls.account)
 | 
						|
           # Creating network using the network offering created
 | 
						|
           cls.network = Network.create(
 | 
						|
                                    cls.api_client,
 | 
						|
                                    cls.services["network"],
 | 
						|
                                    accountid=cls.account.name,
 | 
						|
                                    domainid=cls.account.domainid,
 | 
						|
                                    networkofferingid=cls.network_offering.id,
 | 
						|
                                    zoneid=cls.zone.id
 | 
						|
                                    )
 | 
						|
 | 
						|
           # Spawn an instance in that network
 | 
						|
           cls.virtual_machine = VirtualMachine.create(
 | 
						|
                                  cls.api_client,
 | 
						|
                                  cls.services["virtual_machine"],
 | 
						|
                                  accountid=cls.account.name,
 | 
						|
                                  domainid=cls.account.domainid,
 | 
						|
                                  serviceofferingid=cls.service_offering.id,
 | 
						|
                                  networkids=[str(cls.network.id)]
 | 
						|
                                  )
 | 
						|
           cls.public_ip = PublicIPAddress.create(
 | 
						|
                                cls.api_client,
 | 
						|
                                accountid=cls.account.name,
 | 
						|
                                zoneid=cls.zone.id,
 | 
						|
                                domainid=cls.account.domainid,
 | 
						|
                                networkid=cls.network.id
 | 
						|
                                )
 | 
						|
           cls.lb_rule = LoadBalancerRule.create(
 | 
						|
                                    cls.api_client,
 | 
						|
                                    cls.services["lbrule"],
 | 
						|
                                    ipaddressid=cls.public_ip.ipaddress.id,
 | 
						|
                                    accountid=cls.account.name,
 | 
						|
                                    networkid=cls.network.id
 | 
						|
                                )
 | 
						|
        except Exception as e:
 | 
						|
           cls.tearDownClass()
 | 
						|
           raise Exception ("Warning: Exception in setUpClass: %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @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)
 | 
						|
        return
 | 
						|
 | 
						|
    def setUp(self):
 | 
						|
        self.apiclient = self.testClient.getApiClient()
 | 
						|
        self.dbclient = self.testClient.getDbConnection()
 | 
						|
        self.cleanup = []
 | 
						|
        return
 | 
						|
 | 
						|
    def tearDown(self):
 | 
						|
        try:
 | 
						|
            self.debug("Cleaning up the resources")
 | 
						|
            #Clean up, terminate the created network offerings
 | 
						|
            cleanup_resources(self.apiclient, self.cleanup)
 | 
						|
            self.debug("Cleanup complete!")
 | 
						|
        except Exception as e:
 | 
						|
            raise Exception("Warning: Exception during cleanup : %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_01_edit_name_lb_rule(self):
 | 
						|
        """Test edit name of LB rule
 | 
						|
        """
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Create an Lb rule for couple of Vms .
 | 
						|
        # 2. Edit the name of the existing LB rule. When all the existing
 | 
						|
        #    Lbrules are listed , we should see the edited name.
 | 
						|
 | 
						|
        self.debug("Assigning VMs to LB rule: %s" % self.lb_rule.name)
 | 
						|
        self.lb_rule.assign(self.apiclient, [self.virtual_machine])
 | 
						|
 | 
						|
        self.debug("Editing name of the LB rule: %s" % self.lb_rule.name)
 | 
						|
        new_name = random_gen()
 | 
						|
        self.lb_rule.update(self.apiclient, name=new_name)
 | 
						|
 | 
						|
        self.debug("Verifing the name change in list Lb rules call")
 | 
						|
        lb_rules = LoadBalancerRule.list(self.apiclient, id=self.lb_rule.id)
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                         isinstance(lb_rules, list),
 | 
						|
                         True,
 | 
						|
                         "List LB call should return a valid list"
 | 
						|
                         )
 | 
						|
        lb = lb_rules[0]
 | 
						|
        self.assertEqual(
 | 
						|
                         lb.name,
 | 
						|
                         new_name,
 | 
						|
                         "LB name should be updated with the new name"
 | 
						|
                         )
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_02_edit_lb_ports(self):
 | 
						|
        """Test edit public port of LB rule
 | 
						|
        """
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Create an Lb rule for couple of Vms .
 | 
						|
        # 2. Edit the public/private of the existing LB rule. When all the
 | 
						|
        #    existing Lbrules are listed, this should not be allowed.
 | 
						|
 | 
						|
        self.debug("Editing public port of the LB rule: %s" % self.lb_rule.name)
 | 
						|
        port = 8888
 | 
						|
        with self.assertRaises(Exception):
 | 
						|
            self.lb_rule.update(self.apiclient, publicport=port)
 | 
						|
 | 
						|
        self.debug("Editing private port of the LB rule: %s" % self.lb_rule.name)
 | 
						|
        with self.assertRaises(Exception):
 | 
						|
            self.lb_rule.update(self.apiclient, privateport=port)
 | 
						|
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_03_delete_lb_rule(self):
 | 
						|
        """Test delete LB rule
 | 
						|
        """
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Delete existing load balancing rule
 | 
						|
        # 2. In netscaler service and port corresponding to LB rule should get
 | 
						|
        #    deleted. Any request to IP should error out.
 | 
						|
 | 
						|
        self.debug("Deleting existing load balancing rule")
 | 
						|
        self.lb_rule.delete(self.apiclient)
 | 
						|
 | 
						|
        self.debug("SSH into Netscaler to verify other resources are deleted")
 | 
						|
        try:
 | 
						|
            ssh_client = SshClient(
 | 
						|
                                    self.services["netscaler"]["ipaddress"],
 | 
						|
                                    self.services["netscaler"]["port"],
 | 
						|
                                    self.services["netscaler"]["username"],
 | 
						|
                                    self.services["netscaler"]["password"],
 | 
						|
                                    )
 | 
						|
            cmd = "show lb vserver Cloud-VirtualServer-%s-%s" % (
 | 
						|
                                        self.public_ip.ipaddress.ipaddress,
 | 
						|
                                        self.lb_rule.publicport)
 | 
						|
            self.debug("command: %s" % cmd)
 | 
						|
            res = ssh_client.execute(cmd)
 | 
						|
            result = str(res)
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
 | 
						|
            self.assertEqual(
 | 
						|
                    result.count("ERROR: No such resource"),
 | 
						|
                    1,
 | 
						|
                    "Virtual server should get deleted after removing LB rule"
 | 
						|
                    )
 | 
						|
 | 
						|
            cmd = "show ip"
 | 
						|
            self.debug("command: %s" % cmd)
 | 
						|
            res = ssh_client.execute(cmd)
 | 
						|
            result = str(res)
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
 | 
						|
            self.assertEqual(
 | 
						|
                    result.count(self.public_ip.ipaddress.ipaddress),
 | 
						|
                    0,
 | 
						|
                    "Virtual server should get deleted after removing LB rule"
 | 
						|
                    )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("SSH Access failed for %s: %s" % \
 | 
						|
                      (self.services["netscaler"]["ipaddress"], e))
 | 
						|
        return
 | 
						|
 | 
						|
 | 
						|
class TestDeleteCreateLBRule(cloudstackTestCase):
 | 
						|
 | 
						|
    @classmethod
 | 
						|
    def setUpClass(cls):
 | 
						|
        cls._cleanup = []
 | 
						|
        cls.testClient = super(TestDeleteCreateLBRule, cls).getClsTestClient()
 | 
						|
        cls.api_client = cls.testClient.getApiClient()
 | 
						|
 | 
						|
        cls.services = Services().services
 | 
						|
        # Get Zone, Domain and templates
 | 
						|
        cls.domain = get_domain(cls.api_client)
 | 
						|
        cls.zone = get_zone(cls.api_client, cls.testClient.getZoneForTests())
 | 
						|
        cls.template = get_template(
 | 
						|
                            cls.api_client,
 | 
						|
                            cls.zone.id,
 | 
						|
                            cls.services["ostype"]
 | 
						|
                            )
 | 
						|
        try:
 | 
						|
           cls.netscaler = add_netscaler(cls.api_client, cls.zone.id, cls.services["netscaler"])
 | 
						|
           cls._cleanup.append(cls.netscaler)
 | 
						|
           cls.network_offering = NetworkOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["network_offering"],
 | 
						|
                                            conservemode=True
 | 
						|
                                            )
 | 
						|
           # Enable Network offering
 | 
						|
           cls.network_offering.update(cls.api_client, state='Enabled')
 | 
						|
           cls.services["virtual_machine"]["zoneid"] = cls.zone.id
 | 
						|
           cls.services["virtual_machine"]["template"] = cls.template.id
 | 
						|
           cls.service_offering = ServiceOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["service_offering"]
 | 
						|
                                            )
 | 
						|
           cls.account = Account.create(
 | 
						|
                                     cls.api_client,
 | 
						|
                                     cls.services["account"],
 | 
						|
                                     admin=True,
 | 
						|
                                     domainid=cls.domain.id
 | 
						|
                                     )
 | 
						|
           cls._cleanup.insert(0,cls.account)
 | 
						|
           # Creating network using the network offering created
 | 
						|
           cls.network = Network.create(
 | 
						|
                                    cls.api_client,
 | 
						|
                                    cls.services["network"],
 | 
						|
                                    accountid=cls.account.name,
 | 
						|
                                    domainid=cls.account.domainid,
 | 
						|
                                    networkofferingid=cls.network_offering.id,
 | 
						|
                                    zoneid=cls.zone.id
 | 
						|
                                    )
 | 
						|
 | 
						|
           # Spawn an instance in that network
 | 
						|
           cls.virtual_machine = VirtualMachine.create(
 | 
						|
                                  cls.api_client,
 | 
						|
                                  cls.services["virtual_machine"],
 | 
						|
                                  accountid=cls.account.name,
 | 
						|
                                  domainid=cls.account.domainid,
 | 
						|
                                  serviceofferingid=cls.service_offering.id,
 | 
						|
                                  networkids=[str(cls.network.id)]
 | 
						|
                                  )
 | 
						|
           cls.public_ip = PublicIPAddress.create(
 | 
						|
                                cls.api_client,
 | 
						|
                                accountid=cls.account.name,
 | 
						|
                                zoneid=cls.zone.id,
 | 
						|
                                domainid=cls.account.domainid,
 | 
						|
                                networkid=cls.network.id
 | 
						|
                                )
 | 
						|
        except Exception as e:
 | 
						|
           cls.tearDownClass()
 | 
						|
           raise Exception ("Warning: Exception in setUpClass: %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @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)
 | 
						|
        return
 | 
						|
 | 
						|
    def setUp(self):
 | 
						|
        self.apiclient = self.testClient.getApiClient()
 | 
						|
        self.dbclient = self.testClient.getDbConnection()
 | 
						|
        self.lb_rule = LoadBalancerRule.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["lbrule"],
 | 
						|
                                    ipaddressid=self.public_ip.ipaddress.id,
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    networkid=self.network.id
 | 
						|
                                )
 | 
						|
        self.cleanup = []
 | 
						|
        return
 | 
						|
 | 
						|
    def tearDown(self):
 | 
						|
        try:
 | 
						|
            self.debug("Cleaning up the resources")
 | 
						|
            #Clean up, terminate the created network offerings
 | 
						|
            cleanup_resources(self.apiclient, self.cleanup)
 | 
						|
            self.debug("Cleanup complete!")
 | 
						|
        except Exception as e:
 | 
						|
            raise Exception("Warning: Exception during cleanup : %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_01_create_with_same_public_port(self):
 | 
						|
        """Test create LB rule with same public port after deleting rule"""
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Delete existing rule and create exactly same rule with different
 | 
						|
        #    public port
 | 
						|
        # 2. Requests should be served correctly
 | 
						|
 | 
						|
        self.debug("Delete the existing LB rule: %s" % self.lb_rule.name)
 | 
						|
        self.lb_rule.delete(self.apiclient)
 | 
						|
        self.debug("LB rule deleted")
 | 
						|
 | 
						|
        self.debug("Create a new LB rule with different public port")
 | 
						|
        self.services["lbrule"]["publicport"] = 23
 | 
						|
        lb_rule = LoadBalancerRule.create(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    self.services["lbrule"],
 | 
						|
                                    ipaddressid=self.public_ip.ipaddress.id,
 | 
						|
                                    accountid=self.account.name,
 | 
						|
                                    networkid=self.network.id
 | 
						|
                                )
 | 
						|
 | 
						|
 | 
						|
class TestVmWithLb(cloudstackTestCase):
 | 
						|
 | 
						|
    @classmethod
 | 
						|
    def setUpClass(cls):
 | 
						|
        cls._cleanup = []
 | 
						|
        cls.testClient = super(TestVmWithLb, cls).getClsTestClient()
 | 
						|
        cls.api_client = cls.testClient.getApiClient()
 | 
						|
 | 
						|
        cls.services = Services().services
 | 
						|
        # Get Zone, Domain and templates
 | 
						|
        cls.domain = get_domain(cls.api_client)
 | 
						|
        cls.zone = get_zone(cls.api_client, cls.testClient.getZoneForTests())
 | 
						|
        cls.template = get_template(
 | 
						|
                            cls.api_client,
 | 
						|
                            cls.zone.id,
 | 
						|
                            cls.services["ostype"]
 | 
						|
                            )
 | 
						|
        try:
 | 
						|
           cls.netscaler = add_netscaler(cls.api_client, cls.zone.id, cls.services["netscaler"])
 | 
						|
           cls._cleanup.append(cls.netscaler)
 | 
						|
           cls.network_offering = NetworkOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["network_offering"],
 | 
						|
                                            conservemode=True
 | 
						|
                                            )
 | 
						|
           # Enable Network offering
 | 
						|
           cls.network_offering.update(cls.api_client, state='Enabled')
 | 
						|
           cls.services["virtual_machine"]["zoneid"] = cls.zone.id
 | 
						|
           cls.services["virtual_machine"]["template"] = cls.template.id
 | 
						|
 | 
						|
           cls.service_offering = ServiceOffering.create(
 | 
						|
                                            cls.api_client,
 | 
						|
                                            cls.services["service_offering"]
 | 
						|
                                            )
 | 
						|
           cls.account = Account.create(
 | 
						|
                                     cls.api_client,
 | 
						|
                                     cls.services["account"],
 | 
						|
                                     admin=True,
 | 
						|
                                     domainid=cls.domain.id
 | 
						|
                                     )
 | 
						|
           cls._cleanup.insert(0,cls.account)
 | 
						|
           # Creating network using the network offering created
 | 
						|
           cls.network = Network.create(
 | 
						|
                                    cls.api_client,
 | 
						|
                                    cls.services["network"],
 | 
						|
                                    accountid=cls.account.name,
 | 
						|
                                    domainid=cls.account.domainid,
 | 
						|
                                    networkofferingid=cls.network_offering.id,
 | 
						|
                                    zoneid=cls.zone.id
 | 
						|
                                    )
 | 
						|
 | 
						|
           # Spawn an instance in that network
 | 
						|
           cls.vm_1 = VirtualMachine.create(
 | 
						|
                                  cls.api_client,
 | 
						|
                                  cls.services["virtual_machine"],
 | 
						|
                                  accountid=cls.account.name,
 | 
						|
                                  domainid=cls.account.domainid,
 | 
						|
                                  serviceofferingid=cls.service_offering.id,
 | 
						|
                                  networkids=[str(cls.network.id)]
 | 
						|
                                  )
 | 
						|
           cls.vm_2 = VirtualMachine.create(
 | 
						|
                                  cls.api_client,
 | 
						|
                                  cls.services["virtual_machine"],
 | 
						|
                                  accountid=cls.account.name,
 | 
						|
                                  domainid=cls.account.domainid,
 | 
						|
                                  serviceofferingid=cls.service_offering.id,
 | 
						|
                                  networkids=[str(cls.network.id)]
 | 
						|
                                  )
 | 
						|
           cls.public_ip_1 = PublicIPAddress.create(
 | 
						|
                                cls.api_client,
 | 
						|
                                accountid=cls.account.name,
 | 
						|
                                zoneid=cls.zone.id,
 | 
						|
                                domainid=cls.account.domainid,
 | 
						|
                                networkid=cls.network.id
 | 
						|
                                )
 | 
						|
           cls.lb_rule_1 = LoadBalancerRule.create(
 | 
						|
                                    cls.api_client,
 | 
						|
                                    cls.services["lbrule"],
 | 
						|
                                    ipaddressid=cls.public_ip_1.ipaddress.id,
 | 
						|
                                    accountid=cls.account.name,
 | 
						|
                                    networkid=cls.network.id
 | 
						|
                                )
 | 
						|
           cls.public_ip_2 = PublicIPAddress.create(
 | 
						|
                                cls.api_client,
 | 
						|
                                accountid=cls.account.name,
 | 
						|
                                zoneid=cls.zone.id,
 | 
						|
                                domainid=cls.account.domainid,
 | 
						|
                                networkid=cls.network.id
 | 
						|
                                )
 | 
						|
           cls.lb_rule_2 = LoadBalancerRule.create(
 | 
						|
                                    cls.api_client,
 | 
						|
                                    cls.services["lbrule"],
 | 
						|
                                    ipaddressid=cls.public_ip_2.ipaddress.id,
 | 
						|
                                    accountid=cls.account.name,
 | 
						|
                                    networkid=cls.network.id
 | 
						|
                                )
 | 
						|
        except Exception as e:
 | 
						|
           cls.tearDownClass()
 | 
						|
           raise Exception ("Warning: Exception in setUpClass: %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @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)
 | 
						|
        return
 | 
						|
 | 
						|
    def setUp(self):
 | 
						|
        self.apiclient = self.testClient.getApiClient()
 | 
						|
        self.dbclient = self.testClient.getDbConnection()
 | 
						|
        self.cleanup = []
 | 
						|
        return
 | 
						|
 | 
						|
    def tearDown(self):
 | 
						|
        try:
 | 
						|
            self.debug("Cleaning up the resources")
 | 
						|
            #Clean up, terminate the created network offerings
 | 
						|
            cleanup_resources(self.apiclient, self.cleanup)
 | 
						|
            self.debug("Cleanup complete!")
 | 
						|
        except Exception as e:
 | 
						|
            raise Exception("Warning: Exception during cleanup : %s" % e)
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_01_delete_public_ip(self):
 | 
						|
        """Test delete one public Ip with LB rules"""
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Associate 2 public Ips and create load balancing rules in it
 | 
						|
        # 2. Delete one of the public Ip
 | 
						|
        # 3. All the LB rules should be removed from that public Ip
 | 
						|
        # 4. In netscaler, make sure that all LB rules associated with that
 | 
						|
        #    public Ip should get removed.
 | 
						|
        # 5. All servers and services service to that public Ip get deleted
 | 
						|
 | 
						|
        self.debug("Deleting public IP: %s from network: %s" % (
 | 
						|
                                        self.public_ip_2.ipaddress.ipaddress,
 | 
						|
                                        self.network.name
 | 
						|
                                        ))
 | 
						|
        self.public_ip_2.delete(self.apiclient)
 | 
						|
        self.debug(
 | 
						|
                "Public Ip: %s is deleted!" %
 | 
						|
                                        self.public_ip_2.ipaddress.ipaddress)
 | 
						|
        lb_rules = LoadBalancerRule.list(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    publicipid=self.public_ip_2.ipaddress.id,
 | 
						|
                                    listall=True,
 | 
						|
                                )
 | 
						|
        self.assertEqual(
 | 
						|
                lb_rules,
 | 
						|
                None,
 | 
						|
                "LB rules associated with the public Ip should get deleted"
 | 
						|
                )
 | 
						|
        self.debug("SSH into Netscaler to verify other resources are deleted")
 | 
						|
        try:
 | 
						|
            ssh_client = SshClient(
 | 
						|
                                    self.services["netscaler"]["ipaddress"],
 | 
						|
                                    self.services["netscaler"]["port"],
 | 
						|
                                    self.services["netscaler"]["username"],
 | 
						|
                                    self.services["netscaler"]["password"],
 | 
						|
                                    )
 | 
						|
            cmd = "show lb vserver Cloud-VirtualServer-%s-%s" % (
 | 
						|
                                        self.public_ip_2.ipaddress.ipaddress,
 | 
						|
                                        self.lb_rule_2.publicport)
 | 
						|
            self.debug("command: %s" % cmd)
 | 
						|
            res = ssh_client.execute(cmd)
 | 
						|
            result = str(res)
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
 | 
						|
            self.assertEqual(
 | 
						|
                    result.count("ERROR: No such resource"),
 | 
						|
                    1,
 | 
						|
                    "Virtual server should get deleted after removing LB rule"
 | 
						|
                    )
 | 
						|
 | 
						|
            cmd = "show ip"
 | 
						|
            self.debug("command: %s" % cmd)
 | 
						|
            res = ssh_client.execute(cmd)
 | 
						|
            result = str(res)
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
 | 
						|
            self.assertEqual(
 | 
						|
                    result.count(self.public_ip_2.ipaddress.ipaddress),
 | 
						|
                    0,
 | 
						|
                    "Virtual server should get deleted after removing LB rule"
 | 
						|
                    )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("SSH Access failed for %s: %s" % \
 | 
						|
                      (self.services["netscaler"]["ipaddress"], e))
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_02_stop_user_vm(self):
 | 
						|
        """Test stop user VM with LB"""
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Create 2 instances and add these two for load balancing
 | 
						|
        # 2. Stop one of the user VM
 | 
						|
        # 3. Test whether the request are not sent to stopped user VM
 | 
						|
        # 4. In netscaler, LB rules for this VM  still remain configured.But
 | 
						|
        #    it will be marked as being down
 | 
						|
 | 
						|
        self.debug("Adding instances: %s, %s to LB rule: %s" % (
 | 
						|
                                                        self.vm_1.name,
 | 
						|
                                                        self.vm_2.name,
 | 
						|
                                                        self.lb_rule_1.name))
 | 
						|
        self.lb_rule_1.assign(self.apiclient, [self.vm_1, self.vm_2])
 | 
						|
        self.debug("Assigned instances: %s, %s to LB rule: %s" % (
 | 
						|
                                                        self.vm_1.name,
 | 
						|
                                                        self.vm_2.name,
 | 
						|
                                                        self.lb_rule_1.name))
 | 
						|
        self.debug("Stopping VM instance: %s" % self.vm_2.name)
 | 
						|
        self.vm_2.stop(self.apiclient)
 | 
						|
        self.debug("Stopped VM: %s" % self.vm_2.name)
 | 
						|
 | 
						|
        try:
 | 
						|
            self.debug(
 | 
						|
                "Verifying request served by only running instances")
 | 
						|
            hostnames = []
 | 
						|
            ssh = self.vm_1.get_ssh_client(
 | 
						|
                                ipaddress=self.public_ip_1.ipaddress.ipaddress)
 | 
						|
            self.debug("Command: hostname")
 | 
						|
            result = ssh.execute("hostname")
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
            hostnames.append(result)
 | 
						|
 | 
						|
            ssh = self.vm_1.get_ssh_client(
 | 
						|
                                ipaddress=self.public_ip_1.ipaddress.ipaddress,
 | 
						|
                                reconnect=True
 | 
						|
                                )
 | 
						|
            self.debug("Command: hostname")
 | 
						|
            result = ssh.execute("hostname")
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
            hostnames.append(result)
 | 
						|
            self.debug("Hostnames: %s" % str(hostnames))
 | 
						|
 | 
						|
            self.assertEqual(
 | 
						|
                    hostnames[0],
 | 
						|
                    hostnames[1],
 | 
						|
                    "Hostnames must be same as another VM is stopped"
 | 
						|
                )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("Exception occured during SSH: %s - %s" % (
 | 
						|
                                        self.public_ip_1.ipaddress.ipaddress,
 | 
						|
                                        e))
 | 
						|
        self.debug("SSH into Netscaler to rules still persist")
 | 
						|
        try:
 | 
						|
            ssh_client = SshClient(
 | 
						|
                                    self.services["netscaler"]["ipaddress"],
 | 
						|
                                    self.services["netscaler"]["port"],
 | 
						|
                                    self.services["netscaler"]["username"],
 | 
						|
                                    self.services["netscaler"]["password"],
 | 
						|
                                    )
 | 
						|
 | 
						|
            cmd = "show server"
 | 
						|
            self.debug("command: %s" % cmd)
 | 
						|
            res = ssh_client.execute(cmd)
 | 
						|
            result = str(res)
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
 | 
						|
            self.assertNotEqual(
 | 
						|
                    result.count(self.vm_2.ipaddress),
 | 
						|
                    0,
 | 
						|
                    "The server should be present in netscaler after VM stop"
 | 
						|
                    )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("SSH Access failed for %s: %s" % \
 | 
						|
                      (self.services["netscaler"]["ipaddress"], e))
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_03_start_user_vm(self):
 | 
						|
        """Test start user VM with LB"""
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Create 2 instances and add these two for load balancing
 | 
						|
        # 2. Stop one of the user VM
 | 
						|
        # 3. Test whether the request are not sent to stopped user VM
 | 
						|
        # 4. In netscaler, LB rules for this VM  still remain configured.But
 | 
						|
        #    it will be marked as being down
 | 
						|
 | 
						|
        self.debug("Starting VM instance: %s" % self.vm_2.name)
 | 
						|
        self.vm_2.start(self.apiclient)
 | 
						|
        self.debug("Starting VM: %s" % self.vm_2.name)
 | 
						|
        self.debug("Sleeping for netscaler to recognize service is up")
 | 
						|
        time.sleep( 120 )
 | 
						|
 | 
						|
        try:
 | 
						|
            self.debug(
 | 
						|
                "Verifying request served by only running instances")
 | 
						|
            hostnames = []
 | 
						|
            ssh_1 = self.vm_1.get_ssh_client(
 | 
						|
                                ipaddress=self.public_ip_1.ipaddress.ipaddress)
 | 
						|
            self.debug("Command: hostname")
 | 
						|
            result = ssh_1.execute("hostname")
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
            hostnames.append(result)
 | 
						|
 | 
						|
            ssh_2 = self.vm_2.get_ssh_client(
 | 
						|
                                ipaddress=self.public_ip_1.ipaddress.ipaddress,
 | 
						|
                                reconnect=True
 | 
						|
                                )
 | 
						|
            self.debug("Command: hostname")
 | 
						|
            result = ssh_2.execute("hostname")
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
            hostnames.append(result)
 | 
						|
            self.debug("Hostnames: %s" % str(hostnames))
 | 
						|
 | 
						|
            self.assertNotEqual(
 | 
						|
                    hostnames[0],
 | 
						|
                    hostnames[1],
 | 
						|
                    "Both request should be served by different instances"
 | 
						|
                )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("Exception occured during SSH: %s - %s" % (
 | 
						|
                                        self.public_ip_1.ipaddress.ipaddress,
 | 
						|
                                        e))
 | 
						|
        self.debug("SSH into Netscaler to rules still persist")
 | 
						|
        try:
 | 
						|
            ssh_client = SshClient(
 | 
						|
                                    self.services["netscaler"]["ipaddress"],
 | 
						|
                                    self.services["netscaler"]["port"],
 | 
						|
                                    self.services["netscaler"]["username"],
 | 
						|
                                    self.services["netscaler"]["password"],
 | 
						|
                                    )
 | 
						|
 | 
						|
            cmd = "show server"
 | 
						|
            self.debug("command: %s" % cmd)
 | 
						|
            res = ssh_client.execute(cmd)
 | 
						|
            result = str(res)
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
 | 
						|
            self.assertNotEqual(
 | 
						|
                    result.count(self.vm_2.ipaddress),
 | 
						|
                    0,
 | 
						|
                    "The server should be present in netscaler after VM stop"
 | 
						|
                    )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("SSH Access failed for %s: %s" % \
 | 
						|
                      (self.services["netscaler"]["ipaddress"], e))
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns", "multihost"])
 | 
						|
    def test_04_migrate_user_vm(self):
 | 
						|
        """Test migrate user VM with LB"""
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Create 2 instances and add these two for load balancing
 | 
						|
        # 2. migrate one Vm to another host.
 | 
						|
        # 3. Test whether the request are sent to stopped user VM after migrate
 | 
						|
        # 4. In netscaler, the LB rules are still configured.
 | 
						|
 | 
						|
        hosts = Host.list(
 | 
						|
                          self.apiclient,
 | 
						|
                          zoneid=self.vm_2.zoneid,
 | 
						|
                          type='Routing'
 | 
						|
                          )
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                         isinstance(hosts, list),
 | 
						|
                         True,
 | 
						|
                         "Check the number of hosts in the zone"
 | 
						|
                         )
 | 
						|
        self.assertGreaterEqual(
 | 
						|
                len(hosts),
 | 
						|
                2,
 | 
						|
                "Atleast 2 hosts should be present in a zone for VM migration"
 | 
						|
                )
 | 
						|
 | 
						|
        # Remove the host of current VM from the hosts list
 | 
						|
        hosts[:] = [host for host in hosts if host.id != self.vm_2.hostid]
 | 
						|
 | 
						|
        host = hosts[0]
 | 
						|
        self.debug("Migrating VM-ID: %s to Host: %s" % (self.vm_2.id, host.id))
 | 
						|
 | 
						|
        cmd = migrateVirtualMachine.migrateVirtualMachineCmd()
 | 
						|
        cmd.hostid = host.id
 | 
						|
        cmd.virtualmachineid = self.vm_2.id
 | 
						|
        self.apiclient.migrateVirtualMachine(cmd)
 | 
						|
 | 
						|
        list_vm_response = list_virtual_machines(
 | 
						|
                                            self.apiclient,
 | 
						|
                                            id=self.vm_2.id
 | 
						|
                                            )
 | 
						|
        self.assertEqual(
 | 
						|
                            isinstance(list_vm_response, list),
 | 
						|
                            True,
 | 
						|
                            "Check list response returns a valid list"
 | 
						|
                        )
 | 
						|
        vm_response = list_vm_response[0]
 | 
						|
 | 
						|
        self.assertEqual(
 | 
						|
                            vm_response.hostid,
 | 
						|
                            host.id,
 | 
						|
                            "Check destination hostID of migrated VM"
 | 
						|
                        )
 | 
						|
        self.debug("Migrated VM-ID: %s to Host: %s" % (self.vm_2.id, host.id))
 | 
						|
        try:
 | 
						|
            self.debug(
 | 
						|
                "Verifying request served by only running instances")
 | 
						|
            hostnames = []
 | 
						|
            ssh_1 = self.vm_1.get_ssh_client(
 | 
						|
                                ipaddress=self.public_ip_1.ipaddress.ipaddress,
 | 
						|
                                reconnect=True
 | 
						|
                                )
 | 
						|
            self.debug("Command: hostname")
 | 
						|
            result = ssh_1.execute("hostname")
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
            hostnames.append(result)
 | 
						|
 | 
						|
            ssh_2 = self.vm_2.get_ssh_client(
 | 
						|
                                ipaddress=self.public_ip_1.ipaddress.ipaddress,
 | 
						|
                                reconnect=True
 | 
						|
                                )
 | 
						|
            self.debug("Command: hostname")
 | 
						|
            result = ssh_2.execute("hostname")
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
            hostnames.append(result)
 | 
						|
            self.debug("Hostnames: %s" % str(hostnames))
 | 
						|
 | 
						|
            self.assertNotEqual(
 | 
						|
                    hostnames[0],
 | 
						|
                    hostnames[1],
 | 
						|
                    "Both request should be served by different instances"
 | 
						|
                )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("Exception occured during SSH: %s - %s" % (
 | 
						|
                                        self.public_ip_1.ipaddress.ipaddress,
 | 
						|
                                        e))
 | 
						|
        self.debug("SSH into Netscaler to rules still persist")
 | 
						|
        try:
 | 
						|
            ssh_client = SshClient(
 | 
						|
                                    self.services["netscaler"]["ipaddress"],
 | 
						|
                                    self.services["netscaler"]["port"],
 | 
						|
                                    self.services["netscaler"]["username"],
 | 
						|
                                    self.services["netscaler"]["password"],
 | 
						|
                                    )
 | 
						|
 | 
						|
            cmd = "show server"
 | 
						|
            self.debug("command: %s" % cmd)
 | 
						|
            res = ssh_client.execute(cmd)
 | 
						|
            result = str(res)
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
 | 
						|
            self.assertNotEqual(
 | 
						|
                    result.count(self.vm_2.ipaddress),
 | 
						|
                    0,
 | 
						|
                    "The server should be present in netscaler after migrate"
 | 
						|
                    )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("SSH Access failed for %s: %s" % \
 | 
						|
                      (self.services["netscaler"]["ipaddress"], e))
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_05_reboot_user_vm(self):
 | 
						|
        """Test reboot user VM with LB"""
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Create 2 instances and add these two for load balancing
 | 
						|
        # 2. Reboot one of the user VM
 | 
						|
        # 3. Test whether the request are sent to both VMs after reboot
 | 
						|
        # 4. In netscaler, LB rules for this VM  still remain configured.
 | 
						|
 | 
						|
        self.debug("Rebooting VM instance: %s" % self.vm_2.name)
 | 
						|
        self.vm_2.reboot(self.apiclient)
 | 
						|
        self.debug("Rebooting VM: %s" % self.vm_2.name)
 | 
						|
        self.debug("Waiting for Netscaler to get the service up")
 | 
						|
        time.sleep( 120 )
 | 
						|
 | 
						|
        try:
 | 
						|
            self.debug(
 | 
						|
                "Verifying request served by only running instances")
 | 
						|
            hostnames = []
 | 
						|
            ssh_1 = self.vm_1.get_ssh_client(
 | 
						|
                                ipaddress=self.public_ip_1.ipaddress.ipaddress)
 | 
						|
            self.debug("Command: hostname")
 | 
						|
            result = ssh_1.execute("hostname")
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
            hostnames.append(result)
 | 
						|
 | 
						|
            ssh_2 = self.vm_2.get_ssh_client(
 | 
						|
                                ipaddress=self.public_ip_1.ipaddress.ipaddress,
 | 
						|
                                reconnect=True
 | 
						|
                                )
 | 
						|
            self.debug("Command: hostname")
 | 
						|
            result = ssh_2.execute("hostname")
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
            hostnames.append(result)
 | 
						|
            self.debug("Hostnames: %s" % str(hostnames))
 | 
						|
 | 
						|
            if hostnames[0] == hostnames[1]:
 | 
						|
                ssh_3 = self.vm_2.get_ssh_client(
 | 
						|
                                ipaddress=self.public_ip_1.ipaddress.ipaddress,
 | 
						|
                                reconnect=True
 | 
						|
                                )
 | 
						|
                self.debug("Command: hostname")
 | 
						|
                result = ssh_3.execute("hostname")
 | 
						|
                self.debug("Output: %s" % result)
 | 
						|
                hostnames.append(result)
 | 
						|
 | 
						|
                self.assertNotEqual(
 | 
						|
                    hostnames[0],
 | 
						|
                    hostnames[2],
 | 
						|
                    "Both request should be served by different instances"
 | 
						|
                )
 | 
						|
            else:
 | 
						|
                 self.assertNotEqual(
 | 
						|
                    hostnames[0],
 | 
						|
                    hostnames[1],
 | 
						|
                    "Both request should be served by different instances"
 | 
						|
                 )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("Exception occured during SSH: %s - %s" % (
 | 
						|
                                        self.public_ip_1.ipaddress.ipaddress,
 | 
						|
                                        e))
 | 
						|
        self.debug("SSH into Netscaler to rules still persist")
 | 
						|
        try:
 | 
						|
            ssh_client = SshClient(
 | 
						|
                                    self.services["netscaler"]["ipaddress"],
 | 
						|
                                    self.services["netscaler"]["port"],
 | 
						|
                                    self.services["netscaler"]["username"],
 | 
						|
                                    self.services["netscaler"]["password"],
 | 
						|
                                    )
 | 
						|
 | 
						|
            cmd = "show server"
 | 
						|
            self.debug("command: %s" % cmd)
 | 
						|
            res = ssh_client.execute(cmd)
 | 
						|
            result = str(res)
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
 | 
						|
            self.assertNotEqual(
 | 
						|
                    result.count(self.vm_2.ipaddress),
 | 
						|
                    0,
 | 
						|
                    "The server should be present in netscaler after reboot"
 | 
						|
                    )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("SSH Access failed for %s: %s" % \
 | 
						|
                      (self.services["netscaler"]["ipaddress"], e))
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_06_destroy_user_vm(self):
 | 
						|
        """Test destroy user VM with LB"""
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Create 2 instances and add these two for load balancing
 | 
						|
        # 2. Destroy one of the user VM
 | 
						|
        # 3. Until the time the Vm is in "Destroyed" state, the servies
 | 
						|
        #    relating to this Vm will be marked as "Down".
 | 
						|
        # 4. Once the Vm gets expunged, then the servers and services
 | 
						|
        #    associated with this VM should get deleted and the LB rules
 | 
						|
        #    should not be pointing to this Vm anymore.
 | 
						|
 | 
						|
        self.debug("Destroying VM instance: %s" % self.vm_2.name)
 | 
						|
        self.vm_2.delete(self.apiclient)
 | 
						|
        self.debug("Destroying VM: %s" % self.vm_2.name)
 | 
						|
 | 
						|
        try:
 | 
						|
            self.debug(
 | 
						|
                "Verifying request served by only running instances")
 | 
						|
            hostnames = []
 | 
						|
            ssh = self.vm_1.get_ssh_client(
 | 
						|
                                ipaddress=self.public_ip_1.ipaddress.ipaddress)
 | 
						|
            self.debug("Command: hostname")
 | 
						|
            result = ssh.execute("hostname")
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
            hostnames.append(result)
 | 
						|
 | 
						|
            ssh = self.vm_1.get_ssh_client(
 | 
						|
                                ipaddress=self.public_ip_1.ipaddress.ipaddress,
 | 
						|
                                reconnect=True
 | 
						|
                                )
 | 
						|
            self.debug("Command: hostname")
 | 
						|
            result = ssh.execute("hostname")
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
            hostnames.append(result)
 | 
						|
            self.debug("Hostnames: %s" % str(hostnames))
 | 
						|
 | 
						|
            self.assertEqual(
 | 
						|
                    hostnames[0],
 | 
						|
                    hostnames[1],
 | 
						|
                    "Both request should be served by same instance"
 | 
						|
                )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("Exception occured during SSH: %s - %s" % (
 | 
						|
                                        self.public_ip_1.ipaddress.ipaddress,
 | 
						|
                                        e))
 | 
						|
        delay = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='expunge.delay'
 | 
						|
                                    )
 | 
						|
        wait = list_configurations(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    name='expunge.interval'
 | 
						|
                                    )
 | 
						|
        # Sleep to ensure that all resources are deleted
 | 
						|
        time.sleep(int(delay[0].value) + int(wait[0].value))
 | 
						|
        self.debug("SSH into Netscaler to rules still persist")
 | 
						|
        try:
 | 
						|
            ssh_client = SshClient(
 | 
						|
                                    self.services["netscaler"]["ipaddress"],
 | 
						|
                                    self.services["netscaler"]["port"],
 | 
						|
                                    self.services["netscaler"]["username"],
 | 
						|
                                    self.services["netscaler"]["password"],
 | 
						|
                                    )
 | 
						|
 | 
						|
            cmd = "show server"
 | 
						|
            self.debug("command: %s" % cmd)
 | 
						|
            res = ssh_client.execute(cmd)
 | 
						|
            result = str(res)
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
 | 
						|
            self.assertEqual(
 | 
						|
                    result.count(self.vm_2.ipaddress),
 | 
						|
                    0,
 | 
						|
                    "The server should not be present in netscaler after destroy"
 | 
						|
                    )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("SSH Access failed for %s: %s" % \
 | 
						|
                      (self.services["netscaler"]["ipaddress"], e))
 | 
						|
        return
 | 
						|
 | 
						|
    @attr(tags = ["advancedns"])
 | 
						|
    def test_07_delete_all_public_ip(self):
 | 
						|
        """Test delete all public Ip with LB rules"""
 | 
						|
 | 
						|
 | 
						|
        # Validate the following
 | 
						|
        # 1. Associate 2 public Ips and create load balancing rules in it
 | 
						|
        # 2. Delete all of the public Ip
 | 
						|
        # 3. All the LB rules should be removed from that public Ip
 | 
						|
        # 4. In netscaler, make sure that all LB rules associated with that
 | 
						|
        #    public Ip should get removed.
 | 
						|
        # 5. All servers and services service to that public Ip get deleted
 | 
						|
 | 
						|
        self.debug("Deleting public IP: %s from network: %s" % (
 | 
						|
                                        self.public_ip_1.ipaddress.ipaddress,
 | 
						|
                                        self.network.name
 | 
						|
                                        ))
 | 
						|
        self.public_ip_1.delete(self.apiclient)
 | 
						|
        self.debug(
 | 
						|
                "Public Ip: %s is deleted!" %
 | 
						|
                                        self.public_ip_1.ipaddress.ipaddress)
 | 
						|
        lb_rules = LoadBalancerRule.list(
 | 
						|
                                    self.apiclient,
 | 
						|
                                    publicipid=self.public_ip_1.ipaddress.id,
 | 
						|
                                    listall=True,
 | 
						|
                                )
 | 
						|
        self.assertEqual(
 | 
						|
                lb_rules,
 | 
						|
                None,
 | 
						|
                "LB rules associated with the public Ip should get deleted"
 | 
						|
                )
 | 
						|
        self.debug("SSH into Netscaler to verify other resources are deleted")
 | 
						|
        try:
 | 
						|
            ssh_client = SshClient(
 | 
						|
                                    self.services["netscaler"]["ipaddress"],
 | 
						|
                                    self.services["netscaler"]["port"],
 | 
						|
                                    self.services["netscaler"]["username"],
 | 
						|
                                    self.services["netscaler"]["password"],
 | 
						|
                                    )
 | 
						|
            cmd = "show lb vserver Cloud-VirtualServer-%s-%s" % (
 | 
						|
                                        self.public_ip_1.ipaddress.ipaddress,
 | 
						|
                                        self.lb_rule_1.publicport)
 | 
						|
            self.debug("command: %s" % cmd)
 | 
						|
            res = ssh_client.execute(cmd)
 | 
						|
            result = str(res)
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
 | 
						|
            self.assertEqual(
 | 
						|
                    result.count("ERROR: No such resource"),
 | 
						|
                    1,
 | 
						|
                    "Virtual server should get deleted after removing LB rule"
 | 
						|
                    )
 | 
						|
 | 
						|
            cmd = "show ip"
 | 
						|
            self.debug("command: %s" % cmd)
 | 
						|
            res = ssh_client.execute(cmd)
 | 
						|
            result = str(res)
 | 
						|
            self.debug("Output: %s" % result)
 | 
						|
 | 
						|
            self.assertEqual(
 | 
						|
                    result.count(self.public_ip_1.ipaddress.ipaddress),
 | 
						|
                    0,
 | 
						|
                    "Virtual server should get deleted after removing LB rule"
 | 
						|
                    )
 | 
						|
        except Exception as e:
 | 
						|
            self.fail("SSH Access failed for %s: %s" % \
 | 
						|
                      (self.services["netscaler"]["ipaddress"], e))
 | 
						|
        return
 |