60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
|
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)
|