如何使用 unittest 在 Django 中使用 request.is_ajax() 和 request.method=='POST' 测试函数？

Question

我有下一个功能

    def edit_resolution_text_view(self, request):
        if request.method == 'POST' and request.is_ajax():
            resolution = request.POST.get('resolution')
            donation_id = request.POST.get('donation')
            result = Donation.objects.filter(pk=donation_id).\
                update(resolution_note=resolution)
            if not result:
                return JsonResponse({'message': 'Error'})
            return JsonResponse({'resolution': resolution})
        return JsonResponse({'message': 'Error'})

如何使用unittest对其进行测试

此函数位于Django modelAdmin 。

我将实例定义为self.donation_admin = DonationAdmin(model=Donation, admin_site=AdminSite())并将函数调用为self.donation_admin.edit_resolution_text_view(...)

Answer 1

最简单的方法是使用 unittest 中的Client并调用解析为您函数的端点。 is_ajax()方法只是检查 HTTP_X_REQUESTED_WITH': 'XMLHttpRequest' 标头是否在请求中。 因此，您可以将其添加到测试请求中。

class TestEditResulutionTextView(unittest.TestCase):

    def test_get(self):
        response = self.client.get('/url_to_my_view/', **{'HTTP_X_REQUESTED_WITH': 'XMLHttpRequest'})
        self.assertEqual(response.json(), {'message': 'Error'})

    def test_post(self):
        expected_resolution = # put here what you are expecting
        response = self.client.post('/url_to_my_view/', **{'HTTP_X_REQUESTED_WITH': 'XMLHttpRequest'})
        self.assertEqual(response.json(), {'resolution': expected_resolution})