5

我之前为我的观点创建了一些测试。像那样

class TestUserRegistrationViewUserCreate(APITestCase):
def setUp(self):
    self.factory = APIRequestFactory()

def test_create_user(self):
    data = {
        'phone_number': '+79513332211',
        'password': 'qwerty'
    }
    request = self.factory.post(reverse('user'), data=data)
    response = CustomUserAPIView.as_view()(request)
    self.assertEqual(response.status_code, status.HTTP_201_CREATED)

一切都很好，直到我被要求添加 API 版本控制。

DRF 原生支持版本控制http://www.django-rest-framework.org/api-guide/versioning/ 所以我只是使用它并在我的 API 中添加了基于名称空间的版本控制

REST_FRAMEWORK = {
    'DEFAULT_VERSIONING_CLASS': 'rest_framework.versioning.NamespaceVersioning'
}

现在我需要重写我的视图单元测试以支持版本控制。

这个问题是为了获得版本化的 url reverse，我必须使用

from rest_framework.reverse import reverse

reverse('bookings-list', request=request)

就像在文档中一样。

但是我在测试中没有request对象，因为我自己制作了一个并且制作它所需的版本化 url。

我应该怎么办？

PS 我可以在不使用 DRF 的情况下实现版本控制，使用视图装饰器和几个 utils 函数来解决这个问题，但是当我重新发明轮子时，这对我来说感觉很糟糕。另外，我也可能会忘记一些边缘情况。

4

2 回答 2

0

我reverse('<VERSION>:<VIEW_NAME>')在我的测试用例中使用。

于 2016-10-31T10:46:48.137 回答
0

很晚了，但对于那些有类似问题的人，您可以在调用视图时传递版本 -

response = CustomUserAPIView.as_view()(request, version='1.0')
于 2019-06-20T12:04:29.747 回答