Дразнить связанного менеджера в Django 2 - PullRequest
Новая
       60

Дразнить связанного менеджера в Django 2

2 голосов
/ 29 апреля 2019

Этот вопрос напрямую связан с этим вопросом , но кажется, что этот вопрос уже устарел.

Я пытаюсь проверить представление без необходимости доступа к базе данных.Для этого мне нужно смоделировать RelatedManager для пользователя.

Я использую pytest и pytest-mock.

models.py 

# truncated for brevity, taken from django-rest-knox
class AuthToken(models.Model):
    user = models.ForeignKey(
        User, 
        null=False, 
        blank=False,
        related_name='auth_token_set', 
        on_delete=models.CASCADE
    )

views.py 

class ChangeEmail(APIView):
    permission_classes = [permissions.IsAdmin]
    serializer_class = serializers.ChangeEmail

    def post(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        serializer.is_valid(raise_exception=True)

        user = request.user
        user.email = request.validated_data['email']
        user.save()

        # Logout user from all devices
        user.auth_token_set.all().delete() # <--- How do I mock this?

        return Response(status=status.HTTP_200_OK)

test_views.py 

def test_valid(mocker, user_factory):
    user = user_factory.build()
    user.id = 1

    data = {
        'email': 'foo@example.com'
    }

    factory = APIRequestFactory()
    request = factory.post('/', data=data)
    force_authenticate(request, user)

    mocker.patch.object(user, "save")

    related_manager = mocker.patch(
        'django.db.models.fields.related.ReverseManyToOneDescriptor.__set__',
        return_vaue=mocker.MagicMock()
    )
    related_manager.all = mocker.MagicMock()
    related_manager.all.delete = mocker.MagicMock()

    response = ChangeEmail.as_view()(request)
    assert response.status_code == status.HTTP_200_OK

Рисунок из ответа в связанном вопросе, который я пытался пропатчить ReverseManyToOneDescriptor.Тем не менее, похоже, что на самом деле это не надо, потому что тест все еще пытается подключиться к базе данных, когда пытается удалить пользователя auth_token_set.

Ответы [ 2 ]

1 голос
/ 29 апреля 2019

Вам нужно будет смоделировать возвращаемое значение фабричной функции create_reverse_many_to_one_manager.Пример: 

def test_valid(mocker):
    mgr = mocker.MagicMock()
    mocker.patch(
        'django.db.models.fields.related_descriptors.create_reverse_many_to_one_manager', 
        return_value=mgr
    )

    user = user_factory.build()
    user.id = 1
    ...
    mgr.assert_called()

Помните, что приведенный выше пример будет издеваться над менеджером оборотов для всех моделей.Если вам нужен более детальный подход (например, исправьте только менеджер оборотов User.auth_token, оставьте остальные не исправленными), предоставьте пользовательский заводской импл, например 

def test_valid(mocker):
    mgr = mocker.MagicMock()
    factory_orig = related_descriptors.create_reverse_many_to_one_manager
    def my_factory(superclass, rel):
        if rel.model == User and rel.name == 'auth_token_set':
            return mgr
        else:
            return factory_orig(superclass, rel)

    mocker.patch(
        'django.db.models.fields.related_descriptors.create_reverse_many_to_one_manager',
        my_factory
    )

    user = user_factory.build()
    user.id = 1
    ...
    mgr.assert_called()
1 голос
/ 29 апреля 2019

Если вы используете django APITestCase, это становится относительно простым. 

class TestChangeEmail(APITestCase):
    def test_valid(self):
        user = UserFactory()
        auth_token = AuthToken.objects.create(user=user)

        response = self.client.post(
            reverse('your endpoint'), 
            data={'email': 'foo@example.com'}
        )

        self.assertEqual(response.status_code, status.HTTP_200_OK)
        self.assertFalse(AuthToken.objects.filter(user=user).exists())

Это полностью исключает насмешки и дает более точное представление о вашей логике.

...