diff options
Diffstat (limited to 'test/functest/utils.py')
| -rw-r--r-- | test/functest/utils.py | 303 |
1 files changed, 0 insertions, 303 deletions
diff --git a/test/functest/utils.py b/test/functest/utils.py deleted file mode 100644 index cbfc88d..0000000 --- a/test/functest/utils.py +++ /dev/null @@ -1,303 +0,0 @@ -#!/usr/bin/python -# -# Copyright (c) 2015 All rights reserved -# This program and the accompanying materials -# are made available under the terms of the Apache License, Version 2.0 -# which accompanies this distribution, and is available at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -import sys -import time - -import functest.utils.functest_logger as ft_logger -import functest.utils.openstack_utils as os_utils -import re - -import config as sdnvpn_config - -logger = ft_logger.Logger("sndvpn_test_utils").getLogger() - -common_config = sdnvpn_config.CommonConfig() - - -def create_net(neutron_client, name): - logger.debug("Creating network %s", name) - net_id = os_utils.create_neutron_net(neutron_client, name) - if not net_id: - logger.error( - "There has been a problem when creating the neutron network") - sys.exit(-1) - return net_id - - -def create_subnet(neutron_client, name, cidr, net_id): - logger.debug("Creating subnet %s in network %s with cidr %s", - name, net_id, cidr) - subnet_id = os_utils.create_neutron_subnet(neutron_client, - name, - cidr, - net_id) - if not subnet_id: - logger.error( - "There has been a problem when creating the neutron subnet") - sys.exit(-1) - return subnet_id - - -def create_network(neutron_client, net, subnet1, cidr1, - router, subnet2=None, cidr2=None): - """Network assoc will not work for networks/subnets created by this function. - - It is an ODL limitation due to it handling routers as vpns. - See https://bugs.opendaylight.org/show_bug.cgi?id=6962""" - network_dic = os_utils.create_network_full(neutron_client, - net, - subnet1, - router, - cidr1) - if not network_dic: - logger.error( - "There has been a problem when creating the neutron network") - sys.exit(-1) - net_id = network_dic["net_id"] - subnet_id = network_dic["subnet_id"] - router_id = network_dic["router_id"] - - if subnet2 is not None: - logger.debug("Creating and attaching a second subnet...") - subnet_id = os_utils.create_neutron_subnet( - neutron_client, subnet2, cidr2, net_id) - if not subnet_id: - logger.error( - "There has been a problem when creating the second subnet") - sys.exit(-1) - logger.debug("Subnet '%s' created successfully" % subnet_id) - return net_id, subnet_id, router_id - - -def create_instance(nova_client, - name, - image_id, - network_id, - sg_id, - secgroup_name=None, - fixed_ip=None, - compute_node='', - userdata=None, - files=None, - **kwargs - ): - if 'flavor' not in kwargs: - kwargs['flavor'] = common_config.default_flavor - - logger.info("Creating instance '%s'..." % name) - logger.debug( - "Configuration:\n name=%s \n flavor=%s \n image=%s \n" - " network=%s\n secgroup=%s \n hypervisor=%s \n" - " fixed_ip=%s\n files=%s\n userdata=\n%s\n" - % (name, kwargs['flavor'], image_id, network_id, sg_id, - compute_node, fixed_ip, files, userdata)) - instance = os_utils.create_instance_and_wait_for_active( - kwargs['flavor'], - image_id, - network_id, - name, - config_drive=True, - userdata=userdata, - av_zone=compute_node, - fixed_ip=fixed_ip, - files=files) - - if instance is None: - logger.error("Error while booting instance.") - sys.exit(-1) - else: - logger.debug("Instance '%s' booted successfully. IP='%s'." % - (name, instance.networks.itervalues().next()[0])) - # Retrieve IP of INSTANCE - # instance_ip = instance.networks.get(network_id)[0] - - if secgroup_name: - logger.debug("Adding '%s' to security group '%s'..." - % (name, secgroup_name)) - else: - logger.debug("Adding '%s' to security group '%s'..." - % (name, sg_id)) - os_utils.add_secgroup_to_instance(nova_client, instance.id, sg_id) - - return instance - - -def generate_ping_userdata(ips_array): - ips = "" - for ip in ips_array: - ips = ("%s %s" % (ips, ip)) - - ips = ips.replace(' ', ' ') - return ("#!/bin/sh\n" - "set%s\n" - "while true; do\n" - " for i do\n" - " ip=$i\n" - " ping -c 1 $ip 2>&1 >/dev/null\n" - " RES=$?\n" - " if [ \"Z$RES\" = \"Z0\" ] ; then\n" - " echo ping $ip OK\n" - " else echo ping $ip KO\n" - " fi\n" - " done\n" - " sleep 1\n" - "done\n" - % ips) - - -def generate_userdata_common(): - return ("#!/bin/sh\n" - "sudo mkdir -p /home/cirros/.ssh/\n" - "sudo chown cirros:cirros /home/cirros/.ssh/\n" - "sudo chown cirros:cirros /home/cirros/id_rsa\n" - "mv /home/cirros/id_rsa /home/cirros/.ssh/\n" - "sudo echo ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAAAgnWtSS98Am516e" - "stBsq0jbyOB4eLMUYDdgzsUHsnxFQCtACwwAg9/2uq3FoGUBUWeHZNsT6jcK9" - "sCMEYiS479CUCzbrxcd8XaIlK38HECcDVglgBNwNzX/WDfMejXpKzZG61s98rU" - "ElNvZ0YDqhaqZGqxIV4ejalqLjYrQkoly3R+2k= " - "cirros@test1>/home/cirros/.ssh/authorized_keys\n" - "sudo chown cirros:cirros /home/cirros/.ssh/authorized_keys\n" - "chmod 700 /home/cirros/.ssh\n" - "chmod 644 /home/cirros/.ssh/authorized_keys\n" - "chmod 600 /home/cirros/.ssh/id_rsa\n" - ) - - -def generate_userdata_with_ssh(ips_array): - u1 = generate_userdata_common() - - ips = "" - for ip in ips_array: - ips = ("%s %s" % (ips, ip)) - - ips = ips.replace(' ', ' ') - u2 = ("#!/bin/sh\n" - "set%s\n" - "while true; do\n" - " for i do\n" - " ip=$i\n" - " hostname=$(ssh -y -i /home/cirros/.ssh/id_rsa " - "cirros@$ip 'hostname' </dev/zero 2>/dev/null)\n" - " RES=$?\n" - " if [ \"Z$RES\" = \"Z0\" ]; then echo $ip $hostname;\n" - " else echo $ip 'not reachable';fi;\n" - " done\n" - " sleep 1\n" - "done\n" - % ips) - return (u1 + u2) - - -def wait_for_instance(instance): - logger.info("Waiting for instance %s to get a DHCP lease..." % instance.id) - # The sleep this function replaced waited for 80s - tries = 40 - sleep_time = 2 - pattern = "Lease of .* obtained, lease time" - expected_regex = re.compile(pattern) - console_log = "" - while tries > 0 and not expected_regex.search(console_log): - console_log = instance.get_console_output() - time.sleep(sleep_time) - tries -= 1 - - if not expected_regex.search(console_log): - logger.error("Instance %s seems to have failed leasing an IP." - % instance.id) - return False - return True - - -def wait_for_instances_up(*args): - check = [wait_for_instance(instance) for instance in args] - return all(check) - - -def wait_for_bgp_net_assoc(neutron_client, bgpvpn_id, net_id): - tries = 30 - sleep_time = 1 - nets = [] - logger.debug("Waiting for network %s to associate with BGPVPN %s " - % (bgpvpn_id, net_id)) - - while tries > 0 and net_id not in nets: - nets = os_utils.get_bgpvpn_networks(neutron_client, bgpvpn_id) - time.sleep(sleep_time) - tries -= 1 - if net_id not in nets: - logger.error("Association of network %s with BGPVPN %s failed" % - (net_id, bgpvpn_id)) - return False - return True - - -def wait_for_bgp_net_assocs(neutron_client, bgpvpn_id, *args): - check = [wait_for_bgp_net_assoc(neutron_client, bgpvpn_id, id) - for id in args] - # Return True if all associations succeeded - return all(check) - - -def wait_for_bgp_router_assoc(neutron_client, bgpvpn_id, router_id): - tries = 30 - sleep_time = 1 - routers = [] - logger.debug("Waiting for router %s to associate with BGPVPN %s " - % (bgpvpn_id, router_id)) - while tries > 0 and router_id not in routers: - routers = os_utils.get_bgpvpn_routers(neutron_client, bgpvpn_id) - time.sleep(sleep_time) - tries -= 1 - if router_id not in routers: - logger.error("Association of router %s with BGPVPN %s failed" % - (router_id, bgpvpn_id)) - return False - return True - - -def wait_for_bgp_router_assocs(neutron_client, bgpvpn_id, *args): - check = [wait_for_bgp_router_assoc(neutron_client, bgpvpn_id, id) - for id in args] - # Return True if all associations succeeded - return all(check) - - -def wait_before_subtest(*args, **kwargs): - ''' This is a placeholder. - TODO: Replace delay with polling logic. ''' - time.sleep(30) - - -def assert_and_get_compute_nodes(nova_client, required_node_number=2): - """Get the compute nodes in the deployment - - Exit if the deployment doesn't have enough compute nodes""" - compute_nodes = os_utils.get_hypervisors(nova_client) - - num_compute_nodes = len(compute_nodes) - if num_compute_nodes < 2: - logger.error("There are %s compute nodes in the deployment. " - "Minimum number of nodes to complete the test is 2." - % num_compute_nodes) - sys.exit(-1) - - logger.debug("Compute nodes: %s" % compute_nodes) - return compute_nodes - - -def open_icmp_ssh(neutron_client, security_group_id): - os_utils.create_secgroup_rule(neutron_client, - security_group_id, - 'ingress', - 'icmp') - os_utils.create_secgroup_rule(neutron_client, - security_group_id, - 'tcp', - 80, 80) |