diff -r 5ff1fc726848 -r c6bca38c1cbf parts/django/tests/regressiontests/formwizard/tests.py --- /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)