无法使用 TestCase 库对 Django APIviews 进行单元测试？答案

【问题标题】：Unable to unittest Django APIviews with TestCase library?无法使用 TestCase 库对 Django APIviews 进行单元测试？
【发布时间】：2018-07-10 00:17:23
【问题描述】：

这是一个非常具体的项目问题，但我无法弄清楚我做错了什么。我有一个视图类，允许用户对项目数据库执行 GET 和 POST 请求。我能够使用 curl 成功地 GET/POST，但是当我为此编写单元测试时，我无法让它通过测试，而且我不完全明白为什么。

此命令有效：

curl http://127.0.0.1:8000/list/categories/ -d '{"id":"00001", "name":"some info"}'

view.py：

class CategoryList(APIView):

    def get(self, request, format=None):
        categories = Category.objects.all()
        serializer = CategorySerializer(categories, many=True)
        context = {
            'categories': categories
        }
        return render(request, 'myapp/categories.html', context)


    def post(self, request, format=None):
        print('=======inside post=========')
        data = JSONParser().parse(request)
        serializer = CategorySerializer(data=data)
        if serializer.is_valid():
            serializer.save()
            return Response(serializer.data, status=status.HTTP_201_CREATED)
        return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

urls.py:

urlpatterns = [
# eg: /list/categories/
url(r'^categories/$', CategoryList.as_view()),
]

这是我的测试失败的地方。

tests.py：

class CategoryViewTests(TestCase):
    def setUp(self):
        """ Mock model data """
        cat1 = Category.objects.create(id=55555, name="Shoes")
        cat2 = Category.objects.create(id=12345, name="Wallets")

    def test_new_unique_category(self):
        """
        Create a new category obj
        """
        c = Client()
        print(str(Category.objects.all()))
        new_cat_data = {'id':'54211', 'name':'Soccer Balls'}
        response = c.post('/list/categories', new_cat_data)

        print(str(Category.objects.all()))

        self.assertEqual(response.status_code, 201)

终端输出：

test_new_unique_category (inferbrand.tests.CategoryViewTests) ... 
<QuerySet [<Category: Test Men's Shoes>, <Category: Test Wallets>]>
<QuerySet [<Category: Test Men's Shoes>, <Category: Test Wallets>]>
FAIL


======================================================================
FAIL: test_new_unique_category (inferbrand.tests.CategoryViewTests)
----------------------------------------------------------------------
Traceback (most recent call last):
File "path/to/proj/tests.py", line 104, in test_new_unique_category
self.assertEqual(response.status_code, 201)
AssertionError: 301 != 201

我做错了什么？？任何帮助将不胜感激。

【问题讨论】：

标签： django-views django-testing

【解决方案1】：

curl 命令有效，因为您的 URL 中有斜杠，但在您的单元测试中，该 URL 没有斜杠，因此 Django 将其重定向到带有状态码 301 的斜杠的 URL，即默认行为，除非您在 settings.py 中将 APPEND_SLASH 设置为 False。

详情请咨询https://docs.djangoproject.com/en/2.0/ref/settings/#append-slash。

【讨论】：

我尝试在 url 中使用斜杠运行测试，但它现在输出状态码 400。
阅读文档后，这可能是我在 post 函数中返回响应的方式。我会调查的。

【解决方案2】：

好的。我认为其根本原因是序列化程序无法正确处理来自 django unittests 的传入请求。我最终删除了所有序列化程序，而是像这样替换它们：

class CategoryList(APIView):

    def get(self, request, format=None):
        categories = Category.objects.all()
        context = {
            'categories': categories
        }
        return render(
            request,
            'myapp/categories.html',
            context
            )


    def post(self, request, format=None):
        data = request.POST
        new_cat = Category(id=data['id'],
                           name=data['name'])
        new_cat.save()


        return HttpResponseRedirect(reverse('myapp:category_detail', args=[ new_cat.id ]))

新的卷曲格式：

curl http://127.0.0.1:8000/list/categories/ -d 'id=1299988&name=Tennis123'

后 HttpResponseRedirect 仍然无法正常工作（返回状态 301），但 POST/GET 现在可以使用 curl 和单元测试，这是我遇到的主要问题。

【讨论】：