From 36ed0ec99d49b97384ebaa4ded6911e3104d3f0b Mon Sep 17 00:00:00 2001
From: Karl Hobley <karlhobley10@gmail.com>
Date: Thu, 5 Feb 2015 11:46:44 +0000
Subject: [PATCH] Added tests for FieldBlock

---
 wagtail/wagtailadmin/tests/test_blocks.py | 55 +++++++++++++++++++++++
 1 file changed, 55 insertions(+)
 create mode 100644 wagtail/wagtailadmin/tests/test_blocks.py

diff --git a/wagtail/wagtailadmin/tests/test_blocks.py b/wagtail/wagtailadmin/tests/test_blocks.py
new file mode 100644
index 000000000..eb9c74a1f
--- /dev/null
+++ b/wagtail/wagtailadmin/tests/test_blocks.py
@@ -0,0 +1,55 @@
+import unittest
+
+from django import forms
+from django.core.exceptions import ValidationError
+
+from wagtail.wagtailadmin import blocks
+
+
+class TestFieldBlock(unittest.TestCase):
+    def test_charfield_render(self):
+        block = blocks.FieldBlock(forms.CharField())
+        html = block.render("Hello world!")
+
+        self.assertEqual(html, "Hello world!")
+
+    def test_charfield_render_form(self):
+        block = blocks.FieldBlock(forms.CharField())
+        html = block.render_form("Hello world!")
+
+        self.assertIn('<div class="field char_field">', html)
+        self.assertIn('<input id="" name="" type="text" value="Hello world!" />', html)
+
+    def test_charfield_render_form_with_prefix(self):
+        block = blocks.FieldBlock(forms.CharField())
+        html = block.render_form("Hello world!", prefix='foo')
+
+        self.assertIn('<input id="foo" name="foo" type="text" value="Hello world!" />', html)
+
+    def test_charfield_render_form_with_error(self):
+        block = blocks.FieldBlock(forms.CharField())
+        html = block.render_form("Hello world!", error=ValidationError("This field is required."))
+
+        self.assertIn('This field is required.', html)
+
+    @unittest.expectedFailure
+    def test_choicefield_render(self):
+        block = blocks.FieldBlock(forms.ChoiceField(choices=(
+            ('choice-1', "Choice 1"),
+            ('choice-2', "Choice 2"),
+        )))
+        html = block.render('choice-2')
+
+        self.assertEqual(html, "Choice 2")
+
+    def test_choicefield_render_form(self):
+        block = blocks.FieldBlock(forms.ChoiceField(choices=(
+            ('choice-1', "Choice 1"),
+            ('choice-2', "Choice 2"),
+        )))
+        html = block.render_form('choice-2')
+
+        self.assertIn('<div class="field choice_field">', html)
+        self.assertIn('<select id="" name="">', html)
+        self.assertIn('<option value="choice-1">Choice 1</option>', html)
+        self.assertIn('<option value="choice-2" selected="selected">Choice 2</option>', html)