diff --git a/tests/test_routers.py b/tests/test_routers.py index 06ab8103a..dd9d3db5e 100644 --- a/tests/test_routers.py +++ b/tests/test_routers.py @@ -309,15 +309,20 @@ class TestDynamicListAndDetailRouter(TestCase): class TestRootWithAListlessViewset(TestCase): def setUp(self): - class NoteViewSet(mixins.RetrieveModelMixin, - viewsets.GenericViewSet): + class ListlessNoteViewSet(mixins.RetrieveModelMixin, + viewsets.GenericViewSet): + model = RouterTestModel + + class FullNoteViewSet(viewsets.ModelViewSet): model = RouterTestModel 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 def test_api_root(self): request = factory.get('/') response = self.view(request) - self.assertEqual(response.data, {}) + self.assertIn('listfull', response.data) + self.assertNotIn('listless', response.data)