Django Rest Api - 如何测试POST是否将数据保存在数据库中?

问题描述 投票:0回答:1

所以我发现了这个问题:Django Rest Framework testing save POST request data

我知道只要函数正在运行,就可以访问使用POST请求创建的数据。所以我在一个函数中完成了整个测试:

class PostMovieAPITest(APITestCase):
    def test_correct_request(self):
        title = 'Snatch'
        response = self.client.post('/movies/', data={'title': title}, format='json')
        self.assertEqual(response.status_code, status.HTTP_201_CREATED)
        movies = Movie.objects.all()
        self.assertTrue(Movie.objects.get(title=title))

问题是,即使我在同一个函数中发送CREATE / POST请求,Movie.objects.all()也是空的。当我执行`manage.py runserver'并在浏览器中测试它时,API工作正常。但是如何编写一个合适的测试来检查数据是否实际保存在数据库中?

URLs.朋友:

from django.contrib import admin
from django.urls import path, include
from rest_framework import routers
from api import views


router = routers.DefaultRouter()
router.register(r'movies', views.MovieViewSet)

urlpatterns = [
    path('', include(router.urls)),
    path('admin/', admin.site.urls),
]

views.朋友:

class MovieViewSet(viewsets.ModelViewSet):
    queryset = Movie.objects.all()
    serializer_class = MovieSerializer

    def create(self, request, *args, **kwargs):
        title = request.data.get('title')
        if not title:
            return Response({'Error': "Body should be {'title':'The title of the Movie'}"}, status=status.HTTP_400_BAD_REQUEST)

        data = get_data_from_omdb(title)

        if len(data) == 0:
            return Response({"Error": "Title does not exist in OMDB database"}, status=status.HTTP_400_BAD_REQUEST)

        serializer = MovieSerializer(data=data, context={'request': request})

        if serializer.is_valid(raise_exception=False):
            serializer.save()
            return Response(serializer.data, status=status.HTTP_201_CREATED)
        else:
            return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
django django-rest-framework django-testing
1个回答
1
投票

由于Ozgur Akcali的评论,发现了这个问题。我无法获得电影,因为Movie对象是基于外部API的数据创建的,包括标题。

“雇用我!” - >“雇用我!”

有时脚本让我想要杀人。非常感谢,抱歉浪费你的时间。

def test_correct_request(self):
    title = 'Hire Me!' # was: 'Hire me!'
    response = self.client.post('/movies/', data={'title': title}, format='json')
    self.assertEqual(response.status_code, status.HTTP_201_CREATED)
    self.assertTrue(Movie.objects.get(title=title))
© www.soinside.com 2019 - 2024. All rights reserved.