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)
