From 7e7885b2b514856de0b53c513f2450201be573a6 Mon Sep 17 00:00:00 2001
From: Karl Hobley <karl@torchbox.com>
Date: Wed, 4 Jun 2014 15:55:25 +0100
Subject: [PATCH] Added tests for wagtailforms page modes

---
 wagtail/wagtailforms/tests.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/wagtail/wagtailforms/tests.py b/wagtail/wagtailforms/tests.py
index 019f992d8..ab65f909b 100644
--- a/wagtail/wagtailforms/tests.py
+++ b/wagtail/wagtailforms/tests.py
@@ -56,6 +56,29 @@ class TestFormSubmission(TestCase):
         self.assertTemplateUsed(response, 'tests/form_page_landing.html')
 
 
+class TestPageModes(TestCase):
+    fixtures = ['test.json']
+
+    def setUp(self):
+        self.form_page = Page.objects.get(url_path='/home/contact-us/').specific
+
+    def test_form(self):
+        response = self.form_page.show_as_mode('form')
+
+        # Check response
+        self.assertContains(response, """<label for="id_your-email">Your email</label>""")
+        self.assertTemplateUsed(response, 'tests/form_page.html')
+        self.assertTemplateNotUsed(response, 'tests/form_page_landing.html')
+
+    def test_landing(self):
+        response = self.form_page.show_as_mode('landing')
+
+        # Check response
+        self.assertContains(response, "Thank you for your feedback.")
+        self.assertTemplateNotUsed(response, 'tests/form_page.html')
+        self.assertTemplateUsed(response, 'tests/form_page_landing.html')
+
+
 class TestFormBuilder(TestCase):
     fixtures = ['test.json']