add unit tests which show m2m creation failing for through table

1 files changed, 27 insertions, 0 deletions

diff --git a/djangorestframework/tests/modelviews.py b/djangorestframework/tests/modelviews.py
index 14be178e..2fd1878a 100644
--- a/djangorestframework/tests/modelviews.py
+++ b/djangorestframework/tests/modelviews.py
@@ -4,6 +4,7 @@ from django.forms import ModelForm
 from django.contrib.auth.models import Group, User
 from djangorestframework.resources import ModelResource
 from djangorestframework.views import ListOrCreateModelView, InstanceModelView
+from djangorestframework.tests.models import CustomUser
 
 class GroupResource(ModelResource):
     model = Group
@@ -16,10 +17,15 @@ class UserForm(ModelForm):
 class UserResource(ModelResource):
     model = User
     form = UserForm
+    
+class CustomUserResource(ModelResource):
+    model = CustomUser    
 
 urlpatterns = patterns('',
     url(r'^users/$', ListOrCreateModelView.as_view(resource=UserResource), name='users'),
     url(r'^users/(?P<id>[0-9]+)/$', InstanceModelView.as_view(resource=UserResource)),
+    url(r'^customusers/$', ListOrCreateModelView.as_view(resource=CustomUserResource), name='customusers'),
+    url(r'^customusers/(?P<id>[0-9]+)/$', InstanceModelView.as_view(resource=CustomUserResource)),
     url(r'^groups/$', ListOrCreateModelView.as_view(resource=GroupResource), name='groups'),
     url(r'^groups/(?P<id>[0-9]+)/$', InstanceModelView.as_view(resource=GroupResource)),
 )
@@ -57,3 +63,24 @@ class ModelViewTests(TestCase):
         
         group = user.groups.all()[0]
         self.assertEqual('foo', group.name)
+        
+    def test_creation_with_m2m_relation_through(self):
+        """
+        Ensure that a model object with a m2m relation can be created where that
+        relation uses a through table
+        """
+        group = Group(name='foo')
+        group.save()
+        self.assertEqual(0, User.objects.count())
+
+        response = self.client.post('/customusers/', {'username': 'bar', 'groups': [group.id]})
+        
+        self.assertEqual(response.status_code, 201)
+        self.assertEqual(1, CustomUser.objects.count())
+       
+        user = CustomUser.objects.all()[0]
+        self.assertEqual('bar', user.username)
+        self.assertEqual(1, user.groups.count())
+        
+        group = user.groups.all()[0]
+        self.assertEqual('foo', group.name)