def setUp(self):
subscribers.register(SubscribersTestModel1)
subscribers.register(SubscribersTestModel2)
self.email1 = SubscribersTestModel1.objects.create(subject="Foo 1")
self.email2 = SubscribersTestModel2.objects.create(subject="Foo 2")
self.subscriber1 = Subscriber.objects.subscribe(email="*****@*****.**")
self.subscriber2 = Subscriber.objects.subscribe(email="*****@*****.**")
# Create the emails.
for email in (self.email1, self.email2):
for subscriber in (self.subscriber1, self.subscriber2):
subscribers.dispatch_email(email, subscriber)
def testRegistration(self):
# Register the model and test.
subscribers.register(SubscribersTestModel1)
self.assertTrue(subscribers.is_registered(SubscribersTestModel1))
self.assertRaises(RegistrationError, lambda: subscribers.register(SubscribersTestModel1))
self.assertTrue(SubscribersTestModel1 in subscribers.get_registered_models())
self.assertTrue(isinstance(subscribers.get_adapter(SubscribersTestModel1), subscribers.EmailAdapter))
# Unregister the model and text.
subscribers.unregister(SubscribersTestModel1)
self.assertFalse(subscribers.is_registered(SubscribersTestModel1))
self.assertRaises(RegistrationError, lambda: subscribers.unregister(SubscribersTestModel1))
self.assertTrue(SubscribersTestModel1 not in subscribers.get_registered_models())
self.assertRaises(RegistrationError, lambda: isinstance(subscribers.get_adapter(SubscribersTestModel1)))
def setUp(self):
subscribers.register(SubscribersTestModel1)
subscribers.register(SubscribersTestModel2)
self.email1 = SubscribersTestModel1.objects.create(subject="Foo 1")
self.email2 = SubscribersTestModel2.objects.create(subject="Foo 1")
self.subscriber1 = Subscriber.objects.subscribe(email="*****@*****.**")
