def test_bomb_reply_Y(self): """ Bomb reply Y response """ ret = Bomb.reply_handler(self.action, "Y", self.p2a) 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 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)
def test_bomb_reply_n(self): """ Bomb reply n response """ ret = Bomb.reply_handler(self.action, "n", self.p2a) self.assertEqual(len(ret), 0) self.assertEqual(self.p2a.state, "ALIVE")