def test_add_comment_signal(self):
comment = Comments.objects.filter().order_by('?')[0]
with mock_signal_receiver(snippet_add_comment) as receiver:
snippet_add_comment.send(sender=self, snippet=comment.snippet,
comment=comment)
self.assertEqual(receiver.call_count, 1)
self.assertGreater(mail.outbox, 0)
def test_follow_notification(self):
follow = Follow.objects.filter().order_by('?')[0]
with mock_signal_receiver(follow_done) as receiver:
follow_done.send(sender=self, follow=follow)
self.assertEqual(receiver.call_count, 1)
self.assertEqual(len(mail.outbox), 1)
send_mail = mail.outbox[0]
expected_to = [follow.following.email]
self.assertEqual(expected_to, send_mail.to)
def test_welcome_email(self):
user = User.objects.filter().order_by('?')[0]
with mock_signal_receiver(welcome_email) as receiver:
welcome_email.send(sender=self, user=user)
self.assertEqual(receiver.call_count, 1)
send_mail = mail.outbox[0]
self.assertEqual(len(mail.outbox), 1)
expected_to = [user.email]
self.assertEqual(expected_to, send_mail.to)
