Added more logs to track better errors
[functest.git] / testcases / features / sfc / sfc.py
index 562d410..122ed5d 100755 (executable)
@@ -1,13 +1,14 @@
-import argparse
 import os
 import subprocess
 import sys
 import time
+
+import argparse
+import paramiko
+
 import functest.utils.functest_logger as ft_logger
 import functest.utils.functest_utils as ft_utils
 import functest.utils.openstack_utils as os_utils
-import paramiko
-
 
 parser = argparse.ArgumentParser()
 
@@ -20,14 +21,15 @@ args = parser.parse_args()
 """ logging configuration """
 logger = ft_logger.Logger("ODL_SFC").getLogger()
 
-REPO_PATH = os.environ['repos_dir'] + '/functest/'
+FUNCTEST_REPO = ft_utils.FUNCTEST_REPO
+
 HOME = os.environ['HOME'] + "/"
 
 VM_BOOT_TIMEOUT = 180
 INSTANCE_NAME = "client"
-FLAVOR = "m1.small"
-IMAGE_NAME = "sf_summit2016"
-IMAGE_FILENAME = "sf_summit2016.qcow2"
+FLAVOR = "custom"
+IMAGE_NAME = "sf_nsh_colorado"
+IMAGE_FILENAME = "sf_nsh_colorado.qcow2"
 IMAGE_FORMAT = "qcow2"
 IMAGE_PATH = "/home/opnfv/functest/data" + "/" + IMAGE_FILENAME
 
@@ -50,9 +52,56 @@ TACKER_SCRIPT = 'sfc_tacker.bash'
 TEARDOWN_SCRIPT = "sfc_teardown.bash"
 TACKER_CHANGECLASSI = "sfc_change_classi.bash"
 
+ssh_options = '-o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no'
+
+
+def check_ssh(ip):
+    cmd = "sshpass -p opnfv ssh " + ssh_options + " -q " + ip + " exit"
+    success = subprocess.call(cmd, shell=True) == 0
+    if not success:
+        logger.debug("Wating for SSH connectivity in SF with IP: %s" % ip)
+    return success
+
 
 def main():
 
+    # Allow any port so that tacker commands reaches the server.
+    # This will be deleted when tacker is included in OPNFV installation
+
+    status = "PASS"
+    failures = 0
+    start_time = time.time()
+    json_results = {}
+
+    contr_cmd = ("sshpass -p r00tme ssh " + ssh_options + " root@10.20.0.2"
+                 " 'fuel node'|grep controller|awk '{print $10}'")
+    logger.info("Executing script to get ip_server: '%s'" % contr_cmd)
+    process = subprocess.Popen(contr_cmd,
+                               shell=True,
+                               stdout=subprocess.PIPE,
+                               stderr=subprocess.PIPE)
+    ip_server = process.stdout.readline().rstrip()
+
+    iptable_cmd1 = ("sshpass -p r00tme ssh " + ssh_options + " root@10.20.0.2"
+                    " ssh " + ip_server + " iptables -P INPUT ACCEPT ")
+    iptable_cmd2 = ("sshpass -p r00tme ssh " + ssh_options + " root@10.20.0.2"
+                    " ssh " + ip_server + " iptables -t nat -P INPUT ACCEPT ")
+    iptable_cmd3 = ("sshpass -p r00tme ssh " + ssh_options + " root@10.20.0.2"
+                    " ssh " + ssh_options + " " + ip_server +
+                    " iptables -A INPUT -m state"
+                    " --state NEW,ESTABLISHED,RELATED -j ACCEPT")
+
+    logger.info("Changing firewall policy in controller: '%s'" % iptable_cmd1)
+    subprocess.call(iptable_cmd1, shell=True, stderr=subprocess.PIPE)
+
+    logger.info("Changing firewall policy in controller: '%s'" % iptable_cmd2)
+    subprocess.call(iptable_cmd2, shell=True, stderr=subprocess.PIPE)
+
+    logger.info("Changing firewall policy in controller: '%s'" % iptable_cmd3)
+    subprocess.call(iptable_cmd3, shell=True, stderr=subprocess.PIPE)
+
+# Getting the different clients
+
     nova_client = os_utils.get_nova_client()
     neutron_client = os_utils.get_neutron_client()
     glance_client = os_utils.get_glance_client()
@@ -62,30 +111,11 @@ def main():
     if not os.path.isfile(IMAGE_PATH):
         logger.info("Downloading image")
         ft_utils.download_url(
-            "http://artifacts.opnfv.org/sfc/demo/sf_summit2016.qcow2",
+            "http://artifacts.opnfv.org/sfc/demo/sf_nsh_colorado.qcow2",
             "/home/opnfv/functest/data/")
     else:
         logger.info("Using old image")
 
-# Allow any port so that tacker commands reaches the server.
-# CHECK IF THIS STILL MAKES SENSE WHEN TACKER IS INCLUDED IN OPNFV INSTALATION
-
-    controller_command = ("sshpass -p r00tme ssh root@10.20.0.2"
-                          " 'fuel node'|grep controller|awk '{print $10}'")
-    logger.info("Executing tacker script: '%s'" % controller_command)
-    process = subprocess.Popen(controller_command,
-                               shell=True,
-                               stdout=subprocess.PIPE)
-    ip = process.stdout.readline()
-
-    iptable_command1 = ("sshpass -p r00tme ssh root@10.20.0.2 ssh"
-                        " " + ip + " iptables -P INPUT ACCEPT ")
-    iptable_command2 = ("sshpass -p r00tme ssh root@10.20.0.2 ssh"
-                        " " + ip + " iptables -t nat -P INPUT ACCEPT ")
-
-    subprocess.call(iptable_command1, shell=True)
-    subprocess.call(iptable_command2, shell=True)
-
 # Create glance image and the neutron network
 
     image_id = os_utils.create_glance_image(glance_client,
@@ -110,7 +140,33 @@ def main():
     sg_id = os_utils.create_security_group_full(neutron_client,
                                                 SECGROUP_NAME, SECGROUP_DESCR)
 
-    # boot INTANCE
+    secgroups = os_utils.get_security_groups(neutron_client)
+
+    for sg in secgroups:
+        os_utils.create_secgroup_rule(neutron_client, sg['id'],
+                                      'ingress', 'tcp',
+                                      port_range_min=22,
+                                      port_range_max=22)
+        os_utils.create_secgroup_rule(neutron_client, sg['id'],
+                                      'egress', 'tcp',
+                                      port_range_min=22,
+                                      port_range_max=22)
+        os_utils.create_secgroup_rule(neutron_client, sg['id'],
+                                      'ingress', 'tcp',
+                                      port_range_min=80,
+                                      port_range_max=80)
+        os_utils.create_secgroup_rule(neutron_client, sg['id'],
+                                      'egress', 'tcp',
+                                      port_range_min=80,
+                                      port_range_max=80)
+
+    _, custom_flv_id = os_utils.get_or_create_flavor(
+        'custom', 1500, 10, 1, public=True)
+    if not custom_flv_id:
+        logger.error("Failed to create custom flavor")
+        sys.exit(1)
+
+    # boot INSTANCE
     logger.info("Creating instance '%s'..." % INSTANCE_NAME)
     logger.debug(
         "Configuration:\n name=%s \n flavor=%s \n image=%s \n "
@@ -118,7 +174,8 @@ def main():
     instance = os_utils.create_instance_and_wait_for_active(FLAVOR,
                                                             image_id,
                                                             network_id,
-                                                            INSTANCE_NAME)
+                                                            INSTANCE_NAME,
+                                                            av_zone='nova')
 
     if instance is None:
         logger.error("Error while booting instance.")
@@ -151,14 +208,15 @@ def main():
 # STARTING SECOND VM (server) ###
 
     # boot INTANCE
-    logger.info("Creating instance '%s'..." % INSTANCE_NAME)
+    logger.info("Creating instance '%s'..." % INSTANCE_NAME_2)
     logger.debug(
         "Configuration:\n name=%s \n flavor=%s \n image=%s \n "
-        "network=%s \n" % (INSTANCE_NAME, FLAVOR, image_id, network_id))
+        "network=%s \n" % (INSTANCE_NAME_2, FLAVOR, image_id, network_id))
     instance_2 = os_utils.create_instance_and_wait_for_active(FLAVOR,
                                                               image_id,
                                                               network_id,
-                                                              INSTANCE_NAME_2)
+                                                              INSTANCE_NAME_2,
+                                                              av_zone='nova')
 
     if instance_2 is None:
         logger.error("Error while booting instance.")
@@ -193,8 +251,8 @@ def main():
 
     # CREATION OF THE 2 SF ####
 
-    tacker_script = "/home/opnfv/repos/functest/testcases/features/sfc/" + \
-        TACKER_SCRIPT
+    tacker_script = "%s/testcases/features/sfc/%s" % \
+                    (FUNCTEST_REPO, TACKER_SCRIPT)
     logger.info("Executing tacker script: '%s'" % tacker_script)
     subprocess.call(tacker_script, shell=True)
 
@@ -228,131 +286,210 @@ def main():
     except:
         logger.debug("Problems assigning floating IP to SFs")
 
+    # If no IPs were obtained, then we cant continue
+    if not ips:
+        logger.error('Failed to obtain IPs, cant continue, exiting')
+        return
+
     logger.debug("Floating IPs for SFs: %s..." % ips)
+
+    # Check SSH connectivity to VNFs
+    r = 0
+    retries = 100
+    check = [False, False]
+
+    logger.info("Checking SSH connectivity to the SFs with ips %s" % str(ips))
+    while r < retries and not all(check):
+        try:
+            check = [check_ssh(ips[0]), check_ssh(ips[1])]
+        except Exception:
+            logger.exception("SSH check failed")
+            check = [False, False]
+        time.sleep(3)
+        r += 1
+
+    if not all(check):
+        logger.error("Cannot establish SSH connection to the SFs")
+        sys.exit(1)
+
+    logger.info("SSH connectivity to the SFs established")
+
     # SSH TO START THE VXLAN_TOOL ON SF1
     logger.info("Configuring the SFs")
     try:
         ssh.connect(ips[0], username="root",
                     password="opnfv", timeout=2)
         command = ("nohup python vxlan_tool.py -i eth0 "
-                   "-d forward -v off -f -b 80 &")
+                   "-d forward -v off -b 80 > /dev/null 2>&1 &")
         (stdin, stdout, stderr) = ssh.exec_command(command)
     except:
         logger.debug("Waiting for %s..." % ips[0])
         time.sleep(6)
         # timeout -= 1
 
+    try:
+        while 1:
+            (stdin, stdout, stderr) = ssh.exec_command(
+                "ps aux | grep \"vxlan_tool.py\" | grep -v grep")
+            if len(stdout.readlines()) > 0:
+                logger.debug("HTTP firewall started")
+                break
+            else:
+                logger.debug("HTTP firewall not started")
+                time.sleep(3)
+    except Exception:
+        logger.exception("vxlan_tool not started in SF1")
+
     # SSH TO START THE VXLAN_TOOL ON SF2
     try:
         ssh.connect(ips[1], username="root",
                     password="opnfv", timeout=2)
         command = ("nohup python vxlan_tool.py -i eth0 "
-                   "-d forward -v off -f -b 22 &")
+                   "-d forward -v off -b 22 > /dev/null 2>&1 &")
         (stdin, stdout, stderr) = ssh.exec_command(command)
     except:
         logger.debug("Waiting for %s..." % ips[1])
         time.sleep(6)
         # timeout -= 1
 
-    # SSH TO EXECUTE cmd_client
+    try:
+        while 1:
+            (stdin, stdout, stderr) = ssh.exec_command(
+                "ps aux | grep \"vxlan_tool.py\" | grep -v grep")
+            if len(stdout.readlines()) > 0:
+                logger.debug("SSH firewall started")
+                break
+            else:
+                logger.debug("SSH firewall not started")
+                time.sleep(3)
+    except Exception:
+        logger.exception("vxlan_tool not started in SF2")
+
+    i = 0
 
+    # SSH TO EXECUTE cmd_client
     logger.info("TEST STARTED")
     try:
         ssh.connect(floatip_client, username="root",
                     password="opnfv", timeout=2)
-        command = "nc -w 5 -zv " + floatip_server + " 22 2>&1"
+        command = "nc -w 5 -zv " + instance_ip_2 + " 22 2>&1"
         (stdin, stdout, stderr) = ssh.exec_command(command)
+
+        # WRITE THE CORRECT WAY TO DO LOGGING
+        if "timed out" in stdout.readlines()[0]:
+            logger.info('\033[92m' + "TEST 1 [PASSED] "
+                        "==> SSH BLOCKED" + '\033[0m')
+            i = i + 1
+            json_results.update({"Test 1: SSH Blocked": "Passed"})
+        else:
+            logger.error('\033[91m' + "TEST 1 [FAILED] "
+                         "==> SSH NOT BLOCKED" + '\033[0m')
+            status = "FAIL"
+            json_results.update({"Test 1: SSH Blocked": "Failed"})
+            failures += 1
     except:
         logger.debug("Waiting for %s..." % floatip_client)
         time.sleep(6)
         # timeout -= 1
 
-    # WRITE THE CORRECT WAY TO DO LOGGING
-    i = 0
-    if "timed out" in stdout.readlines()[0]:
-        logger.info('\033[92m' + "TEST 1 [PASSED] "
-                    "==> SSH BLOCKED" + '\033[0m')
-        i = i + 1
-    else:
-        logger.error('\033[91m' + "TEST 1 [FAILED] "
-                     "==> SSH NOT BLOCKED" + '\033[0m')
-        return
-
     # SSH TO EXECUTE cmd_client
-
     try:
         ssh.connect(floatip_client, username="root",
                     password="opnfv", timeout=2)
-        command = "nc -w 5 -zv " + floatip_server + " 80 2>&1"
+        command = "nc -w 5 -zv " + instance_ip_2 + " 80 2>&1"
         (stdin, stdout, stderr) = ssh.exec_command(command)
+
+        if "succeeded" in stdout.readlines()[0]:
+            logger.info('\033[92m' + "TEST 2 [PASSED] "
+                        "==> HTTP WORKS" + '\033[0m')
+            i = i + 1
+            json_results.update({"Test 2: HTTP works": "Passed"})
+        else:
+            logger.error('\033[91m' + "TEST 2 [FAILED] "
+                         "==> HTTP BLOCKED" + '\033[0m')
+            status = "FAIL"
+            json_results.update({"Test 2: HTTP works": "Failed"})
+            failures += 1
     except:
         logger.debug("Waiting for %s..." % floatip_client)
         time.sleep(6)
         # timeout -= 1
 
-    if "succeeded" in stdout.readlines()[0]:
-        logger.info('\033[92m' + "TEST 2 [PASSED] "
-                    "==> HTTP WORKS" + '\033[0m')
-        i = i + 1
-    else:
-        logger.error('\033[91m' + "TEST 2 [FAILED] "
-                     "==> HTTP BLOCKED" + '\033[0m')
-        return
-
     # CHANGE OF CLASSIFICATION #
     logger.info("Changing the classification")
-    tacker_classi = "/home/opnfv/repos/functest/testcases/features/sfc/" + \
-        TACKER_CHANGECLASSI
+    tacker_classi = "%s/testcases/features/sfc/%s" % \
+                    (FUNCTEST_REPO, TACKER_CHANGECLASSI)
     subprocess.call(tacker_classi, shell=True)
 
+    logger.info("Wait for ODL to update the classification rules in OVS")
+    time.sleep(100)
+
     # SSH TO EXECUTE cmd_client
 
     try:
         ssh.connect(floatip_client, username="root",
                     password="opnfv", timeout=2)
-        command = "nc -w 5 -zv " + floatip_server + " 80 2>&1"
+        command = "nc -w 5 -zv " + instance_ip_2 + " 80 2>&1"
         (stdin, stdout, stderr) = ssh.exec_command(command)
+
+        if "timed out" in stdout.readlines()[0]:
+            logger.info('\033[92m' + "TEST 3 [PASSED] "
+                        "==> HTTP BLOCKED" + '\033[0m')
+            i = i + 1
+            json_results.update({"Test 3: HTTP Blocked": "Passed"})
+        else:
+            logger.error('\033[91m' + "TEST 3 [FAILED] "
+                         "==> HTTP NOT BLOCKED" + '\033[0m')
+            status = "FAIL"
+            json_results.update({"Test 3: HTTP Blocked": "Failed"})
+            failures += 1
     except:
         logger.debug("Waiting for %s..." % floatip_client)
         time.sleep(6)
         # timeout -= 1
 
-    if "timed out" in stdout.readlines()[0]:
-        logger.info('\033[92m' + "TEST 3 [WORKS] "
-                    "==> HTTP BLOCKED" + '\033[0m')
-        i = i + 1
-    else:
-        logger.error('\033[91m' + "TEST 3 [FAILED] "
-                     "==> HTTP NOT BLOCKED" + '\033[0m')
-        return
-
     # SSH TO EXECUTE cmd_client
-
     try:
         ssh.connect(floatip_client, username="root",
                     password="opnfv", timeout=2)
-        command = "nc -w 5 -zv " + floatip_server + " 22 2>&1"
+        command = "nc -w 5 -zv " + instance_ip_2 + " 22 2>&1"
         (stdin, stdout, stderr) = ssh.exec_command(command)
+
+        if "succeeded" in stdout.readlines()[0]:
+            logger.info('\033[92m' + "TEST 4 [PASSED] "
+                        "==> SSH WORKS" + '\033[0m')
+            i = i + 1
+            json_results.update({"Test 4: SSH works": "Passed"})
+        else:
+            logger.error('\033[91m' + "TEST 4 [FAILED] "
+                         "==> SSH BLOCKED" + '\033[0m')
+            status = "FAIL"
+            json_results.update({"Test 4: SSH works": "Failed"})
+            failures += 1
     except:
         logger.debug("Waiting for %s..." % floatip_client)
         time.sleep(6)
         # timeout -= 1
 
-    if "succeeded" in stdout.readlines()[0]:
-        logger.info('\033[92m' + "TEST 4 [WORKS] "
-                    "==> SSH WORKS" + '\033[0m')
-        i = i + 1
-    else:
-        logger.error('\033[91m' + "TEST 4 [FAILED] "
-                     "==> SSH BLOCKED" + '\033[0m')
-        return
-
     if i == 4:
         for x in range(0, 5):
             logger.info('\033[92m' + "SFC TEST WORKED"
                         " :) \n" + '\033[0m')
 
-    sys.exit(0)
+    if args.report:
+        stop_time = time.time()
+        json_results.update({"tests": "4", "failures": int(failures)})
+        logger.debug("Promise Results json: " + str(json_results))
+        ft_utils.push_results_to_db("sfc",
+                                    "functest-odl-sfc",
+                                    start_time,
+                                    stop_time,
+                                    status,
+                                    json_results)
+    if status == "PASS":
+        sys.exit(0)
+    else:
+        sys.exit(1)
 
 if __name__ == '__main__':
     main()