vHello.py

#!/usr/bin/python
# coding: utf8
#######################################################################
#
#   Copyright (c) 2015 Orange
#   valentin.boucher@orange.com
#
# 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
#
# Modifications of the original vIMS blueprint by Orange, for use in the 
# OPNFV vHello blueprint for the OPNFV Models project are: 
# Copyright 2016 AT&T Intellectual Property, Inc
#  
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#  
# http://www.apache.org/licenses/LICENSE-2.0
#  
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
########################################################################

import argparse
import datetime
import json
import os
import pprint
import requests
import subprocess
import time
import yaml

import keystoneclient.v2_0.client as ksclient
import glanceclient.client as glclient
import novaclient.client as nvclient
from neutronclient.v2_0 import client as ntclient

from helloworld import helloworld
from orchestrator import orchestrator

import functest.utils.functest_logger as ft_logger
import functest.utils.functest_utils as functest_utils
import functest.utils.openstack_utils as os_utils


pp = pprint.PrettyPrinter(indent=4)


parser = argparse.ArgumentParser()
parser.add_argument("-d", "--debug", help="Debug mode", action="store_true")
parser.add_argument("-r", "--report",
                    help="Create json result file",
                    action="store_true")
parser.add_argument("-n", "--noclean",
                    help="Don't clean the created resources for this test.",
                    action="store_true")
args = parser.parse_args()

""" logging configuration """
logger = ft_logger.Logger("vHello").getLogger()

REPO_PATH = os.environ['repos_dir'] + '/functest/'
if not os.path.exists(REPO_PATH):
    logger.error("Functest repository directory not found '%s'" % REPO_PATH)
    exit(-1)

with open(os.environ["CONFIG_FUNCTEST_YAML"]) as f:
    functest_yaml = yaml.safe_load(f)
f.close()

# Cloudify parameters
vHello_DIR = (REPO_PATH +
            functest_yaml.get("general").get("directories").get("dir_vHello"))
vHello_DATA_DIR = functest_yaml.get("general").get(
    "directories").get("dir_vHello_data") + "/"
vHello_TEST_DIR = functest_yaml.get("general").get(
    "directories").get("dir_repo_vHello_test") + "/"
DB_URL = functest_yaml.get("results").get("test_db_url")

TENANT_NAME = functest_yaml.get("vHello").get("general").get("tenant_name")
TENANT_DESCRIPTION = functest_yaml.get("vHello").get(
    "general").get("tenant_description")
IMAGES = functest_yaml.get("vHello").get("general").get("images")

CFY_MANAGER_BLUEPRINT = functest_yaml.get(
    "vHello").get("cloudify").get("blueprint")
CFY_MANAGER_REQUIERMENTS = functest_yaml.get(
    "vHello").get("cloudify").get("requierments")
CFY_INPUTS = functest_yaml.get("vHello").get("cloudify").get("inputs")

CW_BLUEPRINT = functest_yaml.get("vHello").get("helloworld").get("blueprint")
CW_DEPLOYMENT_NAME = functest_yaml.get("vHello").get(
    "helloworld").get("deployment-name")
CW_INPUTS = functest_yaml.get("vHello").get("helloworld").get("inputs")
CW_REQUIERMENTS = functest_yaml.get("vHello").get(
    "helloworld").get("requierments")

CFY_DEPLOYMENT_DURATION = 0
CW_DEPLOYMENT_DURATION = 0

TESTCASE_START_TIME = time.time()
RESULTS = {'orchestrator': {'duration': 0, 'result': ''},
           'vHello': {'duration': 0, 'result': ''},
           'sig_test': {'duration': 0, 'result': ''}}


def download_and_add_image_on_glance(glance, image_name, image_url):
    dest_path = vHello_DATA_DIR + "tmp/"
    if not os.path.exists(dest_path):
        os.makedirs(dest_path)
    file_name = image_url.rsplit('/')[-1]
    if not functest_utils.download_url(image_url, dest_path):
        logger.error("Failed to download image %s" % file_name)
        return False

    image = os_utils.create_glance_image(
        glance, image_name, dest_path + file_name)
    if not image:
        logger.error("Failed to upload image on glance")
        return False

    return image


def step_failure(step_name, error_msg):
    logger.error(error_msg)
    set_result(step_name, 0, error_msg)
    status = "FAIL"
    # in case of failure starting and stoping time are not correct
    stop_time = time.time()
    if step_name == "sig_test":
        status = "PASS"
    functest_utils.push_results_to_db("functest",
                                      "vHello",
                                      None,
                                      TESTCASE_START_TIME,
                                      stop_time,
                                      status,
                                      RESULTS)
    exit(-1)


def set_result(step_name, duration=0, result=""):
    RESULTS[step_name] = {'duration': duration, 'result': result}


def test_helloworld():
    script = "source " + vHello_DATA_DIR + "venv_cloudify/bin/activate; "
    script += "cd " + vHello_DATA_DIR + "; "
    script += "cfy status | grep -Eo \"([0-9]{1,3}\.){3}[0-9]{1,3}\""
    cmd = "/bin/bash -c '" + script + "'"

    try:
        logger.debug("Trying to get helloworld manager IP ... ")
        mgr_ip = os.popen(cmd).read()
        mgr_ip = mgr_ip.splitlines()[0]
    except:
        step_failure("sig_test", "Unable to retrieve the IP of the "
                     "cloudify manager server !")

    api_url = "http://" + mgr_ip + "/api/v2"
    dep_outputs = requests.get(api_url + "/deployments/" +
                               CW_DEPLOYMENT_NAME + "/outputs")
#    dns_ip = dep_outputs.json()['outputs']['dns_ip']
#    ellis_ip = dep_outputs.json()['outputs']['ellis_ip']

#    ellis_url = "http://" + ellis_ip + "/"
#    url = ellis_url + "accounts"

#    params = {"password": "functest",
#              "full_name": "opnfv functest user",
#              "email": "functest@opnfv.fr",
#              "signup_code": "secret"}

#    rq = requests.post(url, data=params)
#    i = 20
#    while rq.status_code != 201 and i > 0:
#        rq = requests.post(url, data=params)
#        i = i - 1
#        time.sleep(10)

#    if rq.status_code == 201:
#        url = ellis_url + "session"
#        rq = requests.post(url, data=params)
#        cookies = rq.cookies

#    url = ellis_url + "accounts/" + params['email'] + "/numbers"
#    if cookies != "":
#        rq = requests.post(url, cookies=cookies)
#        i = 24
#        while rq.status_code != 200 and i > 0:
#            rq = requests.post(url, cookies=cookies)
#            i = i - 1
#            time.sleep(25)

#    if rq.status_code != 200:
#        step_failure("sig_test", "Unable to create a number: %s"
#                     % rq.json()['reason'])

#    start_time_ts = time.time()
#    end_time_ts = start_time_ts
#     logger.info("vHello functional test Start Time:'%s'" % (
#         datetime.datetime.fromtimestamp(start_time_ts).strftime(
#             '%Y-%m-%d %H:%M:%S')))
#    nameservers = functest_utils.get_resolvconf_ns()
#    resolvconf = ""
#    for ns in nameservers:
#        resolvconf += "\nnameserver " + ns

#    if dns_ip != "":
#        script = ('echo -e "nameserver ' + dns_ip + resolvconf +
#                  '" > /etc/resolv.conf; ')
#        script += 'source /etc/profile.d/rvm.sh; '
#        script += 'cd ' + vHello_TEST_DIR + '; '
#        script += ('rake test[' + CW_INPUTS["public_domain"] +
#                   '] SIGNUP_CODE="secret"')

#        cmd = "/bin/bash -c '" + script + "'"
#        output_file = "output.txt"
#        f = open(output_file, 'w+')
#        subprocess.call(cmd, shell=True, stdout=f,
#                        stderr=subprocess.STDOUT)
#        f.close()
#        end_time_ts = time.time()
#        duration = round(end_time_ts - start_time_ts, 1)
#        logger.info("vHello functional test duration:'%s'" % duration)
#        f = open(output_file, 'r')
#        result = f.read()
#        if result != "" and logger:
#            logger.debug(result)
#
#        vHello_test_result = ""
#        try:
#            logger.debug("Trying to load test results")
#            with open(vHello_TEST_DIR + "temp.json") as f:
#                vHello_test_result = json.load(f)
#            f.close()
#        except:
#            logger.error("Unable to retrieve test results")
#
#        set_result("sig_test", duration, vHello_test_result)

        # success criteria for vHello (for Brahmaputra)
        # - orchestrator deployed
        # - VNF deployed
        # TODO use test criteria defined in config file
#        status = "FAIL"
#        try:
#            if (RESULTS['orchestrator']['duration'] > 0 and
#                    RESULTS['vHello']['duration'] > 0):
#                status = "PASS"
#        except:
#            logger.error("Unable to set test status")

#        functest_utils.push_results_to_db("functest",
#                                          "vHello",
#                                          logger,
#                                          TESTCASE_START_TIME,
#                                          end_time_ts,
#                                          status,
#                                          RESULTS)
#
#        try:
#            os.remove(vHello_TEST_DIR + "temp.json")
#        except:
#            logger.error("Deleting file failed")


def main():

    # ############### GENERAL INITIALISATION ################

    if not os.path.exists(vHello_DATA_DIR):
        os.makedirs(vHello_DATA_DIR)

    ks_creds = os_utils.get_credentials("keystone")
    nv_creds = os_utils.get_credentials("nova")
    nt_creds = os_utils.get_credentials("neutron")

    logger.info("Prepare OpenStack plateform (create tenant and user)")
    keystone = ksclient.Client(**ks_creds)

    user_id = os_utils.get_user_id(keystone, ks_creds['username'])
    if user_id == '':
        step_failure("init", "Error : Failed to get id of " +
                     ks_creds['username'])

    tenant_id = os_utils.create_tenant(
        keystone, TENANT_NAME, TENANT_DESCRIPTION)
    if tenant_id == '':
        step_failure("init", "Error : Failed to create " +
                     TENANT_NAME + " tenant")

    roles_name = ["admin", "Admin"]
    role_id = ''
    for role_name in roles_name:
        if role_id == '':
            role_id = os_utils.get_role_id(keystone, role_name)

    if role_id == '':
        logger.error("Error : Failed to get id for %s role" % role_name)

    if not os_utils.add_role_user(keystone, user_id, role_id, tenant_id):
        logger.error("Error : Failed to add %s on tenant" %
                     ks_creds['username'])

    user_id = os_utils.create_user(
        keystone, TENANT_NAME, TENANT_NAME, None, tenant_id)
    if user_id == '':
        logger.error("Error : Failed to create %s user" % TENANT_NAME)

    logger.info("Update OpenStack creds informations")
    ks_creds.update({
        "username": TENANT_NAME,
        "password": TENANT_NAME,
        "tenant_name": TENANT_NAME,
    })

    nt_creds.update({
        "tenant_name": TENANT_NAME,
    })

    nv_creds.update({
        "project_id": TENANT_NAME,
    })

    logger.info("Upload some OS images if it doesn't exist")
    glance_endpoint = keystone.service_catalog.url_for(
        service_type='image', endpoint_type='publicURL')
    glance = glclient.Client(1, glance_endpoint, token=keystone.auth_token)

    for img in IMAGES.keys():
        image_name = IMAGES[img]['image_name']
        image_url = IMAGES[img]['image_url']

        image_id = os_utils.get_image_id(glance, image_name)

        if image_id == '':
            logger.info("""%s image doesn't exist on glance repository. Try
            downloading this image and upload on glance !""" % image_name)
            image_id = download_and_add_image_on_glance(
                glance, image_name, image_url)

        if image_id == '':
            step_failure(
                "init",
                "Error : Failed to find or upload required OS "
                "image for this deployment")

    nova = nvclient.Client("2", **nv_creds)

    logger.info("Update security group quota for this tenant")
    neutron = ntclient.Client(**nt_creds)
    if not os_utils.update_sg_quota(neutron, tenant_id, 50, 100):
        step_failure(
            "init",
            "Failed to update security group quota for tenant " + TENANT_NAME)

    logger.info("Update cinder quota for this tenant")
    from cinderclient import client as cinderclient

    creds_cinder = os_utils.get_credentials("cinder")
    cinder_client = cinderclient.Client('1', creds_cinder['username'],
                                        creds_cinder['api_key'],
                                        creds_cinder['project_id'],
                                        creds_cinder['auth_url'],
                                        service_type="volume")
    if not os_utils.update_cinder_quota(cinder_client, tenant_id, 20, 10, 150):
        step_failure(
            "init", "Failed to update cinder quota for tenant " + TENANT_NAME)

    # ############### CLOUDIFY INITIALISATION ################

    cfy = orchestrator(vHello_DATA_DIR, CFY_INPUTS, logger)

    cfy.set_credentials(username=ks_creds['username'], password=ks_creds[
                        'password'], tenant_name=ks_creds['tenant_name'],
                        auth_url=ks_creds['auth_url'])

    logger.info("Collect flavor id for cloudify manager server")
    nova = nvclient.Client("2", **nv_creds)

    flavor_name = "m1.medium"
    flavor_id = os_utils.get_flavor_id(nova, flavor_name)
    for requirement in CFY_MANAGER_REQUIERMENTS:
        if requirement == 'ram_min':
            flavor_id = os_utils.get_flavor_id_by_ram_range(
                nova, CFY_MANAGER_REQUIERMENTS['ram_min'], 8196)

    if flavor_id == '':
        logger.error(
            "Failed to find %s flavor. "
            "Try with ram range default requirement !" % flavor_name)
        flavor_id = os_utils.get_flavor_id_by_ram_range(nova, 4000, 8196)

    if flavor_id == '':
        step_failure("orchestrator",
                     "Failed to find required flavor for this deployment")

    cfy.set_flavor_id(flavor_id)

    image_name = "centos_7"
    image_id = os_utils.get_image_id(glance, image_name)
    for requirement in CFY_MANAGER_REQUIERMENTS:
        if requirement == 'os_image':
            image_id = os_utils.get_image_id(
                glance, CFY_MANAGER_REQUIERMENTS['os_image'])

    if image_id == '':
        step_failure(
            "orchestrator",
            "Error : Failed to find required OS image for cloudify manager")

    cfy.set_image_id(image_id)

    ext_net = os_utils.get_external_net(neutron)
    if not ext_net:
        step_failure("orchestrator", "Failed to get external network")

    cfy.set_external_network_name(ext_net)

    ns = functest_utils.get_resolvconf_ns()
    if ns:
        cfy.set_nameservers(ns)

    logger.info("Prepare virtualenv for cloudify-cli")
    cmd = "chmod +x " + vHello_DIR + "create_venv.sh"
    functest_utils.execute_command(cmd, logger)
    time.sleep(3)
    cmd = vHello_DIR + "create_venv.sh " + vHello_DATA_DIR
    functest_utils.execute_command(cmd, logger)

    cfy.download_manager_blueprint(
        CFY_MANAGER_BLUEPRINT['url'], CFY_MANAGER_BLUEPRINT['branch'])

    # ############### CLOUDIFY DEPLOYMENT ################
    start_time_ts = time.time()
    end_time_ts = start_time_ts
    logger.info("Cloudify deployment Start Time:'%s'" % (
        datetime.datetime.fromtimestamp(start_time_ts).strftime(
            '%Y-%m-%d %H:%M:%S')))

    error = cfy.deploy_manager()
    if error:
        step_failure("orchestrator", error)

    end_time_ts = time.time()
    duration = round(end_time_ts - start_time_ts, 1)
    logger.info("Cloudify deployment duration:'%s'" % duration)
    set_result("orchestrator", duration, "")

    # ############### helloworld INITIALISATION ################

    cw = helloworld(CW_INPUTS, cfy, logger)

    logger.info("Collect flavor id for all helloworld vm")
    nova = nvclient.Client("2", **nv_creds)

    flavor_name = "m1.small"
    flavor_id = os_utils.get_flavor_id(nova, flavor_name)
    for requirement in CW_REQUIERMENTS:
        if requirement == 'ram_min':
            flavor_id = os_utils.get_flavor_id_by_ram_range(
                nova, CW_REQUIERMENTS['ram_min'], 8196)

    if flavor_id == '':
        logger.error(
            "Failed to find %s flavor. Try with ram range "
            "default requirement !" % flavor_name)
        flavor_id = os_utils.get_flavor_id_by_ram_range(nova, 4000, 8196)

    if flavor_id == '':
        step_failure(
            "vHello", "Failed to find required flavor for this deployment")

    cw.set_flavor_id(flavor_id)

    image_name = "ubuntu_14.04"
    image_id = os_utils.get_image_id(glance, image_name)
    for requirement in CW_REQUIERMENTS:
        if requirement == 'os_image':
            image_id = os_utils.get_image_id(
                glance, CW_REQUIERMENTS['os_image'])

    if image_id == '':
        step_failure(
            "vHello",
            "Error : Failed to find required OS image for cloudify manager")

    cw.set_image_id(image_id)

    ext_net = os_utils.get_external_net(neutron)
    if not ext_net:
        step_failure("vHello", "Failed to get external network")

    cw.set_external_network_name(ext_net)

    # ############### helloworld DEPLOYMENT ################

    start_time_ts = time.time()
    end_time_ts = start_time_ts
    logger.info("vHello VNF deployment Start Time:'%s'" % (
        datetime.datetime.fromtimestamp(start_time_ts).strftime(
            '%Y-%m-%d %H:%M:%S')))

    error = cw.deploy_vnf(CW_BLUEPRINT)
    if error:
        step_failure("vHello", error)

    end_time_ts = time.time()
    duration = round(end_time_ts - start_time_ts, 1)
    logger.info("vHello VNF deployment duration:'%s'" % duration)
    set_result("vHello", duration, "")

    # ############### helloworld TEST ################

#    test_helloworld()

    # ########## helloworld UNDEPLOYMENT ############

#    cw.undeploy_vnf()

    # ########### CLOUDIFY UNDEPLOYMENT #############

#    cfy.undeploy_manager()

    # ############## GENERAL CLEANUP ################
#    if args.noclean:
#        exit(0)

#    ks_creds = os_utils.get_credentials("keystone")

#    keystone = ksclient.Client(**ks_creds)

#    logger.info("Removing %s tenant .." % CFY_INPUTS['keystone_tenant_name'])
#    tenant_id = os_utils.get_tenant_id(
#        keystone, CFY_INPUTS['keystone_tenant_name'])
#    if tenant_id == '':
#        logger.error("Error : Failed to get id of %s tenant" %
#                     CFY_INPUTS['keystone_tenant_name'])
#    else:
#        if not os_utils.delete_tenant(keystone, tenant_id):
#            logger.error("Error : Failed to remove %s tenant" %
#                         CFY_INPUTS['keystone_tenant_name'])

#    logger.info("Removing %s user .." % CFY_INPUTS['keystone_username'])
#    user_id = os_utils.get_user_id(
#        keystone, CFY_INPUTS['keystone_username'])
#    if user_id == '':
#        logger.error("Error : Failed to get id of %s user" %
#                     CFY_INPUTS['keystone_username'])
#    else:
#        if not os_utils.delete_user(keystone, user_id):
#            logger.error("Error : Failed to remove %s user" %
#                         CFY_INPUTS['keystone_username'])


if __name__ == '__main__':
    main()