django - 在 Django Oscar 中使用单元测试调用视图访问购物篮

Question

如何将在单元测试中创建的 Django Oscar 篮子附加到request对象？

# views.py

from django.contrib.auth.mixins import LoginRequiredMixin

from rest_framework.response import Response
from rest_framework.views import APIView

class BasketAPIAddView(LoginRequiredMixin, APIView):
    """
    Update basket via REST API.
    """
    def delete(self, request, format=None):
        #
        # cannot access `request.basket` here
        #
        return Response({})

 

# tests.py

from django.contrib.auth import get_user_model
from django.urls import reverse
from oscar.test.factories import create_basket
from rest_framework.test import APITestCase

User = get_user_model()

class BasketAPITests(APITestCase):
    """
    Basket view test cases.
    """
    def test_remove_basket_line(self):
        basket = create_basket()
        basket.owner = User.objects.create_user('user', password='pass')
        basket.save()

        self.client.login(username='user', password='password')
        self.client.delete(reverse('delete-basket'))

标签： djangodjango-rest-frameworkdjango-oscar

Answer 1

您可以执行以下操作：

from django.contrib.auth import get_user_model

from rest_framework import status
from rest_framework.test import APIRequestFactory
from oscar.test.factories import create_basket


User = get_user_model()

class BasketAPITests(APITestCase):

    def test_remove_basket_line(self):
        """Check the method BasketAPIAddView.delete works as should."""

        # Create the request, note the  .delete which is the operation (BasketAPIAddView.delete)
        request = APIRequestFactory().delete(reverse('<your url name>'))

        # Create the basket
        basket = create_basket()
        basket.owner = User.objects.create_user('user', password='pass')
        basket.save()

        # Attach the basket to the request obj
        request.basket =  create_basket()

        # Call the endpoint, with the proper request obj
        response = BasketAPIAddView.as_view()(request)

        # Some verifications
        ...
        self.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT)
        ...

关键是使用请求工厂并随心所欲地拥有对象（添加您需要的东西）。

有关更多详细信息，您可以查看https://www.django-rest-framework.org/api-guide/testing/