Fixed #138 -- UserCreateForm and UserUpdateForm get floppified.

example/blog/admin2.py

@@ -6,10 +6,15 @@ from django.contrib.auth.models import Group, User
 from rest_framework.relations import PrimaryKeyRelatedField
 
 import djadmin2
+from djadmin2.forms import floppify_form
 from djadmin2.models import ModelAdmin2, Admin2Inline
 from djadmin2.apiviews import Admin2APISerializer
 
 
+UserCreationForm = floppify_form(UserCreationForm)
+UserChangeForm = floppify_form(UserChangeForm)
+
+
 class GroupSerializer(Admin2APISerializer):
     permissions = PrimaryKeyRelatedField(many=True)
 

example/blog/tests/__init__.py

@@ -1,3 +1,4 @@
+from test_auth_admin import *
 from test_apiviews import *
 from test_builtin_api_resources import *
 from test_views import *

example/blog/tests/test_apiviews.py

@@ -19,7 +19,6 @@ class ViewTest(TestCase):
 
 
 class IndexAPIViewTest(ViewTest):
-
     def test_response_ok(self):
         request = self.factory.get(reverse('admin2:api-index'))
         view = apiviews.IndexAPIView.as_view(**default.get_api_index_kwargs())
@@ -28,7 +27,6 @@ class IndexAPIViewTest(ViewTest):
 
 
 class ListCreateAPIViewTest(ViewTest):
-
     def test_response_ok(self):
         request = self.factory.get(reverse('admin2:blog_post_api-list'))
         model_admin = self.get_model_admin(Post)

example/blog/tests/test_auth_admin.py

@@ -0,0 +1,53 @@
+import floppyforms
+from django.contrib.auth.models import User
+from django.core.urlresolvers import reverse
+from django.test import TestCase
+from django.test.client import RequestFactory
+import djadmin2
+from blog.admin2 import UserAdmin2
+
+
+class UserAdminTest(TestCase):
+    def setUp(self):
+        self.factory = RequestFactory()
+        self.user = User(
+            username='admin',
+            is_staff=True,
+            is_superuser=True)
+        self.user.set_password('admin')
+        self.user.save()
+
+    def test_create_form_uses_floppyform_widgets(self):
+        form = UserAdmin2.create_form_class()
+        self.assertTrue(
+            isinstance(form.fields['username'].widget,
+                       floppyforms.TextInput))
+
+        request = self.factory.get(reverse('admin2:auth_user_create'))
+        request.user = self.user
+        model_admin = UserAdmin2(User, djadmin2.default)
+        view = model_admin.create_view.as_view(
+            **model_admin.get_create_kwargs())
+        response = view(request)
+        form = response.context_data['form']
+        self.assertTrue(
+            isinstance(form.fields['username'].widget,
+                       floppyforms.TextInput))
+
+    def test_update_form_uses_floppyform_widgets(self):
+        form = UserAdmin2.update_form_class()
+        self.assertTrue(
+            isinstance(form.fields['username'].widget,
+                       floppyforms.TextInput))
+
+        request = self.factory.get(
+            reverse('admin2:auth_user_update', args=(self.user.pk,)))
+        request.user = self.user
+        model_admin = UserAdmin2(User, djadmin2.default)
+        view = model_admin.update_view.as_view(
+            **model_admin.get_update_kwargs())
+        response = view(request, pk=self.user.pk)
+        form = response.context_data['form']
+        self.assertTrue(
+            isinstance(form.fields['username'].widget,
+                       floppyforms.TextInput))