summaryrefslogtreecommitdiffstats
path: root/testsuites
diff options
context:
space:
mode:
Diffstat (limited to 'testsuites')
-rwxr-xr-xtestsuites/rubbos/run_rubbos.py159
1 files changed, 159 insertions, 0 deletions
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()