--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/parts/django/tests/regressiontests/formwizard/tests.py Sat Jan 08 11:20:57 2011 +0530
@@ -0,0 +1,59 @@
+import re
+from django import forms
+from django.test import TestCase
+
+class FormWizardWithNullBooleanField(TestCase):
+ urls = 'regressiontests.formwizard.urls'
+
+ input_re = re.compile('name="([^"]+)" value="([^"]+)"')
+
+ wizard_url = '/wiz/'
+ wizard_step_data = (
+ {
+ '0-name': 'Pony',
+ '0-thirsty': '2',
+ },
+ {
+ '1-address1': '123 Main St',
+ '1-address2': 'Djangoland',
+ },
+ {
+ '2-random_crap': 'blah blah',
+ }
+ )
+
+ def grabFieldData(self, response):
+ """
+ Pull the appropriate field data from the context to pass to the next wizard step
+ """
+ previous_fields = response.context['previous_fields']
+ fields = {'wizard_step': response.context['step0']}
+
+ def grab(m):
+ fields[m.group(1)] = m.group(2)
+ return ''
+
+ self.input_re.sub(grab, previous_fields)
+ return fields
+
+ def checkWizardStep(self, response, step_no):
+ """
+ Helper function to test each step of the wizard
+ - Make sure the call succeeded
+ - Make sure response is the proper step number
+ - return the result from the post for the next step
+ """
+ step_count = len(self.wizard_step_data)
+
+ self.assertEqual(response.status_code, 200)
+ self.assertContains(response, 'Step %d of %d' % (step_no, step_count))
+
+ data = self.grabFieldData(response)
+ data.update(self.wizard_step_data[step_no - 1])
+
+ return self.client.post(self.wizard_url, data)
+
+ def testWizard(self):
+ response = self.client.get(self.wizard_url)
+ for step_no in range(1, len(self.wizard_step_data) + 1):
+ response = self.checkWizardStep(response, step_no)