yardstick/tests/unit/service/test_environment.py

   1 ##############################################################################
   2 # Copyright (c) 2016 Huawei Technologies Co.,Ltd and others.
   3 #
   4 # All rights reserved. This program and the accompanying materials
   5 # are made available under the terms of the Apache License, Version 2.0
   6 # which accompanies this distribution, and is available at
   7 # http://www.apache.org/licenses/LICENSE-2.0
   8 ##############################################################################
   9
  10 import mock
  11
  12 from yardstick.common.exceptions import UnsupportedPodFormatError
  13 from yardstick.service.environment import Environment
  14 from yardstick.service.environment import AnsibleCommon
  15 from yardstick.tests.unit import base as ut_base
  16
  17
  18 class EnvironmentTestCase(ut_base.BaseUnitTestCase):
  19
  20     def test_get_sut_info(self):
  21         pod_info = {
  22             'nodes': [
  23                 {
  24                     'name': 'node1',
  25                     'host_name': 'host1',
  26                     'role': 'Controller',
  27                     'ip': '10.1.0.50',
  28                     'user': 'root',
  29                     'passward': 'root'
  30                 }
  31             ]
  32         }
  33
  34         with mock.patch.object(AnsibleCommon, 'gen_inventory_ini_dict'), \
  35                 mock.patch.object(AnsibleCommon, 'get_sut_info',
  36                                   return_value={'node1': {}}):
  37             env = Environment(pod=pod_info)
  38             env.get_sut_info()
  39
  40     def test_get_sut_info_pod_str(self):
  41         pod_info = 'nodes'
  42
  43         env = Environment(pod=pod_info)
  44         with self.assertRaises(UnsupportedPodFormatError):
  45             env.get_sut_info()