Skip to content

Commit 5ddf120

Browse files
author
Tim Csitkovics
committed
Make import backward compatible
1 parent bd2a0c8 commit 5ddf120

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

example/tests/test_serializers.py

+7-3
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
from collections import OrderedDict
2-
from unittest.mock import patch
32

43
import pytest
54
from django.test import TestCase
@@ -18,6 +17,11 @@
1817
from example.models import Author, Blog, Entry
1918
from example.serializers import BlogSerializer
2019

20+
try:
21+
from unittest import mock
22+
except ImportError:
23+
import mock
24+
2125
factory = APIRequestFactory()
2226
pytestmark = pytest.mark.django_db
2327

@@ -44,7 +48,7 @@ def setUp(self):
4448
)
4549

4650
def test_forward_relationship_not_loaded_when_not_included(self):
47-
with patch('example.serializers.BlogSerializer.to_representation') as mock:
51+
with mock.patch('example.serializers.BlogSerializer.to_representation') as mocked_serializer:
4852
class EntrySerializer(ModelSerializer):
4953
blog = BlogSerializer()
5054

@@ -56,7 +60,7 @@ class Meta:
5660
serializer = EntrySerializer(context={'request': request_without_includes})
5761
serializer.to_representation(self.entry)
5862

59-
mock.assert_not_called()
63+
mocked_serializer.assert_not_called()
6064

6165
def test_forward_relationship_optimization_correct_representation(self):
6266
class EntrySerializer(ModelSerializer):

0 commit comments

Comments
 (0)