def setUp(self):
self._mock_ssh = mock.patch.object(attacker_baremetal, 'ssh')
self.mock_ssh = self._mock_ssh.start()
self._mock_subprocess = mock.patch.object(attacker_baremetal,
'subprocess')
self.mock_subprocess = self._mock_subprocess.start()
self.addCleanup(self._stop_mocks)
self.mock_ssh.SSH.from_node().execute.return_value = (0, "running", '')
host = {
"ipmi_ip": "10.20.0.5",
"ipmi_user": "******",
"ipmi_password": "******",
"ip": "10.20.0.5",
"user": "******",
"key_filename": "/root/.ssh/id_rsa"
}
self.context = {"node1": host}
self.attacker_cfg = {
'fault_type': 'bear-metal-down',
'host': 'node1',
}
self.ins = attacker_baremetal.BaremetalAttacker(
self.attacker_cfg, self.context)
def test__attacker_baremetal_all_successful(self, mock_ssh, mock_subprocess):
mock_ssh.SSH.from_node().execute.return_value = (0, "running", '')
ins = attacker_baremetal.BaremetalAttacker(self.attacker_cfg,
self.context)
ins.setup()
ins.inject_fault()
ins.recover()
def test__attacker_baremetal_recover_successful(self):
self.attacker_cfg["jump_host"] = 'node1'
self.context["node1"]["password"] = "******"
ins = attacker_baremetal.BaremetalAttacker(self.attacker_cfg,
self.context)
ins.setup()
ins.recover()
def test__attacker_baremetal_recover_successful(self, mock_ssh, mock_subprocess):
self.attacker_cfg["jump_host"] = 'node1'
self.context["node1"]["pwd"] = "123456"
mock_ssh.SSH.from_node().execute.return_value = (0, "running", '')
ins = attacker_baremetal.BaremetalAttacker(self.attacker_cfg,
self.context)
ins.setup()
ins.recover()
def test__attacker_baremetal_check_failuer(self, mock_ssh, mock_subprocess):
mock_ssh.SSH.from_node().execute.return_value = (0, "error check", '')
ins = attacker_baremetal.BaremetalAttacker(self.attacker_cfg,
self.context)
ins.setup()
def test__attacker_baremetal_check_failuer(self, mock_ssh):
ins = attacker_baremetal.BaremetalAttacker(self.attacker_cfg,
self.context)
mock_ssh.SSH().execute.return_value = (0, "error check", '')
ins.setup()