Add suffix to integer field

Author: jamiefalcus

lung_cancer_screening/nhsuk_forms/integer_field.py

@@ -12,6 +12,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/tests/unit/test_integer_field.py

@@ -3,11 +3,10 @@
 from ...integer_field import IntegerField
 
 
-class TestForm(Form):
-    field = IntegerField(label="Abc", initial=1, max_value=10)
-
 class TestIntegerField(TestCase):
     def test_renders_nhs_input(self):
+        class TestForm(Form):
+            field = IntegerField(label="Abc", initial=1, max_value=10)
         self.assertHTMLEqual(
             TestForm()["field"].as_field_group(),
             """
@@ -19,3 +18,20 @@ def test_renders_nhs_input(self):
             """,
         )
 
+    def test_renders_nhs_input_with_suffix(self):
+        class TestForm(Form):
+            field = IntegerField(label="Abc", initial=1, max_value=10, suffix="suffix")
+        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">suffix</div>
+                </div>
+            </div>
+            """,
+        )