This commit is contained in:
Rocky Meza 2014-12-28 11:13:27 +00:00
commit 91019b4960

View File

@ -309,15 +309,20 @@ class TestDynamicListAndDetailRouter(TestCase):
class TestRootWithAListlessViewset(TestCase):
def setUp(self):
class NoteViewSet(mixins.RetrieveModelMixin,
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)