def test_submit_recovered_images_notice_pt_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_pt(
request, BroadcastEmail.objects.filter(pk=email.pk))
taskMock.assert_not_called()
def test_submit_recovered_images_notice_pt_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_pt(
request, BroadcastEmail.objects.filter(pk=email.pk))
taskMock.assert_not_called()
