def test_submit_recovered_images_notice_es_no_users(self, taskMock): request = RequestFactory().get("/") email = BroadcastEmail.objects.create(subject="test") admin = BroadcastEmailAdmin(model=BroadcastEmail, admin_site=AdminSite()) admin.submit_recovered_images_notice_es( request, BroadcastEmail.objects.filter(pk=email.pk)) taskMock.assert_not_called()
def test_submit_recovered_images_notice_es_already_sent(self, taskMock): request = RequestFactory().get("/") email = BroadcastEmail.objects.create(subject="test") admin = BroadcastEmailAdmin(model=BroadcastEmail, admin_site=AdminSite()) user = Generators.user() user.userprofile.recovered_images_notice_sent = datetime.now() user.userprofile.save(keep_deleted=True) Generators.image(user=user, recovered=datetime.now()) admin.submit_recovered_images_notice_es( request, BroadcastEmail.objects.filter(pk=email.pk)) taskMock.assert_not_called()
def test_submit_recovered_images_notice_es_sent(self, messageUserMock, taskMock): request = RequestFactory().get("/") email = BroadcastEmail.objects.create(subject="test") admin = BroadcastEmailAdmin(model=BroadcastEmail, admin_site=AdminSite()) user = Generators.user() user.userprofile.language = 'es' user.userprofile.save(keep_deleted=True) Generators.image(user=user, recovered=datetime.now()) admin.submit_recovered_images_notice_es( request, BroadcastEmail.objects.filter(pk=email.pk)) taskMock.assert_called() user.userprofile.refresh_from_db() self.assertIsNotNone(user.userprofile.recovered_images_notice_sent)