sdnvpn/test/functest/testcase_2.py

   1 #!/usr/bin/python
   2 #
   3 # Copyright (c) 2017 All rights reserved
   4 # This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 #
   8 # http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10
  11 import logging
  12 import sys
  13
  14 from functest.utils import openstack_utils as os_utils
  15 from random import randint
  16 from sdnvpn.lib import config as sdnvpn_config
  17 from sdnvpn.lib import utils as test_utils
  18 from sdnvpn.lib.results import Results
  19
  20 logger = logging.getLogger(__name__)
  21
  22 COMMON_CONFIG = sdnvpn_config.CommonConfig()
  23 TESTCASE_CONFIG = sdnvpn_config.TestcaseConfig(
  24     'sdnvpn.test.functest.testcase_2')
  25
  26
  27 def main():
  28     results = Results(COMMON_CONFIG.line_length)
  29
  30     results.add_to_summary(0, "=")
  31     results.add_to_summary(2, "STATUS", "SUBTEST")
  32     results.add_to_summary(0, "=")
  33
  34     nova_client = os_utils.get_nova_client()
  35     neutron_client = os_utils.get_neutron_client()
  36     glance_client = os_utils.get_glance_client()
  37
  38     (floatingip_ids, instance_ids, router_ids, network_ids, image_ids,
  39      subnet_ids, interfaces, bgpvpn_ids) = ([] for i in range(8))
  40
  41     logger.debug("Using private key %s injected to the VMs."
  42                  % COMMON_CONFIG.keyfile_path)
  43     keyfile = open(COMMON_CONFIG.keyfile_path, 'r')
  44     key = keyfile.read()
  45     keyfile.close()
  46     files = {"/home/cirros/id_rsa": key}
  47
  48     image_id = os_utils.create_glance_image(glance_client,
  49                                             TESTCASE_CONFIG.image_name,
  50                                             COMMON_CONFIG.image_path,
  51                                             disk=COMMON_CONFIG.image_format,
  52                                             container="bare",
  53                                             public='public')
  54     image_ids.append(image_id)
  55
  56     network_1_id = test_utils.create_net(
  57         neutron_client,
  58         TESTCASE_CONFIG.net_1_name)
  59     subnet_1a_id = test_utils.create_subnet(
  60         neutron_client,
  61         TESTCASE_CONFIG.subnet_1a_name,
  62         TESTCASE_CONFIG.subnet_1a_cidr,
  63         network_1_id)
  64     subnet_1b_id = test_utils.create_subnet(
  65         neutron_client,
  66         TESTCASE_CONFIG.subnet_1b_name,
  67         TESTCASE_CONFIG.subnet_1b_cidr,
  68         network_1_id)
  69
  70     network_2_id = test_utils.create_net(
  71         neutron_client,
  72         TESTCASE_CONFIG.net_2_name)
  73     subnet_2a_id = test_utils.create_subnet(
  74         neutron_client,
  75         TESTCASE_CONFIG.subnet_2a_name,
  76         TESTCASE_CONFIG.subnet_2a_cidr,
  77         network_2_id)
  78     subnet_2b_id = test_utils.create_subnet(
  79         neutron_client,
  80         TESTCASE_CONFIG.subnet_2b_name,
  81         TESTCASE_CONFIG.subnet_2b_cidr,
  82         network_2_id)
  83     network_ids.extend([network_1_id, network_2_id])
  84     subnet_ids.extend([subnet_1a_id, subnet_1b_id, subnet_2a_id, subnet_2b_id])
  85
  86     sg_id = os_utils.create_security_group_full(neutron_client,
  87                                                 TESTCASE_CONFIG.secgroup_name,
  88                                                 TESTCASE_CONFIG.secgroup_descr)
  89
  90     compute_nodes = test_utils.assert_and_get_compute_nodes(nova_client)
  91
  92     av_zone_1 = "nova:" + compute_nodes[0]
  93     av_zone_2 = "nova:" + compute_nodes[1]
  94
  95     # boot INTANCES
  96     userdata_common = test_utils.generate_userdata_common()
  97     vm_2 = test_utils.create_instance(
  98         nova_client,
  99         TESTCASE_CONFIG.instance_2_name,
 100         image_id,
 101         network_1_id,
 102         sg_id,
 103         fixed_ip=TESTCASE_CONFIG.instance_2_ip,
 104         secgroup_name=TESTCASE_CONFIG.secgroup_name,
 105         compute_node=av_zone_1,
 106         userdata=userdata_common)
 107
 108     vm_3 = test_utils.create_instance(
 109         nova_client,
 110         TESTCASE_CONFIG.instance_3_name,
 111         image_id,
 112         network_1_id,
 113         sg_id,
 114         fixed_ip=TESTCASE_CONFIG.instance_3_ip,
 115         secgroup_name=TESTCASE_CONFIG.secgroup_name,
 116         compute_node=av_zone_2,
 117         userdata=userdata_common)
 118
 119     vm_5 = test_utils.create_instance(
 120         nova_client,
 121         TESTCASE_CONFIG.instance_5_name,
 122         image_id,
 123         network_2_id,
 124         sg_id,
 125         fixed_ip=TESTCASE_CONFIG.instance_5_ip,
 126         secgroup_name=TESTCASE_CONFIG.secgroup_name,
 127         compute_node=av_zone_2,
 128         userdata=userdata_common)
 129
 130     # We boot vm5 first because we need vm5_ip for vm4 userdata
 131     u4 = test_utils.generate_userdata_with_ssh(
 132         [TESTCASE_CONFIG.instance_1_ip,
 133          TESTCASE_CONFIG.instance_3_ip,
 134          TESTCASE_CONFIG.instance_5_ip])
 135     vm_4 = test_utils.create_instance(
 136         nova_client,
 137         TESTCASE_CONFIG.instance_4_name,
 138         image_id,
 139         network_2_id,
 140         sg_id,
 141         fixed_ip=TESTCASE_CONFIG.instance_4_ip,
 142         secgroup_name=TESTCASE_CONFIG.secgroup_name,
 143         compute_node=av_zone_1,
 144         userdata=u4,
 145         files=files)
 146
 147     # We boot VM1 at the end because we need to get the IPs first to generate
 148     # the userdata
 149     u1 = test_utils.generate_userdata_with_ssh(
 150         [TESTCASE_CONFIG.instance_2_ip,
 151          TESTCASE_CONFIG.instance_3_ip,
 152          TESTCASE_CONFIG.instance_4_ip,
 153          TESTCASE_CONFIG.instance_5_ip])
 154     vm_1 = test_utils.create_instance(
 155         nova_client,
 156         TESTCASE_CONFIG.instance_1_name,
 157         image_id,
 158         network_1_id,
 159         sg_id,
 160         fixed_ip=TESTCASE_CONFIG.instance_1_ip,
 161         secgroup_name=TESTCASE_CONFIG.secgroup_name,
 162         compute_node=av_zone_1,
 163         userdata=u1,
 164         files=files)
 165     instance_ids.extend([vm_1.id, vm_2.id, vm_3.id, vm_4.id, vm_5.id])
 166
 167     msg = ("Create VPN1 with eRT=iRT")
 168     results.record_action(msg)
 169     vpn1_name = "sdnvpn-1-" + str(randint(100000, 999999))
 170     kwargs = {"import_targets": TESTCASE_CONFIG.targets2,
 171               "export_targets": TESTCASE_CONFIG.targets2,
 172               "route_targets": TESTCASE_CONFIG.targets2,
 173               "route_distinguishers": TESTCASE_CONFIG.route_distinguishers1,
 174               "name": vpn1_name}
 175     bgpvpn1 = test_utils.create_bgpvpn(neutron_client, **kwargs)
 176     bgpvpn1_id = bgpvpn1['bgpvpn']['id']
 177     logger.debug("VPN1 created details: %s" % bgpvpn1)
 178     bgpvpn_ids.append(bgpvpn1_id)
 179
 180     msg = ("Associate network '%s' to the VPN." % TESTCASE_CONFIG.net_1_name)
 181     results.record_action(msg)
 182     results.add_to_summary(0, "-")
 183
 184     test_utils.create_network_association(
 185         neutron_client, bgpvpn1_id, network_1_id)
 186
 187     # Wait for VMs to get ips.
 188     instances_up = test_utils.wait_for_instances_up(vm_1, vm_2,
 189                                                     vm_3, vm_4,
 190                                                     vm_5)
 191
 192     if not instances_up:
 193         logger.error("One or more instances is down")
 194         sys.exit(-1)
 195
 196     logger.info("Waiting for the VMs to connect to each other using the"
 197                 " updated network configuration")
 198     test_utils.wait_before_subtest()
 199
 200     # 10.10.10.12 should return sdnvpn-2 to sdnvpn-1
 201     results.check_ssh_output(vm_1, vm_2,
 202                              expected=TESTCASE_CONFIG.instance_2_name,
 203                              timeout=200)
 204     # 10.10.11.13 should return sdnvpn-3 to sdnvpn-1
 205     results.check_ssh_output(vm_1, vm_3,
 206                              expected=TESTCASE_CONFIG.instance_3_name,
 207                              timeout=30)
 208
 209     results.add_to_summary(0, "-")
 210     msg = ("Create VPN2 with eRT=iRT")
 211     results.record_action(msg)
 212     vpn2_name = "sdnvpn-2-" + str(randint(100000, 999999))
 213     kwargs = {"import_targets": TESTCASE_CONFIG.targets1,
 214               "export_targets": TESTCASE_CONFIG.targets1,
 215               "route_targets": TESTCASE_CONFIG.targets1,
 216               "route_distinguishers": TESTCASE_CONFIG.route_distinguishers2,
 217               "name": vpn2_name}
 218     bgpvpn2 = test_utils.create_bgpvpn(neutron_client, **kwargs)
 219     bgpvpn2_id = bgpvpn2['bgpvpn']['id']
 220     logger.debug("VPN created details: %s" % bgpvpn2)
 221     bgpvpn_ids.append(bgpvpn2_id)
 222
 223     msg = ("Associate network '%s' to the VPN2." % TESTCASE_CONFIG.net_2_name)
 224     results.record_action(msg)
 225     results.add_to_summary(0, "-")
 226
 227     test_utils.create_network_association(
 228         neutron_client, bgpvpn2_id, network_2_id)
 229
 230     test_utils.wait_for_bgp_net_assoc(neutron_client, bgpvpn1_id, network_1_id)
 231     test_utils.wait_for_bgp_net_assoc(neutron_client, bgpvpn2_id, network_2_id)
 232
 233     logger.info("Waiting for the VMs to connect to each other using the"
 234                 " updated network configuration")
 235     test_utils.wait_before_subtest()
 236
 237     # 10.10.11.13 should return sdnvpn-5 to sdnvpn-4
 238     results.check_ssh_output(vm_4, vm_5,
 239                              expected=TESTCASE_CONFIG.instance_5_name,
 240                              timeout=30)
 241
 242     # 10.10.10.11 should return "not reachable" to sdnvpn-4
 243     results.check_ssh_output(vm_4, vm_1,
 244                              expected="not reachable",
 245                              timeout=30)
 246
 247     test_utils.cleanup_nova(nova_client, instance_ids, image_ids)
 248     test_utils.cleanup_neutron(neutron_client, floatingip_ids, bgpvpn_ids,
 249                                interfaces, subnet_ids, router_ids,
 250                                network_ids)
 251     return results.compile_summary()
 252
 253
 254 if __name__ == '__main__':
 255     logging.basicConfig(level=logging.INFO)
 256     sys.exit(main())