1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
|
#!/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 logging
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()
|
