def test_submit_february_2020_data_loss_ultimate_already_got_ultimate(
self, messageUserMock, taskMock):
request = RequestFactory().get("/")
email = BroadcastEmail.objects.create(subject="test")
admin = BroadcastEmailAdmin(model=BroadcastEmail,
admin_site=AdminSite())
user = User.objects.create(username="******", password="******")
premium = UserSubscription.objects.create(
user=user,
subscription=Subscription.objects.create(
name="AstroBin Premium",
price=1,
group=Group.objects.create(name="astrobin_premium")),
expires=date(2020, 2, 20))
premium.subscribe()
ultimate = UserSubscription.objects.create(
user=user,
subscription=Subscription.objects.create(
name="AstroBin Ultimate 2020+",
price=1,
group=Group.objects.create(name="astrobin_ultimate_2020")),
expires=date(2021, 3, 28))
ultimate.subscribe()
admin.submit_february_2020_data_loss_ultimate_upgrade(
request, BroadcastEmail.objects.filter(pk=email.pk))
taskMock.assert_not_called()
def test_submit_february_2020_data_loss_ultimate_upgrade_no_users(
self, taskMock):
request = RequestFactory().get("/")
email = BroadcastEmail.objects.create(subject="test")
admin = BroadcastEmailAdmin(model=BroadcastEmail,
admin_site=AdminSite())
admin.submit_february_2020_data_loss_ultimate_upgrade(
request, BroadcastEmail.objects.filter(pk=email.pk))
taskMock.assert_not_called()
def test_submit_february_2020_data_loss_premium_ultimate_too_early_expiration(
self, messageUserMock, taskMock):
request = RequestFactory().get("/")
email = BroadcastEmail.objects.create(subject="test")
admin = BroadcastEmailAdmin(model=BroadcastEmail,
admin_site=AdminSite())
UserSubscription.objects.create(
user=User.objects.create(username="******", password="******"),
subscription=Subscription.objects.create(
name="AstroBin Premium",
price=1,
group=Group.objects.create(name="astrobin_premium")),
expires=date(2020, 2, 14))
admin.submit_february_2020_data_loss_ultimate_upgrade(
request, BroadcastEmail.objects.filter(pk=email.pk))
taskMock.assert_not_called()
