Changeset 8603
- Timestamp:
- 08/26/08 16:33:56 (3 months ago)
- Files:
-
- django/trunk/django/contrib/formtools/tests.py (modified) (2 diffs)
- django/trunk/django/contrib/formtools/wizard.py (modified) (2 diffs)
Legend:
- Unmodified
- Added
- Removed
- Modified
- Copied
- Moved
django/trunk/django/contrib/formtools/tests.py
r8597 r8603 1 1 from django import forms 2 from django.contrib.formtools import preview 2 from django.contrib.formtools import preview, wizard 3 3 from django import http 4 4 from django.test import TestCase … … 102 102 self.assertEqual(response.content, success_string) 103 103 104 # 105 # FormWizard tests 106 # 107 108 class WizardPageOneForm(forms.Form): 109 field = forms.CharField() 110 111 class WizardPageTwoForm(forms.Form): 112 field = forms.CharField() 113 114 class WizardClass(wizard.FormWizard): 115 def render_template(self, *args, **kw): 116 return "" 117 118 def done(self, request, cleaned_data): 119 return http.HttpResponse(success_string) 120 121 class DummyRequest(object): 122 def __init__(self, POST=None): 123 self.method = POST and "POST" or "GET" 124 self.POST = POST 125 126 class WizardTests(TestCase): 127 def test_step_starts_at_zero(self): 128 """ 129 step should be zero for the first form 130 """ 131 wizard = WizardClass([WizardPageOneForm, WizardPageTwoForm]) 132 request = DummyRequest() 133 wizard(request) 134 self.assertEquals(0, wizard.step) 135 136 def test_step_increments(self): 137 """ 138 step should be incremented when we go to the next page 139 """ 140 wizard = WizardClass([WizardPageOneForm, WizardPageTwoForm]) 141 request = DummyRequest(POST={"0-field":"test", "wizard_step":"0"}) 142 response = wizard(request) 143 self.assertEquals(1, wizard.step) 144 django/trunk/django/contrib/formtools/wizard.py
r8597 r8603 93 93 else: 94 94 form = self.get_form(next_step) 95 current_step = next_step95 self.step = current_step = next_step 96 96 97 97 return self.render(form, request, current_step) … … 204 204 context = context or {} 205 205 context.update(self.extra_context) 206 return render_to_response(self.get_template(s elf.step), dict(context,206 return render_to_response(self.get_template(step), dict(context, 207 207 step_field=self.step_field_name, 208 208 step0=step,
