vPing.py : changed #cloud-config to inject an sh script
[functest.git] / testcases / config_functest.py
index dd9af06..7602004 100644 (file)
@@ -11,6 +11,8 @@
 import re, json, os, urllib2, argparse, logging, shutil, subprocess, yaml
 from git import Repo
 
+from neutronclient.v2_0 import client
+
 actions = ['start', 'check', 'clean']
 
 
@@ -52,13 +54,15 @@ if not os.path.exists(dest):
     logger.info("functest.yaml stored in %s" % dest)
 else:
     logger.info("functest.yaml found in %s" % dest)
-    
-with open('functest.yaml') as f:
+
+
+with open('./functest.yaml') as f:
     functest_yaml = yaml.safe_load(f)
 f.close()
 
 
 """ global variables """
+# Directories
 HOME = os.environ['HOME']+"/"
 FUNCTEST_BASE_DIR = HOME + functest_yaml.get("general").get("directories").get("dir_functest")
 RALLY_REPO_DIR = HOME + functest_yaml.get("general").get("directories").get("dir_rally_repo")
@@ -67,33 +71,53 @@ RALLY_INSTALLATION_DIR = HOME + functest_yaml.get("general").get("directories").
 BENCH_TESTS_DIR = HOME + functest_yaml.get("general").get("directories").get("dir_rally_scn")
 VPING_DIR = HOME + functest_yaml.get("general").get("directories").get("dir_vping")
 ODL_DIR = HOME + functest_yaml.get("general").get("directories").get("dir_odl")
+
+# NEUTRON Public Network parameters
+NEUTRON_PUBLIC_NET_NAME = functest_yaml.get("general").get("openstack").get("neutron_public_net_name")
+NEUTRON_PUBLIC_SUBNET_NAME = functest_yaml.get("general").get("openstack").get("neutron_public_subnet_name")
+NEUTRON_PUBLIC_SUBNET_CIDR = functest_yaml.get("general").get("openstack").get("neutron_public_subnet_cidr")
+NEUTRON_PUBLIC_SUBNET_START = functest_yaml.get("general").get("openstack").get("neutron_public_subnet_start")
+NEUTRON_PUBLIC_SUBNET_END = functest_yaml.get("general").get("openstack").get("neutron_public_subnet_end")
+
+# NEUTRON Private Network parameters
+NEUTRON_PRIVATE_NET_NAME = functest_yaml.get("general").get("openstack").get("neutron_private_net_name")
+NEUTRON_PRIVATE_SUBNET_NAME = functest_yaml.get("general").get("openstack").get("neutron_private_subnet_name")
+NEUTRON_PRIVATE_SUBNET_CIDR = functest_yaml.get("general").get("openstack").get("neutron_private_subnet_cidr")
+ROUTER_NAME = functest_yaml.get("general").get("openstack").get("neutron_router_name")
+
+#GLANCE image parameters
 IMAGE_URL = functest_yaml.get("general").get("openstack").get("image_url")
 IMAGE_DISK_FORMAT = functest_yaml.get("general").get("openstack").get("image_disk_format")
 IMAGE_NAME = functest_yaml.get("general").get("openstack").get("image_name")
 IMAGE_FILE_NAME = IMAGE_URL.rsplit('/')[-1]
 IMAGE_DOWNLOAD_PATH = FUNCTEST_BASE_DIR + IMAGE_FILE_NAME
 
-
-
+credentials = None
+neutron_client = None
 
 def config_functest_start():
     """
     Start the functest environment installation
     """
+
     if config_functest_check():
         logger.info("Functest environment already installed in %s. Nothing to do." %FUNCTEST_BASE_DIR)
         exit(0)
-    elif not check_internet_connectivity():
+
+    if not check_internet_connectivity():
         logger.error("There is no Internet connectivity. Please check the network configuration.")
         exit(-1)
     elif not check_credentials():
         logger.error("Please source the openrc credentials and run the script again.")
         #TODO: source the credentials in this script
         exit(-1)
+
     else:
+        # Clean in case there are left overs
+
         config_functest_clean()
 
-        logger.info("Starting installationg of functest environment in %s" %FUNCTEST_BASE_DIR)
+        logger.info("Starting installationg of functest environment in %s" % FUNCTEST_BASE_DIR)
         os.makedirs(FUNCTEST_BASE_DIR)
         if not os.path.exists(FUNCTEST_BASE_DIR):
             logger.error("There has been a problem while creating the environment directory.")
@@ -117,6 +141,34 @@ def config_functest_start():
             config_functest_clean()
             exit(-1)
 
+        credentials = get_credentials()
+        neutron_client = client.Client(**credentials)
+
+        logger.info("Configuring Neutron...")
+        logger.info("Checking if public network '%s' exists..." % NEUTRON_PUBLIC_NET_NAME)
+        #Now: if exists we don't create it again (the clean command does not clean the neutron networks yet)
+        if check_neutron_net(neutron_client, NEUTRON_PUBLIC_NET_NAME):
+            logger.info("Public network '%s' found. No need to create another one." % NEUTRON_PUBLIC_NET_NAME)
+        else:
+            logger.info("Public network '%s' not found. Creating..." % NEUTRON_PUBLIC_NET_NAME)
+            if not create_public_neutron_net(neutron_client):
+                logger.error("There has been a problem while creating the Neutron network.")
+                #config_functest_clean()
+                exit(-1)
+
+
+        logger.info("Checking if private network '%s' exists..." % NEUTRON_PRIVATE_NET_NAME)
+        #Now: if exists we don't create it again (the clean command does not clean the neutron networks yet)
+        if check_neutron_net(neutron_client, NEUTRON_PRIVATE_NET_NAME):
+            logger.info("Private network '%s' found. No need to create another one." % NEUTRON_PRIVATE_NET_NAME)
+        else:
+            logger.info("Private network '%s' not found. Creating..." % NEUTRON_PRIVATE_NET_NAME)
+            if not create_private_neutron_net(neutron_client):
+                logger.error("There has been a problem while creating the Neutron network.")
+                #config_functest_clean()
+                exit(-1)
+
+
         logger.info("Donwloading image...")
         if not download_url_with_progress(IMAGE_URL, FUNCTEST_BASE_DIR):
             logger.error("There has been a problem while downloading the image.")
@@ -137,27 +189,67 @@ def config_functest_check():
     """
     Check if the functest environment is properly installed
     """
+    errors_all = False
+
     logger.info("Checking current functest configuration...")
+    credentials = get_credentials()
+    neutron_client = client.Client(**credentials)
 
     logger.debug("Checking directories...")
+    errors = False
     dirs = [FUNCTEST_BASE_DIR, RALLY_INSTALLATION_DIR, RALLY_REPO_DIR, RALLY_TEST_DIR, BENCH_TESTS_DIR, VPING_DIR, ODL_DIR]
     for dir in dirs:
         if not os.path.exists(dir):
-            logger.debug("The directory %s does not exist." %dir)
-            return False
-        logger.debug("   %s found" % dir)
+            logger.debug("The directory '%s' does NOT exist." % dir)
+            errors = True
+            errors_all = True
+        else:
+            logger.debug("   %s found" % dir)
+    if not errors:
+        logger.debug("...OK")
+    else:
+        logger.debug("...FAIL")
+
 
-    logger.debug("...OK")
     logger.debug("Checking Rally deployment...")
     if not check_rally():
-        logger.debug("Rally deployment not found.")
-        return False
-    logger.debug("...OK")
+        logger.debug("Rally deployment NOT found.")
+        errors_all = True
+        logger.debug("...FAIL")
+    else:
+        logger.debug("...OK")
+
+
+    logger.debug("Checking Neutron...")
+    errors = False
+    if not check_neutron_net(neutron_client, NEUTRON_PRIVATE_NET_NAME):
+        logger.debug("   Private network '%s' NOT found." % NEUTRON_PRIVATE_NET_NAME)
+        errors = True
+        errors_all = True
+    else:
+        logger.debug("   Private network '%s' found." % NEUTRON_PRIVATE_NET_NAME)
+
+    if not check_neutron_net(neutron_client, NEUTRON_PUBLIC_NET_NAME):
+        logger.debug("   Public network '%s' NOT found." % NEUTRON_PUBLIC_NET_NAME)
+        errors = True
+        errors_all = True
+    else:
+        logger.debug("   Public network '%s' found." % NEUTRON_PUBLIC_NET_NAME)
+
+    if not errors:
+        logger.debug("...OK")
+    else:
+        logger.debug("...FAIL")
+
 
     logger.debug("Checking Image...")
+    errors = False
     if not os.path.isfile(IMAGE_DOWNLOAD_PATH):
-        return False
-    logger.debug("   Image file found in %s" %IMAGE_DOWNLOAD_PATH)
+        logger.debug("   Image file '%s' NOT found." % IMAGE_DOWNLOAD_PATH)
+        errors = True
+        errors_all = True
+    else:
+        logger.debug("   Image file found in %s" % IMAGE_DOWNLOAD_PATH)
 
     cmd="glance image-list | grep " + IMAGE_NAME
     FNULL = open(os.devnull, 'w');
@@ -166,14 +258,22 @@ def config_functest_check():
     #if the command does not exist or there is no glance image
     line = p.stdout.readline()
     if line == "":
-        logger.debug("   Glance image not found")
-        return False
-    logger.debug("   Glance image found")
-    logger.debug("...OK")
+        logger.debug("   Glance image NOT found.")
+        errors = True
+        errors_all = True
+    else:
+        logger.debug("   Glance image found.")
 
-    #TODO: check OLD environment setup
+    if not errors:
+        logger.debug("...OK")
+    else:
+        logger.debug("...FAIL")
 
-    return True
+    #TODO: check OLD environment setup
+    if errors_all:
+        return False
+    else:
+        return True
 
 
 
@@ -192,6 +292,10 @@ def config_functest_clean():
         cmd = "sudo rm -rf " + FUNCTEST_BASE_DIR #need to be sudo, not possible with rmtree
         execute_command(cmd)
 
+    #logger.debug("Deleting Neutron network %s" % NEUTRON_PRIVATE_NET_NAME)
+    #if not delete_neutron_net() :
+    #    logger.error("Error deleting the network. Remove it manually.")
+
     logger.debug("Deleting glance images")
     cmd = "glance image-list | grep "+IMAGE_NAME+" | cut -c3-38"
     p = os.popen(cmd,"r")
@@ -204,6 +308,9 @@ def config_functest_clean():
     return True
 
 
+
+
+
 def install_rally():
     if check_rally():
         logger.info("Rally is already installed.")
@@ -298,15 +405,33 @@ def check_credentials():
     return True
 
 
+def get_credentials():
+    d = {}
+    d['username'] = os.environ['OS_USERNAME']
+    d['password'] = os.environ['OS_PASSWORD']
+    d['auth_url'] = os.environ['OS_AUTH_URL']
+    d['tenant_name'] = os.environ['OS_TENANT_NAME']
+    return d
+
+def get_nova_credentials():
+    d = {}
+    d['username'] = os.environ['OS_USERNAME']
+    d['api_key'] = os.environ['OS_PASSWORD']
+    d['auth_url'] = os.environ['OS_AUTH_URL']
+    d['project_id'] = os.environ['OS_TENANT_NAME']
+    return d
+
+
 def download_tests():
     os.makedirs(VPING_DIR)
     os.makedirs(ODL_DIR)
     os.makedirs(BENCH_TESTS_DIR)
 
-    logger.info("Downloading functest.yaml...")
-    yaml_url = 'https://git.opnfv.org/cgit/functest/plain/testcases/functest.yaml'
-    if not download_url(yaml_url,FUNCTEST_BASE_DIR):
-        logger.error("Unable to download the configuration file functest.yaml")
+    logger.info("Copying functest.yaml to functest environment...")
+    try:
+        shutil.copy("./functest.yaml", FUNCTEST_BASE_DIR+"functest.yaml")
+    except:
+        print "Error copying the file:", sys.exc_info()[0]
         return False
 
     logger.info("Downloading vPing test...")
@@ -339,6 +464,106 @@ def download_tests():
 
     return True
 
+def create_public_neutron_net(neutron):
+    try:
+        neutron.format = 'json'
+        logger.debug('Creating Neutron network %s...' % NEUTRON_PUBLIC_NET_NAME)
+        json_body = {'network': {'name': NEUTRON_PUBLIC_NET_NAME, 'admin_state_up': True, 'router:external': True}}
+        netw = neutron.create_network(body=json_body)
+        net_dict = netw['network']
+        network_id = net_dict['id']
+        logger.debug("Network '%s' created successfully" % network_id)
+
+        logger.debug('Creating Subnet....')
+        json_body = {'subnets': [{'cidr': NEUTRON_PUBLIC_SUBNET_CIDR,
+                           'ip_version': 4, 'network_id': network_id}]}
+        subnet = neutron.create_subnet(body=json_body)
+        logger.debug("Subnet '%s' created successfully" % subnet)
+    except:
+        print "Error:", sys.exc_info()[0]
+        return False
+
+    logger.info("Public Neutron network created successfully.")
+    return True
+
+def create_private_neutron_net(neutron):
+    try:
+        neutron.format = 'json'
+        logger.debug('Creating Neutron network %s...' % NEUTRON_PRIVATE_NET_NAME)
+        json_body = {'network': {'name': NEUTRON_PRIVATE_NET_NAME,
+                    'admin_state_up': True}}
+        netw = neutron.create_network(body=json_body)
+        net_dict = netw['network']
+        network_id = net_dict['id']
+        logger.debug("Network '%s' created successfully" % network_id)
+
+        logger.debug('Creating Subnet....')
+        json_body = {'subnets': [{'cidr': NEUTRON_PRIVATE_SUBNET_CIDR,
+                           'ip_version': 4, 'network_id': network_id}]}
+        subnet = neutron.create_subnet(body=json_body)
+        logger.debug("Subnet '%s' created successfully" % subnet)
+
+        logger.debug('Creating Router...')
+        json_body = {'router': {'name': ROUTER_NAME, 'admin_state_up': True}}
+        router = neutron.create_router(json_body)
+        logger.debug("Router '%s' created successfully" % router)
+        router_id = router['router']['id']
+
+        logger.debug('Creating Port')
+        json_body = {'port': {
+         'admin_state_up': True,
+         'device_id': router_id,
+         'name': 'port1',
+         'network_id': network_id,
+        }}
+        response = neutron.create_port(body=json_body)
+        logger.debug("Port created successfully.")
+
+        logger.debug('Setting up gateway...')
+        public_network_id = get_network_id(neutron,NEUTRON_PUBLIC_NET_NAME)
+        json_body = {'network_id': public_network_id, 'enable_snat' :  True}
+        gateway = neutron.add_gateway_router(router_id,body=json_body)
+        logger.debug("Gateway '%s' added successfully" % gateway)
+    except:
+        print "Error:", sys.exc_info()[0]
+        return False
+
+    logger.info("Private Neutron network created successfully.")
+    return True
+
+def get_network_id(neutron, network_name):
+    networks = neutron.list_networks()['networks']
+    id  = ''
+    for n in networks:
+        if n['name'] == network_name:
+            id = n['id']
+            break
+    return id
+
+def check_neutron_net(neutron, net_name):
+    for network in neutron.list_networks()['networks']:
+        if network['name'] == net_name :
+            for subnet in network['subnets']:
+                return True
+    return False
+
+def delete_neutron_net(neutron):
+    #TODO: remove router, ports
+    try:
+        #https://github.com/isginf/openstack_tools/blob/master/openstack_remove_tenant.py
+        for network in neutron.list_networks()['networks']:
+            if network['name'] == NEUTRON_PRIVATE_NET_NAME :
+                for subnet in network['subnets']:
+                    print "Deleting subnet " + subnet
+                    neutron.delete_subnet(subnet)
+                print "Deleting network " + network['name']
+                neutron.delete_neutron_net(network['id'])
+    finally:
+        return True
+    return False
+
+
+
 
 
 def create_glance_image(path,name,disk_format):
@@ -350,6 +575,9 @@ def create_glance_image(path,name,disk_format):
     return True
 
 
+
+
+
 def download_url(url, dest_path):
     """
     Download a file to a destination path given a URL