Ticket #20116: formwizard_formset_test.diff

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

@@ -13 +13 @@
 from django.contrib.formtools.tests.wizard.wizardtests.tests import (
     SessionWizardTests,
     CookieWizardTests,
+    FormsetWizardTests,
     WizardTestKwargs,
     WizardTestGenericViewInterface,
     WizardFormKwargsOverrideTests,

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

@@ -404 +404 @@
         self.assertEqual(formset.initial_form_count(), 1)
         self.assertEqual(['staff@example.com'],
             list(formset.queryset.values_list('email', flat=True)))
+
+
+class FormsetWizardTests(TestCase):
+    urls = 'django.contrib.formtools.tests.wizard.wizardtests.urls'
+    wizard_url = '/wiz_formset/'
+    wizard_step_data = (
+        {
+            'form1-INITIAL_FORMS': '0',
+            'form1-TOTAL_FORMS': '2',
+            'form1-MAX_NUM_FORMS': '0',
+            'form1-0-random_crap': 'blah blah',
+            'form1-1-random_crap': 'blah blah',
+            'cookie_contact_wizard-current_step': 'form1',
+        },
+        {
+            'form2-random_crap': 'blah blah',
+            'cookie_contact_wizard-current_step': 'form2',
+        },
+    )
+
+    def test_resend_data(self):
+        response = self.client.post(self.wizard_url, self.wizard_step_data[0])
+        self.assertEqual(response.status_code, 200)
+
+        response = self.client.post(self.wizard_url, self.wizard_step_data[1])
+        self.assertEqual(response.status_code, 200)
+
+        response = self.client.post(self.wizard_url, self.wizard_step_data[1])
+        self.assertEqual(response.status_code, 200)

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

@@ -19 +19 @@
           ('form3', Page3),
           ('form4', Page4)],
           template_name='other_wizard_form.html')),
+    url(r'^wiz_formset/$', CookieContactWizard.as_view(
+        [('form1', Page4),  # The test needs the formset as first form
+         ('form2', Page3)])),
 )