无法使用TestCase库对Django API视图进行单元测试?

时间:2018-07-10 00:17:24

标签: django-views django-testing

这是一个非常特定于项目的问题,但是我无法弄清楚我在做什么错。我有一个视图类,该类允许用户对项目数据库执行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

我在做什么错?任何帮助将不胜感激。

2 个答案:

答案 0 :(得分:0)

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

有关更多详细信息,请参阅https://docs.djangoproject.com/en/2.0/ref/settings/#append-slash

答案 1 :(得分:0)

好的。我认为这的根本原因是序列化程序无法正确处理Django单元测试的传入请求。我最终删除了所有序列化器,而是这样替换它们:

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和unittesting进行工作,这是我遇到的主要问题。