DJANGO REST 和 mongoengine 中的 PUT 测试用例

Question

我使用 DJango 1.8.3 和 mongo 作为数据库，使用 mongoengine 开发基于 rest 的 API。

我的models.py文件是

class Client(Document):
    client_name = StringField(max_length=50)
    client_email = EmailField()
    city = StringField(max_length=50)
    country = StringField(max_length=200, verbose_name="Country") 

序列化器.py 类 ClientSerializer(DocumentSerializer):

    class Meta:
        model = Client
        depth = 1

views.py

class ClientDetails(generics.RetrieveUpdateDestroyAPIView):
    serializer_class = ClientSerializer
    queryset = Client.objects.all()

class ClientList(generics.ListCreateAPIView):
    serializer_class = ClientSerializer
    queryset = Client.objects.all()

urls.py

urlpatterns = [
url(r'^client/$', views.ClientList.as_view(), name='client'),
url(r'^client/(?P<id>[\w]{24})/$', views.ClientDetails.as_view(),
    name='client_details'),
]

test.py

def test_get_client(self):
    """
    Ensure we get existing clients from mongo database.
    """
    url = '/client_management/client/'
    factory = APIRequestFactory()
    user = User.objects.get(username='jacob')
    view = views.ClientList.as_view()

    # Make an authenticated request to the view...
    request = factory.get(url)
    force_authenticate(request, user=user)
    response = view(request)
    self.assertEqual(response.status_code, 200)

def test_delete_client(self):
    """
    Ensure we can delete new clients in mongo database.
    """
    url = '/client_management/client/'
    data = {'client_name'='XX'}
    factory = APIRequestFactory()
    user = User.objects.get(username='jacob')
    view = views.ClientList.as_view()

    # Make an authenticated request to the view...
    request = factory.post(url, data=data,)
    force_authenticate(request, user=user)
    response = view(request)
    ##self.assertEqual(response.status_code, status.HTTP_201_CREATED)

    g_id = response.data['id']
    urls = '/client_management/client/?id=' + g_id
    ## even tried
    ## urls = '/client_management/client/ + g_id + /
    data['id'] = g_id
    #data['client_name'] = 'MAA'
    view = views.ClientDetails.as_view()
    request = factory.delete(urls)
    ## tried this also
    ## request = factory.delete('/client_management/client/', id=g_id)
    print request
    force_authenticate(request, user=user)
    response = view(request)
    self.assertEqual(response.status_code, 201)

按预期工作。 GET 和 POST 测试用例运行良好

当我为 PUT 编写测试用例时，DELETE

我收到了错误

Expected view ClientDetails to be called with a URL keyword argument named "id".
Fix your URL conf, or set the `.lookup_field` attribute on the view correctly

从错误很明显，我尝试在 url 中传递客户端 ID 作为请求中的单独参数。

但错误仍然相同，在这种情况下不知道如何传递 id 任何帮助表示赞赏。

提前感谢您的帮助

Answer 1

终于明白了

def test_delete_client(self):
    """
    Ensure we can delete new clients in mongo database.
    """
    url = '/client_management/client/'
    data = {'client_name'='XX'}
    factory = APIRequestFactory()
    user = User.objects.get(username='jacob')
    view = views.ClientList.as_view()

    # Make an authenticated request to the view...
    request = factory.post(url, data=data,)
    force_authenticate(request, user=user)
    response = view(request)
    ##self.assertEqual(response.status_code, status.HTTP_201_CREATED)

    g_id = response.data['id']
    urls = '/client_management/client/?id=' + g_id
    ## even tried
    ## urls = '/client_management/client/ + g_id + /
    data['id'] = g_id
    #data['client_name'] = 'MAA'
    view = views.ClientDetails.as_view()
    request = factory.delete(urls)
    ## tried this also
    ## request = factory.delete('/client_management/client/')
    print request
    force_authenticate(request, user=user)
    response = view(request, id=str(g_id))
    self.assertEqual(response.status_code, 201)

调用 view(request) 时必须发送 ID 作为 response = view(request, id=str(g_id))。

Answer 2

docs 似乎表明默认的lookup_field 是pk 而不是id。我会尝试将 url 模式更改为

...
url(r'^client/(?P<pk>[\w]{24})/$', views.ClientDetails.as_view(),
    name='client_details'),
...

或在视图中添加lookup_field 属性

class ClientDetails(generics.RetrieveUpdateDestroyAPIView):
    serializer_class = ClientSerializer
    queryset = Client.objects.all()
    lookup_field = ('id',) # Note the parenthesis, they are needed so the object is a tuple

Answer 3

更改代码并重新发布

urls.py

urlpatterns = [
    url(r'^client/(?P<id>[\w]{24})/$', views.ClientDetails.as_view(),
    name='client_details'),
]

test.py 在您传递 pk 值的 url 中。在测试用例中，您还需要传递 id 的 pk 值。在上面的代码中，我吸收了你将 id 作为query-parameter 传递，所以我将它作为 pk 传递。

def test_delete_client(self):
    """
    Ensure we can delete new clients in mongo database.
    """
    g_id = response.data['id']
    urls = '/client_management/client/' + str(g_id) + '/'
    view = views.ClientDetails.as_view()