sdnvpn/test/functest/testcase_4.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 argparse
  12 import logging
  13 from random import randint
  14
  15 import functest.utils.openstack_utils as os_utils
  16
  17 from sdnvpn.lib import utils as test_utils
  18 from sdnvpn.lib import config as sdnvpn_config
  19 from sdnvpn.lib.results import Results
  20
  21 parser = argparse.ArgumentParser()
  22
  23 parser.add_argument("-r", "--report",
  24                     help="Create json result file",
  25                     action="store_true")
  26
  27 args = parser.parse_args()
  28
  29 logger = logging.getLogger('sdnvpn-testcase-4')
  30
  31 COMMON_CONFIG = sdnvpn_config.CommonConfig()
  32 TESTCASE_CONFIG = sdnvpn_config.TestcaseConfig('testcase_4')
  33
  34
  35 def main():
  36     results = Results(COMMON_CONFIG.line_length)
  37
  38     results.add_to_summary(0, "=")
  39     results.add_to_summary(2, "STATUS", "SUBTEST")
  40     results.add_to_summary(0, "=")
  41
  42     nova_client = os_utils.get_nova_client()
  43     neutron_client = os_utils.get_neutron_client()
  44     glance_client = os_utils.get_glance_client()
  45
  46     (floatingip_ids, instance_ids, router_ids, network_ids, image_ids,
  47      subnet_ids, interfaces, bgpvpn_ids) = ([] for i in range(8))
  48
  49     image_id = os_utils.create_glance_image(glance_client,
  50                                             TESTCASE_CONFIG.image_name,
  51                                             COMMON_CONFIG.image_path,
  52                                             disk=COMMON_CONFIG.image_format,
  53                                             container="bare",
  54                                             public='public')
  55     image_ids.append(image_id)
  56
  57     network_1_id, subnet_1_id, router_1_id = test_utils.create_network(
  58         neutron_client,
  59         TESTCASE_CONFIG.net_1_name,
  60         TESTCASE_CONFIG.subnet_1_name,
  61         TESTCASE_CONFIG.subnet_1_cidr,
  62         TESTCASE_CONFIG.router_1_name)
  63
  64     network_2_id = test_utils.create_net(
  65         neutron_client,
  66         TESTCASE_CONFIG.net_2_name)
  67
  68     subnet_2_id = test_utils.create_subnet(
  69         neutron_client,
  70         TESTCASE_CONFIG.subnet_2_name,
  71         TESTCASE_CONFIG.subnet_2_cidr,
  72         network_2_id)
  73     interfaces.append(tuple((router_1_id, subnet_1_id)))
  74     network_ids.extend([network_1_id, network_2_id])
  75     router_ids.append(router_1_id)
  76     subnet_ids.extend([subnet_1_id, subnet_2_id])
  77
  78     sg_id = os_utils.create_security_group_full(
  79         neutron_client,
  80         TESTCASE_CONFIG.secgroup_name,
  81         TESTCASE_CONFIG.secgroup_descr)
  82
  83     compute_nodes = test_utils.assert_and_get_compute_nodes(nova_client)
  84
  85     av_zone_1 = "nova:" + compute_nodes[0]
  86     av_zone_2 = "nova:" + compute_nodes[1]
  87
  88     # boot INTANCES
  89     vm_2 = test_utils.create_instance(
  90         nova_client,
  91         TESTCASE_CONFIG.instance_2_name,
  92         image_id,
  93         network_1_id,
  94         sg_id,
  95         secgroup_name=TESTCASE_CONFIG.secgroup_name,
  96         compute_node=av_zone_1)
  97     vm_2_ip = test_utils.get_instance_ip(vm_2)
  98
  99     vm_3 = test_utils.create_instance(
 100         nova_client,
 101         TESTCASE_CONFIG.instance_3_name,
 102         image_id,
 103         network_1_id,
 104         sg_id,
 105         secgroup_name=TESTCASE_CONFIG.secgroup_name,
 106         compute_node=av_zone_2)
 107     vm_3_ip = test_utils.get_instance_ip(vm_3)
 108
 109     vm_5 = test_utils.create_instance(
 110         nova_client,
 111         TESTCASE_CONFIG.instance_5_name,
 112         image_id,
 113         network_2_id,
 114         sg_id,
 115         secgroup_name=TESTCASE_CONFIG.secgroup_name,
 116         compute_node=av_zone_2)
 117     vm_5_ip = test_utils.get_instance_ip(vm_5)
 118
 119     # We boot vm5 first because we need vm5_ip for vm4 userdata
 120     u4 = test_utils.generate_ping_userdata([vm_5_ip])
 121     vm_4 = test_utils.create_instance(
 122         nova_client,
 123         TESTCASE_CONFIG.instance_4_name,
 124         image_id,
 125         network_2_id,
 126         sg_id,
 127         secgroup_name=TESTCASE_CONFIG.secgroup_name,
 128         compute_node=av_zone_1,
 129         userdata=u4)
 130     vm_4_ip = test_utils.get_instance_ip(vm_4)
 131
 132     # We boot VM1 at the end because we need to get the IPs first to generate
 133     # the userdata
 134     u1 = test_utils.generate_ping_userdata([vm_2_ip,
 135                                             vm_3_ip,
 136                                             vm_4_ip,
 137                                             vm_5_ip])
 138     vm_1 = test_utils.create_instance(
 139         nova_client,
 140         TESTCASE_CONFIG.instance_1_name,
 141         image_id,
 142         network_1_id,
 143         sg_id,
 144         secgroup_name=TESTCASE_CONFIG.secgroup_name,
 145         compute_node=av_zone_1,
 146         userdata=u1)
 147
 148     instance_ids.extend([vm_1.id, vm_2.id, vm_3.id, vm_4.id, vm_5.id])
 149
 150     msg = ("Create VPN with eRT<>iRT")
 151     results.record_action(msg)
 152     vpn_name = "sdnvpn-" + str(randint(100000, 999999))
 153     kwargs = {"import_targets": TESTCASE_CONFIG.targets1,
 154               "export_targets": TESTCASE_CONFIG.targets2,
 155               "route_distinguishers": TESTCASE_CONFIG.route_distinguishers,
 156               "name": vpn_name}
 157     bgpvpn = os_utils.create_bgpvpn(neutron_client, **kwargs)
 158     bgpvpn_id = bgpvpn['bgpvpn']['id']
 159     logger.debug("VPN created details: %s" % bgpvpn)
 160     bgpvpn_ids.append(bgpvpn_id)
 161
 162     msg = ("Associate router '%s' to the VPN." % TESTCASE_CONFIG.router_1_name)
 163     results.record_action(msg)
 164     results.add_to_summary(0, "-")
 165
 166     os_utils.create_router_association(
 167         neutron_client, bgpvpn_id, router_1_id)
 168
 169     # Wait for VMs to get ips.
 170     instances_up = test_utils.wait_for_instances_up(vm_1, vm_2,
 171                                                     vm_3, vm_4,
 172                                                     vm_5)
 173
 174     if not instances_up:
 175         logger.error("One or more instances is down")
 176         # TODO Handle appropriately
 177
 178     results.get_ping_status(vm_1, vm_2, expected="PASS", timeout=200)
 179     results.get_ping_status(vm_1, vm_3, expected="PASS", timeout=30)
 180     results.get_ping_status(vm_1, vm_4, expected="FAIL", timeout=30)
 181
 182     msg = ("Associate network '%s' to the VPN." % TESTCASE_CONFIG.net_2_name)
 183     results.add_to_summary(0, "-")
 184     results.record_action(msg)
 185     results.add_to_summary(0, "-")
 186     os_utils.create_network_association(
 187         neutron_client, bgpvpn_id, network_2_id)
 188
 189     test_utils.wait_for_bgp_router_assoc(
 190         neutron_client, bgpvpn_id, router_1_id)
 191     test_utils.wait_for_bgp_net_assoc(
 192         neutron_client, bgpvpn_id, network_2_id)
 193
 194     logger.info("Waiting for the VMs to connect to each other using the"
 195                 " updated network configuration")
 196     test_utils.wait_before_subtest()
 197
 198     results.get_ping_status(vm_4, vm_5, expected="PASS", timeout=30)
 199     # TODO enable again when isolation in VPN with iRT != eRT works
 200     # results.get_ping_status(vm_1, vm_4, expected="FAIL", timeout=30)
 201     # results.get_ping_status(vm_1, vm_5, expected="FAIL", timeout=30)
 202
 203     msg = ("Update VPN with eRT=iRT ...")
 204     results.add_to_summary(0, "-")
 205     results.record_action(msg)
 206     results.add_to_summary(0, "-")
 207     kwargs = {"import_targets": TESTCASE_CONFIG.targets1,
 208               "export_targets": TESTCASE_CONFIG.targets1,
 209               "name": vpn_name}
 210     bgpvpn = os_utils.update_bgpvpn(neutron_client, bgpvpn_id, **kwargs)
 211
 212     logger.info("Waiting for the VMs to connect to each other using the"
 213                 " updated network configuration")
 214     test_utils.wait_before_subtest()
 215
 216     results.get_ping_status(vm_1, vm_4, expected="PASS", timeout=30)
 217     results.get_ping_status(vm_1, vm_5, expected="PASS", timeout=30)
 218
 219     results.add_to_summary(0, "=")
 220     logger.info("\n%s" % results.summary)
 221
 222     test_utils.cleanup_nova(nova_client, instance_ids, image_ids)
 223     test_utils.cleanup_neutron(neutron_client, floatingip_ids, bgpvpn_ids,
 224                                interfaces, subnet_ids, router_ids,
 225                                network_ids)
 226     return results.compile_summary()
 227
 228
 229 if __name__ == '__main__':
 230     logging.basicConfig(level=logging.INFO)
 231     main()