sdnvpn/lib/utils.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 import logging
  11 import os
  12 import sys
  13 import time
  14 import requests
  15 import re
  16 import subprocess
  17
  18 import functest.utils.openstack_utils as os_utils
  19 from opnfv.deployment.factory import Factory as DeploymentFactory
  20
  21 from sdnvpn.lib import config as sdnvpn_config
  22
  23 logger = logging.getLogger('sdnvpn_test_utils')
  24
  25 common_config = sdnvpn_config.CommonConfig()
  26
  27 ODL_USER = 'admin'
  28 ODL_PASS = 'admin'
  29
  30
  31 def create_custom_flavor():
  32     return os_utils.get_or_create_flavor(common_config.custom_flavor_name,
  33                                          common_config.custom_flavor_ram,
  34                                          common_config.custom_flavor_disk,
  35                                          common_config.custom_flavor_vcpus)
  36
  37
  38 def create_net(neutron_client, name):
  39     logger.debug("Creating network %s", name)
  40     net_id = os_utils.create_neutron_net(neutron_client, name)
  41     if not net_id:
  42         logger.error(
  43             "There has been a problem when creating the neutron network")
  44         sys.exit(-1)
  45     return net_id
  46
  47
  48 def create_subnet(neutron_client, name, cidr, net_id):
  49     logger.debug("Creating subnet %s in network %s with cidr %s",
  50                  name, net_id, cidr)
  51     subnet_id = os_utils.create_neutron_subnet(neutron_client,
  52                                                name,
  53                                                cidr,
  54                                                net_id)
  55     if not subnet_id:
  56         logger.error(
  57             "There has been a problem when creating the neutron subnet")
  58         sys.exit(-1)
  59     return subnet_id
  60
  61
  62 def create_network(neutron_client, net, subnet1, cidr1,
  63                    router, subnet2=None, cidr2=None):
  64     """Network assoc won't work for networks/subnets created by this function.
  65     It is an ODL limitation due to it handling routers as vpns.
  66     See https://bugs.opendaylight.org/show_bug.cgi?id=6962"""
  67     network_dic = os_utils.create_network_full(neutron_client,
  68                                                net,
  69                                                subnet1,
  70                                                router,
  71                                                cidr1)
  72     if not network_dic:
  73         logger.error(
  74             "There has been a problem when creating the neutron network")
  75         sys.exit(-1)
  76     net_id = network_dic["net_id"]
  77     subnet_id = network_dic["subnet_id"]
  78     router_id = network_dic["router_id"]
  79
  80     if subnet2 is not None:
  81         logger.debug("Creating and attaching a second subnet...")
  82         subnet_id = os_utils.create_neutron_subnet(
  83             neutron_client, subnet2, cidr2, net_id)
  84         if not subnet_id:
  85             logger.error(
  86                 "There has been a problem when creating the second subnet")
  87             sys.exit(-1)
  88         logger.debug("Subnet '%s' created successfully" % subnet_id)
  89     return net_id, subnet_id, router_id
  90
  91
  92 def create_instance(nova_client,
  93                     name,
  94                     image_id,
  95                     network_id,
  96                     sg_id,
  97                     secgroup_name=None,
  98                     fixed_ip=None,
  99                     compute_node='',
 100                     userdata=None,
 101                     files=None,
 102                     **kwargs
 103                     ):
 104     if 'flavor' not in kwargs:
 105         kwargs['flavor'] = common_config.default_flavor
 106
 107     logger.info("Creating instance '%s'..." % name)
 108     logger.debug(
 109         "Configuration:\n name=%s \n flavor=%s \n image=%s \n"
 110         " network=%s\n secgroup=%s \n hypervisor=%s \n"
 111         " fixed_ip=%s\n files=%s\n userdata=\n%s\n"
 112         % (name, kwargs['flavor'], image_id, network_id, sg_id,
 113            compute_node, fixed_ip, files, userdata))
 114     instance = os_utils.create_instance_and_wait_for_active(
 115         kwargs['flavor'],
 116         image_id,
 117         network_id,
 118         name,
 119         config_drive=True,
 120         userdata=userdata,
 121         av_zone=compute_node,
 122         fixed_ip=fixed_ip,
 123         files=files)
 124
 125     if instance is None:
 126         logger.error("Error while booting instance.")
 127         sys.exit(-1)
 128     else:
 129         logger.debug("Instance '%s' booted successfully. IP='%s'." %
 130                      (name, instance.networks.itervalues().next()[0]))
 131     # Retrieve IP of INSTANCE
 132     # instance_ip = instance.networks.get(network_id)[0]
 133
 134     if secgroup_name:
 135         logger.debug("Adding '%s' to security group '%s'..."
 136                      % (name, secgroup_name))
 137     else:
 138         logger.debug("Adding '%s' to security group '%s'..."
 139                      % (name, sg_id))
 140     os_utils.add_secgroup_to_instance(nova_client, instance.id, sg_id)
 141
 142     return instance
 143
 144
 145 def generate_ping_userdata(ips_array, ping_count=10):
 146     ips = ""
 147     for ip in ips_array:
 148         ips = ("%s %s" % (ips, ip))
 149
 150     ips = ips.replace('  ', ' ')
 151     return ("#!/bin/sh\n"
 152             "set%s\n"
 153             "while true; do\n"
 154             " for i do\n"
 155             "  ip=$i\n"
 156             "  ping -c %s $ip 2>&1 >/dev/null\n"
 157             "  RES=$?\n"
 158             "  if [ \"Z$RES\" = \"Z0\" ] ; then\n"
 159             "   echo ping $ip OK\n"
 160             "  else echo ping $ip KO\n"
 161             "  fi\n"
 162             " done\n"
 163             " sleep 1\n"
 164             "done\n"
 165             % (ips, ping_count))
 166
 167
 168 def generate_userdata_common():
 169     return ("#!/bin/sh\n"
 170             "sudo mkdir -p /home/cirros/.ssh/\n"
 171             "sudo chown cirros:cirros /home/cirros/.ssh/\n"
 172             "sudo chown cirros:cirros /home/cirros/id_rsa\n"
 173             "mv /home/cirros/id_rsa /home/cirros/.ssh/\n"
 174             "sudo echo ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAgnWtSS98Am516e"
 175             "stBsq0jbyOB4eLMUYDdgzsUHsnxFQCtACwwAg9/2uq3FoGUBUWeHZNsT6jcK9"
 176             "sCMEYiS479CUCzbrxcd8XaIlK38HECcDVglgBNwNzX/WDfMejXpKzZG61s98rU"
 177             "ElNvZ0YDqhaqZGqxIV4ejalqLjYrQkoly3R+2k= "
 178             "cirros@test1>/home/cirros/.ssh/authorized_keys\n"
 179             "sudo chown cirros:cirros /home/cirros/.ssh/authorized_keys\n"
 180             "chmod 700 /home/cirros/.ssh\n"
 181             "chmod 644 /home/cirros/.ssh/authorized_keys\n"
 182             "chmod 600 /home/cirros/.ssh/id_rsa\n"
 183             )
 184
 185
 186 def generate_userdata_with_ssh(ips_array):
 187     u1 = generate_userdata_common()
 188
 189     ips = ""
 190     for ip in ips_array:
 191         ips = ("%s %s" % (ips, ip))
 192
 193     ips = ips.replace('  ', ' ')
 194     u2 = ("#!/bin/sh\n"
 195           "set%s\n"
 196           "while true; do\n"
 197           " for i do\n"
 198           "  ip=$i\n"
 199           "  hostname=$(ssh -y -i /home/cirros/.ssh/id_rsa "
 200           "cirros@$ip 'hostname' </dev/zero 2>/dev/null)\n"
 201           "  RES=$?\n"
 202           "  if [ \"Z$RES\" = \"Z0\" ]; then echo $ip $hostname;\n"
 203           "  else echo $ip 'not reachable';fi;\n"
 204           " done\n"
 205           " sleep 1\n"
 206           "done\n"
 207           % ips)
 208     return (u1 + u2)
 209
 210
 211 def get_installerHandler():
 212     installer_type = str(os.environ['INSTALLER_TYPE'].lower())
 213     installer_ip = get_installer_ip()
 214
 215     if installer_type not in ["fuel", "apex"]:
 216         logger.warn("installer type %s is neither fuel nor apex."
 217                     "returning None for installer handler" % installer_type)
 218         return None
 219     else:
 220         if installer_type in ["apex"]:
 221             developHandler = DeploymentFactory.get_handler(
 222                 installer_type,
 223                 installer_ip,
 224                 'root',
 225                 pkey_file="/root/.ssh/id_rsa")
 226
 227         if installer_type in ["fuel"]:
 228             developHandler = DeploymentFactory.get_handler(
 229                 installer_type,
 230                 installer_ip,
 231                 'root',
 232                 'r00tme')
 233         return developHandler
 234
 235
 236 def get_nodes():
 237     developHandler = get_installerHandler()
 238     return developHandler.get_nodes()
 239
 240
 241 def get_installer_ip():
 242     return str(os.environ['INSTALLER_IP'])
 243
 244
 245 def get_instance_ip(instance):
 246     instance_ip = instance.networks.itervalues().next()[0]
 247     return instance_ip
 248
 249
 250 def wait_for_instance(instance):
 251     logger.info("Waiting for instance %s to get a DHCP lease and "
 252                 "prompt for login..." % instance.id)
 253     # The sleep this function replaced waited for 80s
 254     tries = 40
 255     sleep_time = 2
 256     pattern = ".* login:"
 257     expected_regex = re.compile(pattern)
 258     console_log = ""
 259     while tries > 0 and not expected_regex.search(console_log):
 260         console_log = instance.get_console_output()
 261         time.sleep(sleep_time)
 262         tries -= 1
 263
 264     if not expected_regex.search(console_log):
 265         logger.error("Instance %s seems not to boot up properly."
 266                      % instance.id)
 267         return False
 268     return True
 269
 270
 271 def wait_for_instances_up(*args):
 272     check = [wait_for_instance(instance) for instance in args]
 273     return all(check)
 274
 275
 276 def wait_for_bgp_net_assoc(neutron_client, bgpvpn_id, net_id):
 277     tries = 30
 278     sleep_time = 1
 279     nets = []
 280     logger.debug("Waiting for network %s to associate with BGPVPN %s "
 281                  % (bgpvpn_id, net_id))
 282
 283     while tries > 0 and net_id not in nets:
 284         nets = get_bgpvpn_networks(neutron_client, bgpvpn_id)
 285         time.sleep(sleep_time)
 286         tries -= 1
 287     if net_id not in nets:
 288         logger.error("Association of network %s with BGPVPN %s failed" %
 289                      (net_id, bgpvpn_id))
 290         return False
 291     return True
 292
 293
 294 def wait_for_bgp_net_assocs(neutron_client, bgpvpn_id, *args):
 295     check = [wait_for_bgp_net_assoc(neutron_client, bgpvpn_id, id)
 296              for id in args]
 297     # Return True if all associations succeeded
 298     return all(check)
 299
 300
 301 def wait_for_bgp_router_assoc(neutron_client, bgpvpn_id, router_id):
 302     tries = 30
 303     sleep_time = 1
 304     routers = []
 305     logger.debug("Waiting for router %s to associate with BGPVPN %s "
 306                  % (bgpvpn_id, router_id))
 307     while tries > 0 and router_id not in routers:
 308         routers = get_bgpvpn_routers(neutron_client, bgpvpn_id)
 309         time.sleep(sleep_time)
 310         tries -= 1
 311     if router_id not in routers:
 312         logger.error("Association of router %s with BGPVPN %s failed" %
 313                      (router_id, bgpvpn_id))
 314         return False
 315     return True
 316
 317
 318 def wait_for_bgp_router_assocs(neutron_client, bgpvpn_id, *args):
 319     check = [wait_for_bgp_router_assoc(neutron_client, bgpvpn_id, id)
 320              for id in args]
 321     # Return True if all associations succeeded
 322     return all(check)
 323
 324
 325 def wait_before_subtest(*args, **kwargs):
 326     ''' This is a placeholder.
 327         TODO: Replace delay with polling logic. '''
 328     time.sleep(30)
 329
 330
 331 def assert_and_get_compute_nodes(nova_client, required_node_number=2):
 332     """Get the compute nodes in the deployment
 333     Exit if the deployment doesn't have enough compute nodes"""
 334     compute_nodes = os_utils.get_hypervisors(nova_client)
 335
 336     num_compute_nodes = len(compute_nodes)
 337     if num_compute_nodes < 2:
 338         logger.error("There are %s compute nodes in the deployment. "
 339                      "Minimum number of nodes to complete the test is 2."
 340                      % num_compute_nodes)
 341         sys.exit(-1)
 342
 343     logger.debug("Compute nodes: %s" % compute_nodes)
 344     return compute_nodes
 345
 346
 347 def open_icmp(neutron_client, security_group_id):
 348     if os_utils.check_security_group_rules(neutron_client,
 349                                            security_group_id,
 350                                            'ingress',
 351                                            'icmp'):
 352
 353         if not os_utils.create_secgroup_rule(neutron_client,
 354                                              security_group_id,
 355                                              'ingress',
 356                                              'icmp'):
 357             logger.error("Failed to create icmp security group rule...")
 358     else:
 359         logger.info("This rule exists for security group: %s"
 360                     % security_group_id)
 361
 362
 363 def open_http_port(neutron_client, security_group_id):
 364     if os_utils.check_security_group_rules(neutron_client,
 365                                            security_group_id,
 366                                            'ingress',
 367                                            'tcp',
 368                                            80, 80):
 369
 370         if not os_utils.create_secgroup_rule(neutron_client,
 371                                              security_group_id,
 372                                              'ingress',
 373                                              'tcp',
 374                                              80, 80):
 375
 376             logger.error("Failed to create http security group rule...")
 377     else:
 378         logger.info("This rule exists for security group: %s"
 379                     % security_group_id)
 380
 381
 382 def open_bgp_port(neutron_client, security_group_id):
 383     if os_utils.check_security_group_rules(neutron_client,
 384                                            security_group_id,
 385                                            'ingress',
 386                                            'tcp',
 387                                            179, 179):
 388
 389         if not os_utils.create_secgroup_rule(neutron_client,
 390                                              security_group_id,
 391                                              'ingress',
 392                                              'tcp',
 393                                              179, 179):
 394             logger.error("Failed to create bgp security group rule...")
 395     else:
 396         logger.info("This rule exists for security group: %s"
 397                     % security_group_id)
 398
 399
 400 def exec_cmd(cmd, verbose):
 401     success = True
 402     logger.debug("Executing '%s'" % cmd)
 403     p = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE,
 404                          stderr=subprocess.STDOUT)
 405     output = ""
 406     for line in iter(p.stdout.readline, b''):
 407         output += line
 408
 409     if verbose:
 410         logger.debug(output)
 411
 412     p.stdout.close()
 413     returncode = p.wait()
 414     if returncode != 0:
 415         logger.error("Command %s failed to execute." % cmd)
 416         success = False
 417
 418     return output, success
 419
 420
 421 def check_odl_fib(ip, controller_ip):
 422     """Check that there is an entry in the ODL Fib for `ip`"""
 423     url = "http://" + controller_ip + \
 424           ":8181/restconf/config/odl-fib:fibEntries/"
 425     logger.debug("Querring '%s' for FIB entries", url)
 426     res = requests.get(url, auth=(ODL_USER, ODL_PASS))
 427     if res.status_code != 200:
 428         logger.error("OpenDaylight response status code: %s", res.status_code)
 429         return False
 430     logger.debug("Checking whether '%s' is in the OpenDaylight FIB"
 431                  % controller_ip)
 432     logger.debug("OpenDaylight FIB: \n%s" % res.text)
 433     return ip in res.text
 434
 435
 436 def run_odl_cmd(odl_node, cmd):
 437     '''Run a command in the OpenDaylight Karaf shell
 438     This is a bit flimsy because of shell quote escaping, make sure that
 439     the cmd passed does not have any top level double quotes or this
 440     function will break.
 441     The /dev/null is used because client works, but outputs something
 442     that contains "ERROR" and run_cmd doesn't like that.
 443     '''
 444     karaf_cmd = ('/opt/opendaylight/bin/client -h 127.0.0.1 "%s"'
 445                  ' 2>/dev/null' % cmd)
 446     return odl_node.run_cmd(karaf_cmd)
 447
 448
 449 def wait_for_cloud_init(instance):
 450     success = True
 451     # ubuntu images take a long time to start
 452     tries = 20
 453     sleep_time = 30
 454     logger.info("Waiting for cloud init of instance: {}"
 455                 "".format(instance.name))
 456     while tries > 0:
 457         instance_log = instance.get_console_output()
 458         if "Failed to run module" in instance_log:
 459             success = False
 460             logger.error("Cloud init failed to run. Reason: %s",
 461                          instance_log)
 462             break
 463         if re.search(r"Cloud-init v. .+ finished at", instance_log):
 464             success = True
 465             break
 466         time.sleep(sleep_time)
 467         tries = tries - 1
 468
 469     if tries == 0:
 470         logger.error("Cloud init timed out"
 471                      ". Reason: %s",
 472                      instance_log)
 473         success = False
 474     logger.info("Finished waiting for cloud init of instance {} result was {}"
 475                 "".format(instance.name, success))
 476     return success
 477
 478
 479 def attach_instance_to_ext_br(instance, compute_node):
 480     libvirt_instance_name = getattr(instance, "OS-EXT-SRV-ATTR:instance_name")
 481     installer_type = str(os.environ['INSTALLER_TYPE'].lower())
 482     if installer_type == "fuel":
 483         bridge = "br-ex"
 484     elif installer_type == "apex":
 485         # In Apex, br-ex is an ovs bridge and virsh attach-interface
 486         # won't just work. We work around it by creating a linux
 487         # bridge, attaching that to br-ex with a veth pair
 488         # and virsh-attaching the instance to the linux-bridge
 489         bridge = "br-quagga"
 490         cmd = """
 491         set -e
 492         if ! sudo brctl show |grep -q ^{bridge};then
 493           sudo brctl addbr {bridge}
 494           sudo ip link set {bridge} up
 495           sudo ip link add quagga-tap type veth peer name ovs-quagga-tap
 496           sudo ip link set dev ovs-quagga-tap up
 497           sudo ip link set dev quagga-tap up
 498           sudo ovs-vsctl add-port br-ex ovs-quagga-tap
 499           sudo brctl addif {bridge} quagga-tap
 500         fi
 501         """
 502         compute_node.run_cmd(cmd.format(bridge=bridge))
 503
 504     compute_node.run_cmd("sudo virsh attach-interface %s"
 505                          " bridge %s" % (libvirt_instance_name, bridge))
 506
 507
 508 def detach_instance_from_ext_br(instance, compute_node):
 509     libvirt_instance_name = getattr(instance, "OS-EXT-SRV-ATTR:instance_name")
 510     mac = compute_node.run_cmd("for vm in $(sudo virsh list | "
 511                                "grep running | awk '{print $2}'); "
 512                                "do echo -n ; sudo virsh dumpxml $vm| "
 513                                "grep -oP '52:54:[\da-f:]+' ;done")
 514     compute_node.run_cmd("sudo virsh detach-interface --domain %s"
 515                          " --type bridge --mac %s"
 516                          % (libvirt_instance_name, mac))
 517
 518     installer_type = str(os.environ['INSTALLER_TYPE'].lower())
 519     if installer_type == "fuel":
 520         bridge = "br-ex"
 521     elif installer_type == "apex":
 522         # In Apex, br-ex is an ovs bridge and virsh attach-interface
 523         # won't just work. We work around it by creating a linux
 524         # bridge, attaching that to br-ex with a veth pair
 525         # and virsh-attaching the instance to the linux-bridge
 526         bridge = "br-quagga"
 527         cmd = """
 528             sudo brctl delif {bridge} quagga-tap &&
 529             sudo ovs-vsctl del-port br-ex ovs-quagga-tap &&
 530             sudo ip link set dev quagga-tap down &&
 531             sudo ip link set dev ovs-quagga-tap down &&
 532             sudo ip link del quagga-tap type veth peer name ovs-quagga-tap &&
 533             sudo ip link set {bridge} down &&
 534             sudo brctl delbr {bridge}
 535         """
 536         compute_node.run_cmd(cmd.format(bridge=bridge))
 537
 538
 539 def cleanup_neutron(neutron_client, floatingip_ids, bgpvpn_ids, interfaces,
 540                     subnet_ids, router_ids, network_ids):
 541
 542     if len(floatingip_ids) != 0:
 543         for floatingip_id in floatingip_ids:
 544             if not os_utils.delete_floating_ip(neutron_client, floatingip_id):
 545                 logging.error('Fail to delete all floating ips. '
 546                               'Floating ip with id {} was not deleted.'.
 547                               format(floatingip_id))
 548                 return False
 549
 550     if len(bgpvpn_ids) != 0:
 551         for bgpvpn_id in bgpvpn_ids:
 552             delete_bgpvpn(neutron_client, bgpvpn_id)
 553
 554     if len(interfaces) != 0:
 555         for router_id, subnet_id in interfaces:
 556             if not os_utils.remove_interface_router(neutron_client,
 557                                                     router_id, subnet_id):
 558                 logging.error('Fail to delete all interface routers. '
 559                               'Interface router with id {} was not deleted.'.
 560                               format(router_id))
 561
 562     if len(router_ids) != 0:
 563         for router_id in router_ids:
 564             if not os_utils.remove_gateway_router(neutron_client, router_id):
 565                 logging.error('Fail to delete all gateway routers. '
 566                               'Gateway router with id {} was not deleted.'.
 567                               format(router_id))
 568
 569     if len(subnet_ids) != 0:
 570         for subnet_id in subnet_ids:
 571             if not os_utils.delete_neutron_subnet(neutron_client, subnet_id):
 572                 logging.error('Fail to delete all subnets. '
 573                               'Subnet with id {} was not deleted.'.
 574                               format(subnet_id))
 575                 return False
 576
 577     if len(router_ids) != 0:
 578         for router_id in router_ids:
 579             if not os_utils.delete_neutron_router(neutron_client, router_id):
 580                 logging.error('Fail to delete all routers. '
 581                               'Router with id {} was not deleted.'.
 582                               format(router_id))
 583                 return False
 584
 585     if len(network_ids) != 0:
 586         for network_id in network_ids:
 587             if not os_utils.delete_neutron_net(neutron_client, network_id):
 588                 logging.error('Fail to delete all networks. '
 589                               'Network with id {} was not deleted.'.
 590                               format(network_id))
 591                 return False
 592     return True
 593
 594
 595 def cleanup_nova(nova_client, instance_ids, image_ids):
 596     if len(instance_ids) != 0:
 597         for instance_id in instance_ids:
 598             if not os_utils.delete_instance(nova_client, instance_id):
 599                 logging.error('Fail to delete all instances. '
 600                               'Instance with id {} was not deleted.'.
 601                               format(instance_id))
 602                 return False
 603
 604     if len(image_ids) != 0:
 605         for image_id in image_ids:
 606             if not os_utils.delete_glance_image(nova_client, image_id):
 607                 logging.error('Fail to delete all images. '
 608                               'Image with id {} was not deleted.'.
 609                               format(image_id))
 610                 return False
 611     return True
 612
 613
 614 def create_bgpvpn(neutron_client, **kwargs):
 615     # route_distinguishers
 616     # route_targets
 617     json_body = {"bgpvpn": kwargs}
 618     return neutron_client.create_bgpvpn(json_body)
 619
 620
 621 def update_bgpvpn(neutron_client, bgpvpn_id, **kwargs):
 622     json_body = {"bgpvpn": kwargs}
 623     return neutron_client.update_bgpvpn(bgpvpn_id, json_body)
 624
 625
 626 def delete_bgpvpn(neutron_client, bgpvpn_id):
 627     return neutron_client.delete_bgpvpn(bgpvpn_id)
 628
 629
 630 def get_bgpvpn(neutron_client, bgpvpn_id):
 631     return neutron_client.show_bgpvpn(bgpvpn_id)
 632
 633
 634 def get_bgpvpn_routers(neutron_client, bgpvpn_id):
 635     return get_bgpvpn(neutron_client, bgpvpn_id)['bgpvpn']['routers']
 636
 637
 638 def get_bgpvpn_networks(neutron_client, bgpvpn_id):
 639     return get_bgpvpn(neutron_client, bgpvpn_id)['bgpvpn']['networks']
 640
 641
 642 def create_router_association(neutron_client, bgpvpn_id, router_id):
 643     json_body = {"router_association": {"router_id": router_id}}
 644     return neutron_client.create_router_association(bgpvpn_id, json_body)
 645
 646
 647 def create_network_association(neutron_client, bgpvpn_id, neutron_network_id):
 648     json_body = {"network_association": {"network_id": neutron_network_id}}
 649     return neutron_client.create_network_association(bgpvpn_id, json_body)
 650
 651
 652 def is_fail_mode_secure():
 653     """
 654     Checks the value of the attribute fail_mode,
 655     if it is set to secure. This check is performed
 656     on all OVS br-int interfaces, for all OpenStack nodes.
 657     """
 658     is_secure = {}
 659     openstack_nodes = get_nodes()
 660     get_ovs_int_cmd = ("sudo ovs-vsctl show | "
 661                        "grep -i bridge | "
 662                        "awk '{print $2}'")
 663     # Define OVS get fail_mode command
 664     get_ovs_fail_mode_cmd = ("sudo ovs-vsctl get-fail-mode br-int")
 665     for openstack_node in openstack_nodes:
 666         if not openstack_node.is_active():
 667             continue
 668
 669         ovs_int_list = (openstack_node.run_cmd(get_ovs_int_cmd).
 670                         strip().split('\n'))
 671         if 'br-int' in ovs_int_list:
 672             # Execute get fail_mode command
 673             br_int_fail_mode = (openstack_node.
 674                                 run_cmd(get_ovs_fail_mode_cmd).strip())
 675             if br_int_fail_mode == 'secure':
 676                 # success
 677                 is_secure[openstack_node.name] = True
 678             else:
 679                 # failure
 680                 logging.error('The fail_mode for br-int was not secure '
 681                               'in {} node'.format(openstack_node.name))
 682                 is_secure[openstack_node.name] = False
 683     return is_secure
 684
 685
 686 def update_nw_subnet_port_quota(neutron_client, tenant_id, nw_quota,
 687                                 subnet_quota, port_quota):
 688     json_body = {"quota": {
 689         "network": nw_quota,
 690         "subnet": subnet_quota,
 691         "port": port_quota
 692     }}
 693
 694     try:
 695         neutron_client.update_quota(tenant_id=tenant_id,
 696                                     body=json_body)
 697         return True
 698     except Exception as e:
 699         logger.error("Error [update_nw_subnet_port_quota(neutron_client,"
 700                      " '%s', '%s', '%s', '%s')]: %s" %
 701                      (tenant_id, nw_quota, subnet_quota, port_quota, e))
 702         return False
 703
 704
 705 def update_instance_quota_class(nova_client, instances_quota):
 706     try:
 707         nova_client.quota_classes.update("default", instances=instances_quota)
 708         return True
 709     except Exception as e:
 710         logger.error("Error [update_instance_quota_class(nova_client,"
 711                      " '%s' )]: %s" % (instances_quota, e))
 712         return False
 713
 714
 715 def get_neutron_quota(neutron_client, tenant_id):
 716     try:
 717         return neutron_client.show_quota(tenant_id=tenant_id)['quota']
 718     except Exception as e:
 719         logger.error("Error in getting neutron quota for tenant "
 720                      " '%s' )]: %s" % (tenant_id, e))
 721         raise
 722
 723
 724 def get_nova_instances_quota(nova_client):
 725     try:
 726         return nova_client.quota_classes.get("default").instances
 727     except Exception as e:
 728         logger.error("Error in getting nova instances quota: %s" % e)
 729         raise