def test_kill_reply_y(self):
""" Kill reply y response """
ret = Kill.reply_handler(self.action, "y")
self.assertEqual(len(ret), 1)
self.assertEqual(ret[0][0], "*")
self.assertEqual(ret[0][1], "p2a has been killed")
self.assertEqual(self.p2a.state, "DEAD")
def test_kill_reply_y(self):
""" Kill reply y response """
ret = Kill.reply_handler(self.action, "y")
self.assertEqual(len(ret), 1)
self.assertEqual(ret[0][0], "*")
self.assertEqual(ret[0][1], "p2a has been killed")
self.assertEqual(self.p2a.state, "DEAD")
def test_kill_reply_n(self):
""" Kill reply n response """
ret = Kill.reply_handler(self.action, "n")
self.assertEqual(len(ret), 1)
self.assertEqual(ret[0][0], "+1")
self.assertEqual(ret[0][1], "Your victim claims that he/she was not "
"killed. Please check that you have the correct codename")
self.assertEqual(self.p2a.state, "ALIVE")
def test_kill_reply_n(self):
""" Kill reply n response """
ret = Kill.reply_handler(self.action, "n")
self.assertEqual(len(ret), 1)
self.assertEqual(ret[0][0], "+1")
self.assertEqual(
ret[0][1], "Your victim claims that he/she was not "
"killed. Please check that you have the correct codename")
self.assertEqual(self.p2a.state, "ALIVE")
def handler(cls, ref, params, From):
logging.debug("REPLY {}".format(ref))
lookup = Action.get_by_id(int(ref))
if not lookup:
raise ReplyError("ref num", ref)
response = params[0]
if response != "Y" and response != "y" and response != "N" and response != "n":
raise ReplyError(response, ref)
if lookup.action == "DISARM":
return Disarm.reply_handler(lookup, response)
else:
output_msg = []
if lookup.action == "KILL":
output_msg += Kill.reply_handler(lookup, response)
elif lookup.action == "BOMB":
output_msg += Bomb.reply_handler(lookup, response, From)
""" Generate push if necessary """
output_msg += Team.push(Team.get_by_id(From.team))
return output_msg
raise ReplyError(response, ref)
