def test_find_command_wrong_command(self):
all_chaos, _ = ChaosMonkey.get_all_chaos()
command = ChaosMonkey._find_command(all_chaos, 'foo')
self.assertEqual(command, None)
def test_find_command(self):
all_chaos, _ = ChaosMonkey.get_all_chaos()
command = ChaosMonkey._find_command(all_chaos, 'deny-all')
self.assertEqual(command.command_str, 'deny-all')
