vitrage/vitrage_tempest_tests/tests/utils.py

132 lines
3.7 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.
import socket
from oslo_config import cfg
from oslo_config.cfg import NoSuchOptError
from oslo_log import log as logging
from vitrage import service
import os
import oslo_messaging
import re
import subprocess
import vitrage_tempest_tests.tests
LOG = logging.getLogger(__name__)
def opts():
return [
('keystone_authtoken', vitrage_tempest_tests.tests.OPTS)
]
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()
if stderr != '':
LOG.error("The command output error is : " + stderr)
if stdout != '':
LOG.debug("The command output is : \n" + stdout)
return stdout
return None
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_conf():
conf = service.prepare_service([])
for group, options in opts():
conf.register_opts(list(options), group=group)
return conf
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')