encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-08-05 04:50:12 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge 46b9a6acd9 into ef2eff2aba

Browse Source
This commit is contained in:
Rocky Meza 2014-12-28 11:13:27 +00:00
commit 91019b4960
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tests/test_routers.py
View File

@ -309,15 +309,20 @@ class TestDynamicListAndDetailRouter(TestCase):
class TestRootWithAListlessViewset(TestCase): class TestRootWithAListlessViewset(TestCase):
def setUp(self): def setUp(self):
class NoteViewSet(mixins.RetrieveModelMixin, class ListlessNoteViewSet(mixins.RetrieveModelMixin,
viewsets.GenericViewSet): viewsets.GenericViewSet):
model = RouterTestModel
class FullNoteViewSet(viewsets.ModelViewSet):
model = RouterTestModel model = RouterTestModel
self.router = DefaultRouter() self.router = DefaultRouter()
self.router.register(r'notes', NoteViewSet) self.router.register(r'listless', ListlessNoteViewSet, 'listless')
self.router.register(r'listfull', ListlessNoteViewSet, 'listfull')
self.view = self.router.urls[0].callback self.view = self.router.urls[0].callback
def test_api_root(self): def test_api_root(self):
request = factory.get('/') request = factory.get('/')
response = self.view(request) response = self.view(request)
self.assertEqual(response.data, {}) self.assertIn('listfull', response.data)
self.assertNotIn('listless', response.data)