Refactored tests so we have an organized base to scale from

Author: jerel

example/api/api.py

@@ -0,0 +1,55 @@
+from django.contrib.auth import models as auth_models
+from rest_framework import viewsets, generics, renderers, parsers
+from rest_framework.decorators import list_route, detail_route
+from rest_framework.response import Response
+from rest_framework_ember import mixins, utils
+from ..serializers.identity import IdentitySerializer
+from ..serializers.post import PostSerializer
+
+
+class Identity(mixins.MultipleIDMixin, viewsets.ModelViewSet):
+    queryset = auth_models.User.objects.all()
+    serializer_class = IdentitySerializer
+
+    # demonstrate sideloading data for use at app boot time
+    @list_route()
+    def posts(self, request):
+        self.resource_name = False
+
+        identities = self.queryset
+        posts = [{'id': 1, 'title': 'Test Blog Post'}]
+
+        data = {
+            u'identities': IdentitySerializer(identities, many=True).data,
+            u'posts': PostSerializer(posts, many=True).data,
+        }
+        return Response(utils.format_keys(data, format_type='camelize'))
+
+    @detail_route()
+    def manual_resource_name(self, request, *args, **kwargs):
+        self.resource_name = 'data'
+        return super(Identity, self).retrieve(request, args, kwargs)
+
+
+class GenericIdentity(generics.GenericAPIView):
+    """
+    Current user's identity endpoint.
+
+    GET /identities/generic
+    """
+    serializer_class = IdentitySerializer
+    allowed_methods = ['GET']
+    renderer_classes = (renderers.JSONRenderer, )
+    parser_classes = (parsers.JSONParser, )
+
+
+    def get_queryset(self):
+        return auth_models.User.objects.all()
+
+    def get(self, request, pk=None):
+        """
+        GET request
+        """
+        obj = self.get_object()
+        return Response(IdentitySerializer(obj).data)
+

example/api/resources/__init__.py

@@ -0,0 +1,13 @@
+from django.contrib.auth import models as auth_models
+from rest_framework import serializers
+
+
+class IdentitySerializer(serializers.ModelSerializer):
+    """
+    Identity Serializer
+    """
+    class Meta:
+        model = auth_models.User
+        fields = (
+            'id', 'first_name', 'last_name', 'email', )
+

example/api/resources/identity.py

@@ -0,0 +1,9 @@
+from rest_framework import serializers
+
+
+class PostSerializer(serializers.Serializer):
+    """
+    Blog post serializer
+    """
+    title = serializers.CharField(max_length=50)
+

example/api/serializers/__init__.py

@@ -2,25 +2,16 @@
 Example app URLs
 """
 from django.conf.urls import patterns, include, url
-from .api import (
-    User, UserEmber, EmberUserModelViewSet, MultipleIDMixinUserModelViewSet,
-    UserCarResource)
-
 from rest_framework import routers
+from .resources.identity import Identity, GenericIdentity
+
+router = routers.DefaultRouter(trailing_slash=False)
+
+router.register(r'identities', Identity)
+
+urlpatterns = router.urls
 
-urlpatterns = patterns('',
-    url(r'^user-default/(?P<pk>\d+)/$', User.as_view(), name='user-default'),
-    url(r'^user-ember/(?P<pk>\d+)/$', UserEmber.as_view(), name='user-ember'),
-    url(r'^user-mixin-viewset/$',
-        MultipleIDMixinUserModelViewSet.as_view({'get': 'list'}),
-        name='mixin-user-list'),
-    url(r'^user-viewset/$', EmberUserModelViewSet.as_view({'get': 'list'}),
-        name='user-list'),
-    url(r'^user-viewset/(?P<pk>\d+)/$',
-        EmberUserModelViewSet.as_view(
-            {'get': 'retrieve', 'post': 'create', 'put': 'update'}),
-        name='user-detail'),
-    url(r'users-cars/$', UserCarResource.as_view(),
-        name='users-cars'),
+urlpatterns += patterns('',
+    url(r'identities/default/(?P<pk>\d+)', GenericIdentity.as_view(), name='user-default'),
 )
 

example/api/serializers/identity.py

@@ -40,12 +40,12 @@
     'DEFAULT_PAGINATION_SERIALIZER_CLASS':
         'rest_framework_ember.pagination.PaginationSerializer',
     'DEFAULT_PARSER_CLASSES': (
-        # 'rest_framework_ember.parsers.JSONParser',
+        'rest_framework_ember.parsers.JSONParser',
         'rest_framework.parsers.FormParser',
         'rest_framework.parsers.MultiPartParser'
     ),
     'DEFAULT_RENDERER_CLASSES': (
-        # 'rest_framework_ember.renderers.JSONRenderer',
+        'rest_framework_ember.renderers.JSONRenderer',
         'rest_framework.renderers.JSONRenderer',
         'rest_framework.renderers.BrowsableAPIRenderer',
     ),

example/api/serializers/post.py

@@ -48,7 +48,7 @@ def test_camelization(self):
         meta = json_content.get('meta')
 
         self.assertEquals(expected.get('users'), json_content.get('users'))
-        self.assertEqual('http://testserver/user-viewset/?page=2',
+        self.assertEqual(u'http://testserver/identities?page=2',
             meta.get('nextLink'))
 
     def test_pluralization(self):
@@ -76,6 +76,7 @@ def test_pluralization(self):
         json_content = json.loads(response.content)
         self.assertEquals(expected.get('users'), json_content.get('users'))
 
+    def test_empty_pluralization(self):
         #test that the key is still pluralized when there are no records for the
         #model, as long as the endpoint serves a list
         get_user_model().objects.all().delete()
@@ -84,3 +85,4 @@ def test_pluralization(self):
 
         json_content = json.loads(response.content)
         self.assertEqual(json_content.get('users'), [])
+

example/api/urls.py

@@ -32,7 +32,7 @@ def test_ember_expected_renderer(self):
         The :class:`UserEmber` ViewSet has the ``resource_name`` of 'data'
         so that should be the key in the JSON response.
         """
-        url = reverse('user-ember', kwargs={'pk': self.miles.pk})
+        url = reverse('user-manual-resource-name', kwargs={'pk': self.miles.pk})
 
         response = self.client.get(url)
         self.assertEqual(200, response.status_code)