Ticket #9200: wizard_tests.patch

home/david/work/django/django-trunk/django/contrib/formtools/tests.py

@@ -1 +1 @@
 from django import forms
+from django.db import models
 from django.contrib.formtools import preview, wizard
 from django import http
 from django.test import TestCase
+from django.test.client import Client
 
+
 success_string = "Done was called!"
 
 class TestFormPreview(preview.FormPreview):
@@ -141 +144 @@
         request = DummyRequest(POST={"0-field":"test", "wizard_step":"0"})
         response = wizard(request)
         self.assertEquals(1, wizard.step)
+        
 
+#
+# SessionWizard tests
+#
+class SessionWizardModel(models.Model):
+    field = models.CharField(max_length=10)
+    
+class SessionWizardPageOneForm(forms.Form):
+    field = forms.CharField(required=True)
+
+class SessionWizardPageTwoForm(forms.ModelForm):
+    class Meta:
+        model = SessionWizardModel
+
+class SessionWizardPageThreeForm(forms.Form):
+    field = forms.CharField()
+
+class SessionWizardDynamicPageForm(forms.Form):
+    field = forms.CharField()
+        
+class SessionWizardClass(wizard.SessionWizard):
+    def get_page_title(self, session, page0):
+        return "Custom Page Title: %s" % str(page0 + 1)
+
+    def process_show_form(self, request, page0, form):
+        return {'form_title' : 'Form %s' % str(page0 + 1)}
+
+    def preprocess_submit_form(self, request, page0, form):
+        if page0 == 1 and request.POST['field'] == "":
+            self._remove_page(request.session, page0)
+            return page0 - 1
+
+    def process_submit_form(self, request, page0, form):
+        if page0 == 2:
+            self._insert_page(request.session, page0 + 1, 
+                              SessionWizardDynamicPageForm)
+    
+    def get_template(self, page0):
+        return "formtools/form.html"
+
+    def done(self, request):
+        return http.HttpResponse(success_string)
+
+class SessionWizardTests(TestCase):
+    urls = 'django.contrib.formtools.test_urls'
+
+    def test_valid_POST(self):
+        """
+        Tests that a post containing valid data will set session values 
+        correctly and redirect to the next page.
+        """
+        response = self.client.post('/sessionwizard/0', {"page0":"0", 
+                                                          "field":"test"})
+        self.assertEquals(302, response.status_code)
+        self.assertEquals("http://testserver/sessionwizard/1", 
+                          response['Location'])
+        session = self.client.session
+        cleaned_data = session['session_wizard_data']['cleaned_data']
+        post_data = session['session_wizard_data']['POST_data']
+        self.assertEquals('test', cleaned_data[0]['field'])
+        self.assertEquals('test', post_data[0]['field'])
+
+    def test_invalid_POST(self):
+        """
+        Tests that a post containing invalid data will set session values 
+        correctly and redisplay the form.
+        """
+        response = self.client.post('/sessionwizard/0', {"page0":"0", 
+                                                          "field":""})
+        self.assertEquals(200, response.status_code)
+        session = self.client.session
+        post_data = session['session_wizard_data']['POST_data']
+        self.assertEquals('', post_data[0]['field'])
+    
+    def test_GET(self):
+        """
+        Tests that a get will display a page properly.
+        """
+        response = self.client.get('/sessionwizard/0')
+        self.assertEquals(200, response.status_code)
+    
+    def test_preprocess_submit_form(self):
+        """
+        Tests the preprocess_submit_form hook of SessionWizard POSTs.
+        The SessionWizardClass is coded to short-circuit a POST for page index 1
+        when form.cleaned_data['field'] == '' by returning a reference to page0 
+        index 0.
+        """
+        response = self.client.post('/sessionwizard/1', {"page0":"1", 
+                                                          "field":""})
+        self.assertEquals(302, response.status_code)
+        self.assertEquals("http://testserver/sessionwizard/0", 
+                          response['Location'])
+    
+    def test_process_submit_form(self):
+        """
+        Tests the process_submit_form hook of SessionWizard POSTs.
+        The SessionWizardClass is coded to insert a new page at index 3 on a 
+        POST for page index 2.
+        """
+        response = self.client.post('/sessionwizard/2', {"page0":"2", 
+                                                          "field":"test"})
+        self.assertEquals(302, response.status_code)
+        self.assertEquals("http://testserver/sessionwizard/3", 
+                          response['Location'])
+        self.assertEquals([SessionWizardPageOneForm, 
+                           SessionWizardPageTwoForm,
+                           SessionWizardPageThreeForm,
+                           SessionWizardDynamicPageForm], 
+            self.client.session['session_wizard_data']['form_list'])
+    
+    def test_process_show_form(self):
+        """
+        Tests the process_show_form hook.  SessionWizardClass is coded to 
+        return a extra_context having a specific 'form_title' attribute.
+        """
+        response = self.client.get('/sessionwizard/0')
+        self.assertEquals(200, response.status_code)
+        self.assertEquals("Form 1", 
+                          response.context[0]['extra_context']['form_title'])
+    
+    def test_validate_all(self):
+        """
+        Submits all forms, with one of them being invalid, and tests that 
+        submitting the last form will catch an invalid form earlier in the 
+        workflow and redisplay it.
+        """ 
+        self.client.post('/sessionwizard/0', {"page0":"0", "field":""})
+        self.client.post('/sessionwizard/1', {"page0":"1", "field":"test1"})
+        self.client.post('/sessionwizard/2', {"page0":"2", "field":"test2"})
+        response = self.client.post('/sessionwizard/3', 
+                                   {"page0":"3", "field":"test3"})
+        self.assertEquals("Form 1", 
+                          response.context[0]['extra_context']['form_title'])
+    
+    def test_done(self):
+        self.client.post('/sessionwizard/0', {"page0":"0", "field":"test0"})
+        self.client.post('/sessionwizard/1', {"page0":"1", "field":"test1"})
+        self.client.post('/sessionwizard/2', {"page0":"2", "field":"test2"})
+        response = self.client.post('/sessionwizard/3', 
+                                   {"page0":"3", "field":"test3"})
+        self.assertEqual(response.content, success_string)