def test_create_a_mutable_viewset(self): serializer_class = MagicMock(name='serializer_class') model = MagicMock(name='model') model.__name__ = 'BlogPost' model.objects.all.return_value = [] viewset_class = MutableModelViewSet.create(model, serializer_class) viewset = viewset_class() self.assertEqual(viewset.queryset, []) self.assertEqual(viewset.serializer_class, serializer_class)
def autodiscover(self): results = {} for root_name, values in settings.MEZZANINE_REST_SETTINGS.iteritems(): for app_model in values: app_name, model_name = app_model.split('.') model = get_model(app_name, model_name) serializer_class = MutableModelSerializer.create(model) viewset = MutableModelViewSet.create(model, serializer_class) results.update({ '%s/%s' % (root_name.lower(), model_name.lower()): viewset }) return results.iteritems()