Add sufix to decimal field

Author: jamiefalcus

lung_cancer_screening/nhsuk_forms/decimal_field.py

@@ -11,6 +11,7 @@ def __init__(
     ):
         kwargs["template_name"] = "input.jinja"
 
+        self.suffix = kwargs.pop("suffix", None)
         self.hint = hint
         self.classes = classes
         self.label_classes = label_classes

lung_cancer_screening/nhsuk_forms/jinja2/input.jinja

@@ -15,6 +15,7 @@
     "hint": {
       "text": unbound_field.hint
     } if unbound_field.hint,
+    "suffix": unbound_field.suffix if unbound_field.suffix,
     "id": field.auto_id,
     "name": field.html_name,
     "value": field.value() or "",

lung_cancer_screening/nhsuk_forms/tests/unit/test_decimal_field.py

@@ -0,0 +1,39 @@
+from django.test import TestCase
+from django.forms import Form
+from ...decimal_field import DecimalField
+
+
+class TestDecimalField(TestCase):
+    def test_renders_nhs_input(self):
+        class TestForm(Form):
+            field = DecimalField(label="Abc", initial=1, max_value=10)
+        self.assertHTMLEqual(
+            TestForm()["field"].as_field_group(),
+            """
+            <div class="nhsuk-form-group">
+                <label class="nhsuk-label" for="id_field">
+                    Abc
+                </label><input class="nhsuk-input" id="id_field" name="field" type="number" value="1">
+            </div>
+            """,
+        )
+
+    def test_renders_nhs_input_with_suffix(self):
+        class TestForm(Form):
+            field = DecimalField(label="Abc", initial=1, max_value=10, suffix="cm")
+        print(TestForm()["field"].as_field_group())
+        self.assertHTMLEqual(
+            TestForm()["field"].as_field_group(),
+            """
+            <div class="nhsuk-form-group">
+                <label class="nhsuk-label" for="id_field">
+                    Abc
+                </label>
+                <div class="nhsuk-input__wrapper">
+                    <input class="nhsuk-input" id="id_field" name="field" type="number" value="1">
+                    <div class="nhsuk-input__suffix" aria-hidden="true">cm</div>
+                </div>
+            </>
+            """,
+        )
+