class MailingTestCase(MailTestCase): def get_newsletter_kwargs(self): """ Returns the keyword arguments for instanciating the newsletter. """ return { 'title': 'Test newsletter', 'slug': 'test-newsletter', 'sender': 'Test Sender', 'email': '*****@*****.**' } def get_newsletter_sites(self): return get_default_sites() def get_site(self) -> Site: return Site.objects.get_current() def setUp(self): self.n = Newsletter(**self.get_newsletter_kwargs()) self.n.save() self.n.site.set(self.get_newsletter_sites()) self.m = Message(title='Test message', newsletter=self.n, slug='test-message') self.m.save() self.a = Attachment.objects.create(file=os.path.join( 'tests', 'files', 'sample.pdf'), message=self.m) self.s = Subscription.objects.create(name='Test Name', email='*****@*****.**', newsletter=self.n, subscribed=True) self.s2 = Subscription.objects.create(name='René Luçon', email='*****@*****.**', newsletter=self.n, subscribed=True) def send_email(self, action): assert action in ACTIONS + ('message', ), 'Unknown action: %s' % action if action == 'message': # Create submission sub = Submission.from_message(self.m, Site.objects.get_current()) sub.prepared = True sub.publish_date = now() - timedelta(seconds=1) sub.save() # Send message email Submission.submit_queue() else: for subscriber in self.n.get_subscriptions(): subscriber.send_activation_email(Site.objects.get_current(), action)
class MailingTestCase(MailTestCase): def get_newsletter_kwargs(self): """ Returns the keyword arguments for instanciating the newsletter. """ return { 'title': 'Test newsletter', 'slug': 'test-newsletter', 'sender': 'Test Sender', 'email': '*****@*****.**' } def setUp(self): self.n = Newsletter(**self.get_newsletter_kwargs()) self.n.save() try: self.n.site.set(get_default_sites()) except AttributeError: # Django < 1.10 self.n.site = get_default_sites() self.m = Message(title='Test message', newsletter=self.n, slug='test-message') self.m.save() self.s = Subscription.objects.create( name='Test Name', email='*****@*****.**', newsletter=self.n, subscribed=True ) self.s2 = Subscription.objects.create( name='René Luçon', email='*****@*****.**', newsletter=self.n, subscribed=True ) def send_email(self, action): assert action in ACTIONS + ('message', ), 'Unknown action: %s' % action if action == 'message': # Create submission sub = Submission.from_message(self.m) sub.prepared = True sub.publish_date = now() - timedelta(seconds=1) sub.save() # Send message email Submission.submit_queue() else: for subscriber in self.n.get_subscriptions(): subscriber.send_activation_email(action)