模拟修补 class 方法无效

Question

我将 Django 与 DRF 和 python 模拟一起使用。我想做的是测试我的观点并模拟序列化程序和一些方法。

这是我的：

views.py

from gmm_mobile.v1.serializers import RegisterParticipationSerializer
from gmm_mobile.v1.exceptions import GameOrCampaignDoesNotExist
from gmm_util.header import Header
from gmm_util.response import ResponseUtils
from gmm_util.permissions import MobileTokenPermission
from gmm_util.permissions import MobileTokenAuthentication

class ParticipantViewMobile(APIView):
    permission_classes = (MobileTokenPermission, )
    authentication_classes = (MobileTokenAuthentication, )

    def post(self, request, service_id, campaign_id):
        try:
            environment_info = Header.get_environment_info(request)

            request.data[Field.GAME_SERVICE_ID] = service_id
            request.data[Field.CAMPAIGN] = campaign_id
            request.data[Field.LOCATION] = environment_info

            participation_serializer = RegisterParticipationSerializer(data=request.data)
            participation_serializer.is_valid(raise_exception=True)
            participation_serializer.save()

            return ResponseUtils.created()
        except Http404:
            raise GameOrCampaignDoesNotExist()

serializers.py

class RegisterParticipationSerializer(serializers.ModelSerializer):
    location = LocationSerializer(many=False)
    campaign = serializers.IntegerField()
    game_service_id = serializers.CharField(max_length=254)

    class Meta:
        model = Participation
        fields = (Field.PK, Field.EMAIL, Field.GPG_ID, Field.DATE, Field.LOCATION, Field.INFO, Field.CAMPAIGN,
                  Field.GAME_SERVICE_ID, Field.GCM_ID)
        read_only_fields = (Field.GCM_ID,)

test_views.py

from mock import patch
from django.core.urlresolvers import reverse
from rest_framework.test import APITestCase, APIRequestFactory
from rest_framework import status
from rest_framework.exceptions import ValidationError

from gmm_mobile.v1.views import ParticipantViewMobile
from gmm_mobile.v1.urls import CONSULT_PARTICIPANT
from gmm_push.environment_configuration import EnvironmentConfiguration
from gmm_util.util_test import JsonObjects

class ParticipantViewMobileTests(APITestCase):

    factory = APIRequestFactory()
    url = reverse(PARTICIPANT_MOBILE, kwargs={'service_id': 1, 'campaign_id': 1})

    def setUp(self):
        self.view = ParticipantViewMobile.as_view()

    @patch('gmm_mobile.v1.views.RegisterParticipationSerializer')
    def test__post__invalid_data__status_400(self, mock_serializer):
        # Arrange
        mock_serializer.is_valid.side_effect = ValidationError({})
        request = self.factory.post(
            self.url,
            {},
            HTTP_X_ENV=JsonObjects.environment_info_1(),
            HTTP_X_TOKEN=EnvironmentConfiguration.get_token(False),
            format='json')
        # Act
        response = self.view(request, 1, 1)
        # Assert
        self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)

但是当我运行测试时，side_effect不工作。当我在测试中更改 .is_valid 时，它对 ParticipantViewMobile.post.

没有影响

如果我做 @patch('gmm_mobile.v1.views.RegisterParticipationSerializer.is_valid') 和 mock_serializer.side_effect = ValidationError({}) 它有效，但我不希望那样，因为在某些情况下我需要模拟不止一种方法，比如 .save 和 .is_valid.

我也试过更改导入样式 views.py 和 test_views.py 但还是不行。

我是不是做错了什么或遗漏了什么？

编辑 1：

我在视图上打印了照片以试图了解发生了什么：

def post(self, request, service_id, campaign_id):
    try:
        environment_info = Header.get_environment_info(request)

        request.data[Field.GAME_SERVICE_ID] = service_id
        request.data[Field.CAMPAIGN] = campaign_id
        request.data[Field.LOCATION] = environment_info

        participation_serializer = RegisterParticipationSerializer(data=request.data)
        print RegisterParticipationSerializer.is_valid.side_effect
        print participation_serializer.is_valid.side_effect
        participation_serializer.is_valid(raise_exception=True)
        participation_serializer.save()

        return ResponseUtils.created()
    except Http404:
        raise GameOrCampaignDoesNotExist()

并且输出：

{}
None

因此，当我创建 RegisterParticipationSerializer 的实例时，我丢失了模拟方法。如何避免这种情况？

Answer 1

您的模拟不起作用，因为您模拟的是 class 而不是从 class 生成的实例。试试这个：

mock_instance = Mock()
mock_instance.is_valid.side_effect = ValidationError({})
mock_serializer.return_value = mock_instance

模拟修补 class 方法无效

Mocking patched class methods is not working

python

django

django-unittest

python-mock

django-rest-framework