def add_irc_callback(self, callback, func): '''Register a callback for an irc event. Callbacks are registered as irc_CALLBACK. ''' callback = numeric_to_symbolic.get(callback, callback) setattr(self.bot, "irc_%s" % callback, func)
def test_add_nonexisting_callback(self): def names(self, *args): pass self.assertRaises(AttributeError, getattr, self.bot, 'irc_RPL_NAMREPLY') self.disp.add_irc_callback(numeric_to_symbolic.get(RPL_NAMREPLY), names) self.assertEqual(self.bot.irc_RPL_NAMREPLY, names) # make sure to clean up del self.bot.irc_RPL_NAMREPLY