yardstick/network_services/utils.py

   1 # Copyright (c) 2016-2017 Intel Corporation
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #      http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14 """ Helper function to get Network Service testing configuration """
  15
  16 from __future__ import absolute_import
  17 import logging
  18 import os
  19
  20 from oslo_config import cfg
  21 from oslo_config.cfg import NoSuchOptError
  22 from oslo_utils import encodeutils
  23
  24 NSB_ROOT = "/opt/nsb_bin"
  25
  26 CONF = cfg.CONF
  27 OPTS = [
  28     cfg.StrOpt('bin_path',
  29                default=NSB_ROOT,
  30                help='bin_path for VNFs location.'),
  31     cfg.StrOpt('trex_path',
  32                default=os.path.join(NSB_ROOT, 'trex/scripts'),
  33                help='trex automation lib pathh.'),
  34 ]
  35 CONF.register_opts(OPTS, group="nsb")
  36
  37
  38 def get_nsb_option(option, default=None):
  39     """return requested option for yardstick.conf"""
  40
  41     try:
  42         return CONF.nsb.__getitem__(option)
  43     except NoSuchOptError:
  44         logging.debug("Invalid key %s", option)
  45     return default
  46
  47
  48 def provision_tool(connection, tool_path):
  49     """
  50     verify if the tool path exits on the node,
  51     if not push the local binary to remote node
  52
  53     :return - Tool path
  54     """
  55     bin_path = get_nsb_option("bin_path")
  56     exit_status, stdout = connection.execute("which %s" % tool_path)[:2]
  57     if exit_status == 0:
  58         return encodeutils.safe_decode(stdout, incoming='utf-8').rstrip()
  59
  60     logging.warning("%s not found on %s, will try to copy from localhost",
  61                     tool_path, connection.host)
  62     connection.execute('mkdir -p "%s"' % bin_path)
  63     connection.put(tool_path, tool_path)
  64     return tool_path