mirror of https://github.com/django/django.git
Fixed #9473: FormWizard now works with NullBooleanFields. As a bonus, we now have the beginnings of a test suite for FormWizard. Thanks, Keith Bussell.
git-svn-id: http://code.djangoproject.com/svn/django/trunk@10316 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
parent
a274ad9277
commit
da0c6908e8
|
@ -422,7 +422,7 @@ class NullBooleanSelect(Select):
|
|||
|
||||
def value_from_datadict(self, data, files, name):
|
||||
value = data.get(name, None)
|
||||
return {u'2': True, u'3': False, True: True, False: False}.get(value, None)
|
||||
return {u'2': True, u'3': False, 'True': True, 'False': False}.get(value, None)
|
||||
|
||||
def _has_changed(self, initial, data):
|
||||
# Sometimes data or initial could be None or u'' which should be the
|
||||
|
|
|
@ -0,0 +1,18 @@
|
|||
from django import forms
|
||||
from django.contrib.formtools.wizard import FormWizard
|
||||
from django.http import HttpResponse
|
||||
|
||||
class Page1(forms.Form):
|
||||
name = forms.CharField(max_length=100)
|
||||
thirsty = forms.NullBooleanField()
|
||||
|
||||
class Page2(forms.Form):
|
||||
address1 = forms.CharField(max_length=100)
|
||||
address2 = forms.CharField(max_length=100)
|
||||
|
||||
class Page3(forms.Form):
|
||||
random_crap = forms.CharField(max_length=100)
|
||||
|
||||
class ContactWizard(FormWizard):
|
||||
def done(self, request, form_list):
|
||||
return HttpResponse("")
|
|
@ -0,0 +1,13 @@
|
|||
<html>
|
||||
<body>
|
||||
<p>Step {{ step }} of {{ step_count }}</p>
|
||||
<form action="." method="post">
|
||||
<table>
|
||||
{{ form }}
|
||||
</table>
|
||||
<input type="hidden" name="{{ step_field }}" value="{{ step0 }}" />
|
||||
{{ previous_fields|safe }}
|
||||
<input type="submit">
|
||||
</form>
|
||||
</body>
|
||||
</html>
|
|
@ -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)
|
|
@ -0,0 +1,6 @@
|
|||
from django.conf.urls.defaults import *
|
||||
from forms import ContactWizard, Page1, Page2, Page3
|
||||
|
||||
urlpatterns = patterns('',
|
||||
url(r'^wiz/$', ContactWizard([Page1, Page2, Page3])),
|
||||
)
|
Loading…
Reference in New Issue