def test_run_command(self):
chaos = self._get_chaos_object(Net(), 'deny-state-server')
with patch('utility.check_output', autospec=True) as mock:
with patch(
'runner.random.choice', autospec=True, return_value=chaos):
with temp_dir() as directory:
runner = Runner(directory, ChaosMonkey.factory())
runner._run_command(enablement_timeout=0)
self.assertEqual(mock.mock_calls, self._deny_port_call_list())
def test_run_command_select_restart_unit(self):
chaos = self._get_chaos_object(Kill(), Kill.restart_cmd)
with patch('utility.check_output', autospec=True) as mock:
with patch(
'runner.random.choice', autospec=True, return_value=chaos):
with patch('runner.Init', autospec=True) as ri_mock:
with temp_dir() as directory:
runner = Runner(directory, ChaosMonkey.factory())
runner._run_command(enablement_timeout=0)
self.assertEqual(mock.mock_calls, [call(['shutdown', '-r', 'now'])])
ri_mock.upstart.assert_called_once_with()