tests/unit/benchmark/scenarios/availability/test_attacker_baremetal.py

   1 #!/usr/bin/env python
   2
   3 ##############################################################################
   4 # Copyright (c) 2015 Huawei Technologies Co.,Ltd and others.
   5 #
   6 # All rights reserved. This program and the accompanying materials
   7 # are made available under the terms of the Apache License, Version 2.0
   8 # which accompanies this distribution, and is available at
   9 # http://www.apache.org/licenses/LICENSE-2.0
  10 ##############################################################################
  11
  12 # Unittest for yardstick.benchmark.scenarios.availability.attacker.attacker_baremetal
  13
  14 import mock
  15 import unittest
  16
  17 from yardstick.benchmark.scenarios.availability.attacker import baseattacker
  18 from yardstick.benchmark.scenarios.availability.attacker import attacker_baremetal
  19
  20 @mock.patch('yardstick.benchmark.scenarios.availability.attacker.attacker_baremetal.subprocess')
  21 class ExecuteShellTestCase(unittest.TestCase):
  22
  23     def test__fun_execute_shell_command_successful(self, mock_subprocess):
  24         cmd = "env"
  25         mock_subprocess.check_output.return_value = (0, 'unittest')
  26         exitcode, output = attacker_baremetal._execute_shell_command(cmd)
  27         self.assertEqual(exitcode, 0)
  28
  29     def test__fun_execute_shell_command_fail_cmd_exception(self, mock_subprocess):
  30         cmd = "env"
  31         mock_subprocess.check_output.side_effect = RuntimeError
  32         exitcode, output = attacker_baremetal._execute_shell_command(cmd)
  33         self.assertEqual(exitcode, -1)
  34
  35
  36 @mock.patch('yardstick.benchmark.scenarios.availability.attacker.attacker_baremetal.ssh')
  37 class AttackerBaremetalTestCase(unittest.TestCase):
  38
  39     def setUp(self):
  40         host = {
  41             "ipmi_ip": "10.20.0.5",
  42             "ipmi_user": "root",
  43             "ipmi_pwd": "123456",
  44             "ip": "10.20.0.5",
  45             "user": "root",
  46             "key_filename": "/root/.ssh/id_rsa"
  47         }
  48         self.context = {"node1": host}
  49         self.attacker_cfg = {
  50             'fault_type': 'bear-metal-down',
  51             'host': 'node1',
  52         }
  53
  54     def test__attacker_baremetal_all_successful(self, mock_ssh):
  55
  56         ins = attacker_baremetal.BaremetalAttacker(self.attacker_cfg, self.context)
  57
  58         mock_ssh.SSH().execute.return_value = (0, "running", '')
  59         ins.setup()
  60         ins.inject_fault()
  61         ins.recover()
  62
  63     def test__attacker_baremetal_check_failuer(self, mock_ssh):
  64
  65         ins = attacker_baremetal.BaremetalAttacker(self.attacker_cfg, self.context)
  66         mock_ssh.SSH().execute.return_value = (0, "error check", '')
  67         ins.setup()
  68
  69     def test__attacker_baremetal_recover_successful(self, mock_ssh):
  70
  71         self.attacker_cfg["jump_host"] = 'node1'
  72         self.context["node1"]["pwd"] = "123456"
  73         ins = attacker_baremetal.BaremetalAttacker(self.attacker_cfg, self.context)
  74
  75         mock_ssh.SSH().execute.return_value = (0, "running", '')
  76         ins.setup()
  77         ins.recover()