Ticket #17677: 17677.diff

django/contrib/formtools/wizard/views.py

@@ -385 +385 @@
             'initial': self.get_form_initial(step),
         })
         if issubclass(self.form_list[step], forms.ModelForm):
-            # If the form is based on ModelForm, add instance if available.
-            kwargs.update({'instance': self.get_form_instance(step)})
+            # If the form is based on ModelForm, add instance if available
+            # and not previously set.
+            kwargs.setdefault('instance', self.get_form_instance(step))
         elif issubclass(self.form_list[step], forms.models.BaseModelFormSet):
-            # If the form is based on ModelFormSet, add queryset if available.
-            kwargs.update({'queryset': self.get_form_instance(step)})
+            # If the form is based on ModelFormSet, add queryset if available
+            # and not previous set.
+            kwargs.setdefault('queryset', self.get_form_instance(step))
         return self.form_list[step](**kwargs)
 
     def process_step(self, form):

django/contrib/formtools/tests/wizard/wizardtests/views.py

@@ +1 @@
+from django.contrib.auth.models import User
+from django.contrib.formtools.wizard.views import CookieWizardView
+
+
+class ModelWizard(CookieWizardView):
+
+    def get_form_kwargs(self, step):
+        return {'instance': User.objects.get(is_staff=True)}
+
+
+class QuerySetWizard(CookieWizardView):
+
+    def get_form_kwargs(self, step):
+        return {'queryset': User.objects.filter(is_staff=True)}

django/contrib/formtools/tests/wizard/wizardtests/tests.py

@@ -331 +331 @@
         response = view(factory.get('/'))
         self.assertEquals(response.context_data['test_key'], 'test_value')
         self.assertEquals(response.context_data['another_key'], 'another_value')
+
+
+class WizardModelTests(TestCase):
+
+    def setUp(self):
+        super(WizardModelTests, self).setUp()
+        # Create two users so we can filter by is_staff when handing our
+        # wizard a queryset keyword argument.
+        User.objects.create(username='test1', email='test1@example.com')
+        User.objects.create(username='test2', email='test2@example.com',
+            is_staff=True)
+
+    def test_instance_is_maintained(self):
+        from .forms import UserForm
+        from .views import ModelWizard
+
+        factory = RequestFactory()
+
+        view = ModelWizard.as_view([UserForm])
+        response = view(factory.get('/'))
+
+        instance = response.context_data['wizard']['form'].instance
+
+        self.assertNotEqual(None, instance.pk)
+        self.assertEqual('test2@example.com', instance.email)
+
+    def test_queryset_is_maintained(self):
+        from .forms import UserFormSet
+        from .views import QuerySetWizard
+
+        factory = RequestFactory()
+
+        view = QuerySetWizard.as_view([UserFormSet])
+        response = view(factory.get('/'))
+
+        formset = response.context_data['wizard']['form']
+        queryset = formset.queryset
+        emails = list(queryset.values_list('email', flat=True))
+
+        self.assertNotEqual(None, queryset)
+        self.assertEqual(1, formset.total_form_count())
+        self.assertEqual(['test2@example.com'], emails)

django/contrib/formtools/tests/wizard/wizardtests/forms.py

@@ -2 +2 @@
 import tempfile
 
 from django import forms
+from django.contrib.auth.models import User
 from django.core.files.storage import FileSystemStorage
 from django.forms.formsets import formset_factory
+from django.forms.models import modelformset_factory
 from django.http import HttpResponse
 from django.template import Template, Context
 
@@ -50 +52 @@
             context.update({'another_var': True})
         return context
 
+class UserForm(forms.ModelForm):
+    class Meta:
+        model = User
+        fields = ('username', 'email')
+
+UserFormSet = modelformset_factory(User, form=UserForm, extra=0)
+
 class SessionContactWizard(ContactWizard):
     storage_name = 'django.contrib.formtools.wizard.storage.session.SessionStorage'
 

django/contrib/formtools/tests/wizard/__init__.py

@@ -15 +15 @@
     CookieWizardTests,
     WizardTestKwargs,
     WizardTestGenericViewInterface,
+    WizardModelTests,
 )