Skip to content
Closed
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 18 additions & 1 deletion example/tests/integration/test_includes.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,10 @@
from django.core.urlresolvers import reverse
from example.tests.utils import load_json

pytestmark = pytest.mark.django_db
from rest_framework_json_api.pagination import PageNumberPagination
from example.views import EntryViewSet

pytestmark = pytest.mark.django_db

def test_included_data_on_list(multiple_entries, client):
multiple_entries[1].comments = []
Expand All @@ -18,3 +20,18 @@ def test_included_data_on_detail(single_entry, client):

assert [x.get('type') for x in included] == ['comments']

def test_included_all_data(multiple_entries, rf):
class NoPagination(PageNumberPagination):
page_size = None

class NonPaginatedEntryViewSet(EntryViewSet):
pagination_class = NoPagination

request = rf.get( reverse("entry-list") + '?include=comments')
view = NonPaginatedEntryViewSet.as_view({'get': 'list'})
response = view(request)
response.render()

included = load_json(response.content).get('included')

assert len(included) == 2