Ticket #6893: form_wizard_patch_2.diff

django/contrib/formtools/wizard.py

@@ -90 +90 @@
-            # Otherwise, move along to the next step.
-            else:
-                form = self.get_form(next_step)
-
+self.step = 
-
-        return self.render(form, request, current_step)
-
@@ -209 +209 @@
-        """
-        context = context or {}
-        context.update(self.extra_context)
-elf.s
+
-            step_field=self.step_field_name,
-            step0=step,
-            step=step + 1,

django/contrib/formtools/tests.py

@@ -1 +1 @@
-from django import forms
-
+, wizard
-from django import http
-from django.test import TestCase
-
@@ -85 +85 @@
-        response = self.client.post('/test1/', test_data)
-        self.assertEqual(response.content, success_string)
-
+class WizardPageOneForm(forms.Form):
+    field = forms.CharField()
+    
+class WizardPageTwoForm(forms.Form):
+    field = forms.CharField()
+    
+class WizardClass(wizard.FormWizard):
+
+    def get_template(self, step):
+        return "formtools/wizard_form.html"
+    
+    def done(self, request, cleaned_data):
+        return http.HttpResponse(success_string)
+
+class DummyRequest(object):
+    """
+    We'll pass this as request instead of a real request object
+    """
+    
+    def __init__(self, POST=None):
+        self.method = "GET"
+        self.POST = None
+        if POST:
+            self.method = "POST"
+            self.POST = POST
+
+class WizardTests(TestCase):
+    def test_step_starts_at_zero(self):
+        """
+        step should be zero for the first form
+        """
+        wizard = WizardClass([WizardPageOneForm, WizardPageTwoForm])
+        request = DummyRequest()
+        wizard(request)
+        self.assertEquals(0, wizard.step)
+        
+    def test_step_increments(self):
+        """
+        step should be incremented when we go to the next page
+        """
+        wizard = WizardClass([WizardPageOneForm, WizardPageTwoForm])
+        request = DummyRequest(POST={"0-field":"test", "wizard_step":"0"})
+        response = wizard(request)
+        self.assertEquals(1, wizard.step)
+        

django/contrib/formtools/templates/formtools/wizard_form.html

@@ +1 @@
+{% extends "base.html" %}
+
+{% block content %}
+<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>
+{% endblock %}