如何在Django管理员中测试替代功能?

时间:2020-02-11 13:12:33

标签: python django django-admin python-unittest django-unittest

我在admin模块中具有替代功能

@admin.register(Donation)
class DonationAdmin(admin.ModelAdmin):
    def log_change(self, request, object, message):
        log_obj = super().log_change(request, object, message)

        name_map = {name: value.label for name, value
                    in DonationForm.base_fields.items()}

        extended_log = ''
        if message:
            changed_fields = message[0].get('changed').get('fields')
            extended_log = {key: value for key, value in request.POST.items()
                            if key in changed_fields}
            extended_log = self.get_humanize_values(extended_log)

            extended_log = dict(
                zip(
                    map(lambda x: name_map[x], extended_log.keys()),
                    extended_log.values()
                )
            )

        return AdditionalLogEntry.objects.create(
            entry=log_obj, extended_log=extended_log)

我该如何测试?

class DonationAdminTestCase(TestCase):
    def setUp(self) -> None:
        self.user = UserFactory()
        self.donation = DonationFactory()
        self.donation_admin = DonationAdmin(model=Donation, admin_site=AdminSite())

    def test_log_change(self) -> None:
        self.user.is_staff = True
        self.user.save()
        self.client.force_login(self.user)
        # ???

1 个答案:

答案 0 :(得分:1)

我会在更改视图本身上发布一个帖子,这应该触发更改记录。

change_url = reverse('admin:myapp_donation_change', args=(self.donation.id,))
data = {...}  # dictionary with all the data required to make a successful change
self.client.post(change_url, data=data)
self.assertEqual(1, AdditionalLogEntry.objects.count())
...
相关问题