Merge "rubbos testcase running entrance"

2 files changed, 160 insertions, 1 deletions

diff --git a/run_tests.sh b/run_tests.sh
index b87d2fcb..e6e90846 100755
--- a/run_tests.sh
+++ b/run_tests.sh
@@ -79,7 +79,7 @@ function run_test(){
                 #adjust config parameters, different test suite has different methods, take rubbos as an example
                 #run test case, different test suite has different methods
                 file={$BASEDIR}/testsuites/rubbos/testcase_cfg/{$i}.yaml
-                python ${BOTTLENECK_TOP_DIR}/testsuites/rubbos/run_rubbos.py $file
+                python ${BOTTLENECK_TOP_DIR}/testsuites/rubbos/run_rubbos.py -c $file
             done
         ;;
         "vstf")
diff --git a/testsuites/rubbos/run_rubbos.py b/testsuites/rubbos/run_rubbos.py
index e69de29b..d487313a 100755
--- a/testsuites/rubbos/run_rubbos.py
+++ b/testsuites/rubbos/run_rubbos.py
@@ -0,0 +1,159 @@
+#!/usr/bin/env python
+##############################################################################
+# Copyright (c) 2016 Huawei Technologies Co.,Ltd and others.
+#
+# 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 os
+import argparse
+import time
+import heatclient
+import keystoneclient
+import glanceclient
+import novaclient
+
+#------------------------------------------------------
+# parser for configuration files in each test case
+# ------------------------------------------------------
+parser = argparse.ArgumentParser()
+parser.add_argument("-c", "--conf",
+                    help="configuration files for the testcase, in yaml format",
+                    default="rubbos_1-1-1.yaml")
+args = parser.parse_args()
+
+#--------------------------------------------------
+# logging configuration
+#--------------------------------------------------
+logger = logging.getLogger('run_rubbos')
+logger.setLevel(logging.DEBUG)
+
+def _get_keystone_client():
+    keystone_client = keystoneclient.v2_0.client.Client(
+                auth_url=os.environ.get('OS_AUTH_URL'),
+                username=os.environ.get('OS_USERNAME'),
+                password=os.environ.get('OS_PASSWORD'),
+                tenant_name=os.environ.get('OS_TENANT_NAME'),
+                cacert=os.environ.get('OS_CACERT'))
+    return keystone_client
+
+def _get_heat_client():
+    keystone = _get_keystone_client()
+    heat_endpoint = keystone.service_catalog.url_for(service_type='orchestration')
+    heat_client = heatclient.client.Client('1', endpoint=heat_endpoint, token=keystone.auth_token)
+
+    return heat_client
+
+def _get_glance_client():
+    keystone = _get_keystone_client()
+    glance_endpoint = keystone.service_catalog.url_for(service_type='image',
+                                                       endpoint_type='publicURL')
+    return glanceclient.v2.client.Client(glance_endpoint, token=keystone.auth_token)
+
+def _get_nova_client():
+    nova_client = novaclient.client.Client("2", auth_url=os.environ.get('OS_AUTH_URL'),
+                username=os.environ.get('OS_USERNAME'),
+                password=os.environ.get('OS_PASSWORD'),
+                tenant_name=os.environ.get('OS_TENANT_NAME'),
+                region_name=os.environ.get('OS_REGION_NAME'),
+                cacert=os.environ.get('OS_CACERT'))
+    return nova_client
+
+def download_url(url, dest_path):
+    """
+    Download a file to a destination path given a URL
+    """
+    name = url.rsplit('/')[-1]
+    dest = dest_path + "/" + name
+    try:
+        response = urllib2.urlopen(url)
+    except (urllib2.HTTPError, urllib2.URLError):
+        return False
+
+    with open(dest, 'wb') as f:
+        shutil.copyfileobj(response, f)
+    return True
+
+def rubbos_env_prepare(template=None):
+    """ Prepare for rubbos running env
+    """
+    #logger.info("generate heat template for the testcase based on template '%s'." % template)
+    pass
+
+def rubbos_env_cleanup():
+    glance = _get_glance_client()
+    heat = _get_heat_client()
+    nova = _get_nova_client()
+
+    for stack in self.heat.stacks.list():
+        heat.stacks.delete(stack.id)
+
+    for image in self.glance.images.list():
+        glance.images.delete(image.id)
+
+    for keypair in self.nova.keypairs.list():
+        nova.keypairs.delete(keypair.id)
+
+    for flavor in self.nova.flavors.list():
+        nova.flavors.delete(flavor.id)
+
+    logger.info("openstack env cleanup")
+
+def rubbos_load_image(name=None):
+    file_url = '/tmp'
+    download_url(image_url,file_url)
+
+    glance = _get_glance_client()
+    image_args = {'name': name,
+                  'disk-format': 'qcow2',
+                  'container_format': 'bare',
+                  'file': '/tmp/bottlenecks-trusty-server.img'}
+    image = glance.images.create(**image_args)
+    if not (image.id):
+       logger.error("failed to upload rubbos image to openstack")
+       exit(-1)
+
+def rubbos_create_instance(template_file=None, rubbos_parameters=None):
+    heat = _get_heat_client()
+    template = open(template_file, 'r')
+    rubbos_stack = heat.stacks.create(stack_name='rubbos', template=template.read(), parameters=rubbos_parameters)
+    uid = rubbos_stack['stack']['id']
+
+def rubbos_stack_check(stack_name=None):
+    for stack in heat.stacks.list():
+        if stack.stack_name == stack_name:
+           return stack.stack_status
+    return 'NOT_FOUND'
+
+def rubbos_run():
+    pass
+
+def main():
+    global Heat_template
+    global image_url
+
+    image_url = 'http://artifacts.opnfv.org/bottlenecks/rubbos/bottlenecks-trusty-server.img'
+
+    if not (args.conf):
+       logger.error("configuration files are not set for testcase")
+       exit(-1)
+    else:
+       Heat_template = args.conf
+
+    parameters={'image': 'rubbos',
+                'key_name': 'rubbos-key',
+                'public_net': 'net04_ext'}
+
+    rubbos_env_prepare(Heat_template)
+    rubbos_env_cleanup()
+    rubbos_load_image(name='rubbos')
+    rubbos_create_instance(template_file=Heat_template, rubbos_paramters=parameters)
+    time.sleep(400)
+    rubbos_stack_check(stack_name='rubbos')
+    rubbos_run()
+    rubbos_env_cleanup()
+
+if __name__=='__main__':
+    main()