Пример #1
0
    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)
Пример #2
0
    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()