Skip to content

Commit 736837f

Browse files
committed
cleaned up tests
1 parent fe0056d commit 736837f

File tree

5 files changed

+20
-45
lines changed

5 files changed

+20
-45
lines changed

example/models.py

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -34,18 +34,6 @@ def __str__(self):
3434
return self.name
3535

3636

37-
@python_2_unicode_compatible
38-
class RenamedAuthor(Author):
39-
class Meta:
40-
proxy = True
41-
42-
class JSONAPIMeta:
43-
resource_name = "super-author"
44-
45-
def __str__(self):
46-
return self.name
47-
48-
4937
@python_2_unicode_compatible
5038
class Entry(BaseModel):
5139
blog = models.ForeignKey(Blog)

example/serializers.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from rest_framework_json_api import serializers, relations
2-
from example.models import Blog, Entry, Author, Comment, RenamedAuthor
2+
from example.models import Blog, Entry, Author, Comment
33

44

55
class BlogSerializer(serializers.ModelSerializer):
@@ -35,7 +35,7 @@ def get_suggested(self, obj):
3535
class Meta:
3636
model = Entry
3737
fields = ('blog', 'headline', 'body_text', 'pub_date', 'mod_date',
38-
'authors', 'comments', 'suggested',)
38+
'authors', 'comments', 'suggested',)
3939

4040

4141
class AuthorSerializer(serializers.ModelSerializer):
@@ -45,13 +45,6 @@ class Meta:
4545
fields = ('name', 'email',)
4646

4747

48-
class RenamedAuthorSerializer(serializers.ModelSerializer):
49-
50-
class Meta:
51-
model = RenamedAuthor
52-
fields = ('name', 'email',)
53-
54-
5548
class CommentSerializer(serializers.ModelSerializer):
5649

5750
class Meta:

example/tests/integration/test_model_resource_name.py

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -8,20 +8,27 @@
88
pytestmark = pytest.mark.django_db
99

1010

11-
def test_model_resource_name_on_list(single_entry, client):
12-
response = client.get(reverse("renamed-authors-list"))
11+
class _PatchedModel:
12+
class JSONAPIMeta:
13+
resource_name = "resource_name_from_JSONAPIMeta"
14+
15+
16+
def test_match_model_resource_name_on_list(single_entry, client):
17+
models.Comment.__bases__ += (_PatchedModel,)
18+
response = client.get(reverse("comment-list"))
1319
data = load_json(response.content)['data']
1420
# name should be super-author instead of model name RenamedAuthor
15-
assert [x.get('type') for x in data] == ['super-author'], 'List included types are incorrect'
21+
assert [x.get('type') for x in data] == ['resource_name_from_JSONAPIMeta'], 'List included types are incorrect'
22+
1623

1724
@pytest.mark.usefixtures("single_entry")
1825
class ResourceNameConsistencyTest(APITestCase):
19-
26+
2027
def test_type_match_on_included_and_inline_base(self):
2128
self._check_relationship_and_included_comment_type_are_the_same(reverse("entry-list"))
2229

2330
def test_type_match_on_included_and_inline_with_JSONAPIMeta(self):
24-
models.Comment.__bases__ += (self._PatchedModel,)
31+
models.Comment.__bases__ += (_PatchedModel,)
2532

2633
self._check_relationship_and_included_comment_type_are_the_same(reverse("entry-list"))
2734

@@ -31,7 +38,7 @@ def test_type_match_on_included_and_inline_with_serializer_resource_name(self):
3138
self._check_relationship_and_included_comment_type_are_the_same(reverse("entry-list"))
3239

3340
def test_type_match_on_included_and_inline_with_serializer_resource_name_and_JSONAPIMeta(self):
34-
models.Comment.__bases__ += (self._PatchedModel,)
41+
models.Comment.__bases__ += (_PatchedModel,)
3542
serializers.CommentSerializer.Meta.resource_name = "resource_name_from_serializer"
3643

3744
self._check_relationship_and_included_comment_type_are_the_same(reverse("entry-list"))
@@ -45,12 +52,12 @@ def test_resource_and_relationship_type_match_with_serializer_resource_name(self
4552
self._check_resource_and_relationship_comment_type_match()
4653

4754
def test_resource_and_relationship_type_match_with_JSONAPIMeta(self):
48-
models.Comment.__bases__ += (self._PatchedModel,)
55+
models.Comment.__bases__ += (_PatchedModel,)
4956

5057
self._check_resource_and_relationship_comment_type_match()
5158

5259
def test_resource_and_relationship_type_match_with_serializer_resource_name_and_JSONAPIMeta(self):
53-
models.Comment.__bases__ += (self._PatchedModel,)
60+
models.Comment.__bases__ += (_PatchedModel,)
5461
serializers.CommentSerializer.Meta.resource_name = "resource_name_from_serializer"
5562

5663
self._check_resource_and_relationship_comment_type_match()
@@ -81,9 +88,3 @@ def tearDown(self):
8188
delattr(serializers.CommentSerializer.Meta, "resource_name")
8289
except AttributeError:
8390
pass
84-
85-
class _PatchedModel:
86-
87-
class JSONAPIMeta:
88-
resource_name = "resource_name_from_JSONAPIMeta"
89-

example/urls_test.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,15 +2,14 @@
22
from rest_framework import routers
33

44
from example.views import BlogViewSet, EntryViewSet, AuthorViewSet, EntryRelationshipView, BlogRelationshipView, \
5-
CommentRelationshipView, AuthorRelationshipView, RenamedAuthorViewSet, CommentViewSet
5+
CommentRelationshipView, AuthorRelationshipView, CommentViewSet
66
from .api.resources.identity import Identity, GenericIdentity
77

88
router = routers.DefaultRouter(trailing_slash=False)
99

1010
router.register(r'blogs', BlogViewSet)
1111
router.register(r'entries', EntryViewSet)
1212
router.register(r'authors', AuthorViewSet)
13-
router.register(r'renamed-authors', RenamedAuthorViewSet, base_name='renamed-authors')
1413
router.register(r'comments', CommentViewSet)
1514

1615
# for the old tests

example/views.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,8 @@
11
from rest_framework import viewsets
22
from rest_framework_json_api.views import RelationshipView
3-
from example.models import Blog, Entry, Author, Comment, RenamedAuthor
3+
from example.models import Blog, Entry, Author, Comment
44
from example.serializers import (
5-
BlogSerializer, EntrySerializer, AuthorSerializer, CommentSerializer,
6-
RenamedAuthorSerializer)
5+
BlogSerializer, EntrySerializer, AuthorSerializer, CommentSerializer)
76

87

98
class BlogViewSet(viewsets.ModelViewSet):
@@ -22,11 +21,6 @@ class AuthorViewSet(viewsets.ModelViewSet):
2221
serializer_class = AuthorSerializer
2322

2423

25-
class RenamedAuthorViewSet(viewsets.ModelViewSet):
26-
queryset = RenamedAuthor.objects.all()
27-
serializer_class = RenamedAuthorSerializer
28-
29-
3024
class CommentViewSet(viewsets.ModelViewSet):
3125
queryset = Comment.objects.all()
3226
serializer_class = CommentSerializer

0 commit comments

Comments
 (0)