From 8c892582a1eae9e0712654abe7c65bb0e36efeaa Mon Sep 17 00:00:00 2001
From: Hlamalani <hlamalani@praekelt.org>
Date: Thu, 1 Feb 2024 14:25:23 +0200
Subject: [PATCH] test passing with new field

---
 home/models.py                           | 6 ++++++
 home/tests/page_builder.py               | 1 +
 home/tests/test_content_import_export.py | 1 +
 home/tests/test_models.py                | 2 ++
 home/tests/test_page_builder.py          | 4 ++++
 5 files changed, 14 insertions(+)

diff --git a/home/models.py b/home/models.py
index 02a1cc64..b9ffbf49 100644
--- a/home/models.py
+++ b/home/models.py
@@ -268,6 +268,12 @@ class WhatsappBlock(blocks.StructBlock):
         max_num=3,
     )
 
+    footer = blocks.CharBlock(
+        help_text="Footer cannot exceed 60 characters.",
+        required=False,
+        validators=(MaxLengthValidator(60),),
+    )
+
     class Meta:
         icon = "user"
         form_classname = "whatsapp-message-block struct-block"
diff --git a/home/tests/page_builder.py b/home/tests/page_builder.py
index e3b3d39f..f921fbe3 100644
--- a/home/tests/page_builder.py
+++ b/home/tests/page_builder.py
@@ -112,6 +112,7 @@ class WABlk(ContentBlock):
     variation_messages: list[VarMsg] = field(default_factory=list)
     example_values: list[str] = field(default_factory=list)
     buttons: list[Btn] = field(default_factory=list)
+    footer: str | None = None
 
     def to_dict(self) -> dict[str, Any]:
         varmsgs = [vm.to_dict() for vm in self.variation_messages]
diff --git a/home/tests/test_content_import_export.py b/home/tests/test_content_import_export.py
index 31165973..0f83a473 100644
--- a/home/tests/test_content_import_export.py
+++ b/home/tests/test_content_import_export.py
@@ -334,6 +334,7 @@ def add_body_fields(page: DbDict) -> DbDict:
                     "next_prompt": "",
                     "example_values": [],
                     "variation_messages": [],
+                    "footer": None,
                 },
             )
     if "sms_body" in page["fields"]:
diff --git a/home/tests/test_models.py b/home/tests/test_models.py
index dce1405d..f2c5fbc9 100644
--- a/home/tests/test_models.py
+++ b/home/tests/test_models.py
@@ -334,6 +334,7 @@ def create_message_value(
         example_values=None,
         next_prompt="",
         buttons=None,
+        footer="",
     ):
         return {
             "image": image,
@@ -344,6 +345,7 @@ def create_message_value(
             "variation_messages": variation_messages,
             "next_prompt": next_prompt,
             "buttons": buttons or [],
+            "footer": footer,
         }
 
     def create_image(self, width=0, height=0):
diff --git a/home/tests/test_page_builder.py b/home/tests/test_page_builder.py
index f9a34173..6416d30b 100644
--- a/home/tests/test_page_builder.py
+++ b/home/tests/test_page_builder.py
@@ -137,6 +137,7 @@ def test_build_simple_pages() -> None:
                 "buttons": [],
                 "example_values": [],
                 "variation_messages": [],
+                "footer": None,
             },
         ),
     ]
@@ -260,6 +261,7 @@ def test_build_variations() -> None:
                 {"message": "Single male", "variation_restrictions": v_single_male},
                 {"message": "Comp male", "variation_restrictions": v_complicated_male},
             ],
+            "footer": None,
         },
         {
             "message": "Message 2, variable placeholders as well {{0}}",
@@ -271,6 +273,7 @@ def test_build_variations() -> None:
             "variation_messages": [
                 {"message": "Teen", "variation_restrictions": [("age", "15-18")]}
             ],
+            "footer": None,
         },
         {
             "message": "Message 3 with no variation",
@@ -278,6 +281,7 @@ def test_build_variations() -> None:
             "buttons": [("next_message", {"title": "end"})],
             "example_values": [],
             "variation_messages": [],
+            "footer": None,
         },
     ]