def test_get_strict_handler(self): """ The Spamdb.get_handler method should return a strictly registered spam function """ sdb = Spamdb() @sdb.global_handler(CharField) def empty_global_function(): pass @sdb.strict_handler(User.username) def empty_strict_function(): pass sdb2 = Spamdb() @sdb2.strict_handler(User.username) def empty_strict_function2(): pass sdb3 = Spamdb() handler = sdb.get_handler(User, User.username.__class__, 'username') handler2 = sdb2.get_handler(User, User.username.__class__, 'username') handler3 = sdb3.get_handler(User, User.username.__class__, 'username') self.assertEquals(handler, empty_strict_function) self.assertTrue(handler is not empty_global_function) self.assertEquals(handler2, empty_strict_function2) self.assertEquals(handler3, spam_charfield)
def test_get_global_handler(self): """ The Spamdb.get_handler method should return a globally registered spam function """ sdb = Spamdb() @sdb.global_handler(CharField) def empty_function(): pass handler = sdb.get_handler(User, User.username.__class__, 'username') self.assertEquals(handler, empty_function)