128 lines
3.6 KiB
Python
128 lines
3.6 KiB
Python
# Copyright 2016 Nokia
|
|
#
|
|
# 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.
|
|
|
|
from functools import wraps
|
|
|
|
import socket
|
|
|
|
from oslo_config import cfg
|
|
from oslo_config.cfg import NoSuchOptError
|
|
from oslo_log import log as logging
|
|
|
|
import os
|
|
import oslo_messaging
|
|
import re
|
|
import subprocess
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
def get_from_terminal(command):
|
|
proc = os.popen(command)
|
|
text_out = proc.read()
|
|
proc.close()
|
|
return text_out
|
|
|
|
|
|
def run_vitrage_command(command, conf):
|
|
# AUTH_URL
|
|
local_ip = socket.gethostbyname(socket.gethostname())
|
|
auth_url = get_property_value('OS_AUTH_URL', 'auth_url',
|
|
'http://%s:5000/v2.0' % local_ip, conf)
|
|
auth_url_param = '--os-auth-url ' + auth_url
|
|
|
|
# USERNAME
|
|
user = get_property_value('OS_USERNAME', 'username',
|
|
'admin', conf)
|
|
user_param = '--os-user-name ' + user
|
|
|
|
# PASSWORD
|
|
password = get_property_value('OS_PASSWORD', 'password',
|
|
'secretadmin', conf)
|
|
password_param = '--os-password ' + password
|
|
|
|
# PROJECT_NAME
|
|
project_name = get_property_value('OS_TENANT_NAME', 'project_name',
|
|
'admin', conf)
|
|
project_name_param = '--os-project-name ' + project_name
|
|
|
|
full_command = '%s %s %s %s %s' % (command, user_param, password_param,
|
|
project_name_param, auth_url_param)
|
|
|
|
LOG.info('Full command: %s', full_command)
|
|
|
|
p = subprocess.Popen(full_command,
|
|
shell=True,
|
|
executable="/bin/bash",
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE)
|
|
stdout, stderr = p.communicate()
|
|
return stdout
|
|
|
|
|
|
def get_property_value(environment_name, conf_name, default_value, conf):
|
|
if os.environ.get(environment_name):
|
|
return os.environ[environment_name]
|
|
|
|
try:
|
|
return conf.service_credentials[conf_name]
|
|
except NoSuchOptError:
|
|
LOG.debug("Configuration doesn't exist: service_credentials.%s",
|
|
conf_name)
|
|
|
|
return default_value
|
|
|
|
|
|
def run_from_terminal(command):
|
|
proc = os.popen(command)
|
|
proc.close()
|
|
|
|
|
|
def change_terminal_dir(path):
|
|
os.chdir(path)
|
|
LOG.debug("The path is : " + path)
|
|
|
|
|
|
def get_client():
|
|
transport = oslo_messaging.get_transport(cfg.CONF)
|
|
cfg.CONF.set_override('rpc_backend', 'rabbit')
|
|
target = oslo_messaging.Target(topic='rpcapiv1')
|
|
return oslo_messaging.RPCClient(transport, target)
|
|
|
|
|
|
def get_regex_result(pattern, text):
|
|
p = re.compile(pattern)
|
|
m = p.search(text)
|
|
if m:
|
|
LOG.debug("The regex value is " + m.group(1))
|
|
return m.group(1)
|
|
return None
|
|
|
|
|
|
def uni2str(text):
|
|
return text.encode('ascii', 'ignore')
|
|
|
|
|
|
def tempest_logger(func):
|
|
func_name = func.func_name
|
|
|
|
@wraps(func)
|
|
def func_name_print_func(*args, **kwargs):
|
|
LOG.info('Test Start: ' + func_name)
|
|
result = func(*args, **kwargs)
|
|
LOG.info('Test End: ' + func_name)
|
|
return result
|
|
|
|
return func_name_print_func
|