Python BroadcastEmailAdmin.submit_february_2020_data_loss_premium_upgrade示例

示例#1

    def test_submit_february_2020_data_loss_premium_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_premium_upgrade(
            request, BroadcastEmail.objects.filter(pk=email.pk))

        taskMock.assert_not_called()

示例#2

    def test_submit_february_2020_data_loss_premium_upgrade_sent(
            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(2021, 2, 20))

        admin.submit_february_2020_data_loss_premium_upgrade(
            request, BroadcastEmail.objects.filter(pk=email.pk))

        taskMock.assert_called()