Minor fixes

home/export_content_pages.py

@@ -22,6 +22,7 @@
     VariationBlock,
     ViberBlock,
     SMSBlock,
+    USSDBlock,
     WhatsappBlock,
 )
 
@@ -31,7 +32,11 @@
 
 
 MsgBlocks = tuple[
-    WhatsappBlock | None, SMSBlock | None, MessengerBlock | None, ViberBlock | None
+    WhatsappBlock | None,
+    SMSBlock | None,
+    USSDBlock | None,
+    MessengerBlock | None,
+    ViberBlock | None,
 ]
 
 

home/import_content_pages.py

@@ -330,7 +330,7 @@ def add_message_to_shadow_content_page_from_row(
 
         if row.is_ussd_message:
             page.enable_ussd = True
-            page.ussd_body.append(ShadowSMSBlock(message=row.ussd_body))
+            page.ussd_body.append(ShadowUSSDBlock(message=row.ussd_body))
 
         if row.is_messenger_message:
             page.enable_messenger = True

home/models.py

@@ -321,6 +321,14 @@ class Meta:
         icon = "user"
         form_classname = "whatsapp-message-block struct-block"
 
+    def clean(self, value):
+        result = super().clean(value)
+        errors = {}
+
+        if errors:
+            raise StructBlockValidationError(errors)
+        return result
+
 
 class ViberBlock(blocks.StructBlock):
     image = ImageChooserBlock(required=False)

home/tests/test_models.py

@@ -10,6 +10,7 @@
     NextMessageButton,
     PageView,
     WhatsappBlock,
+    USSDBlock,
 )
 
 from .page_builder import PageBuilder, WABlk, WABody
@@ -313,3 +314,21 @@ def test_buttons_char_limit(self):
         with self.assertRaises(StructBlockValidationError) as e:
             GoToPageButton().clean({"title": "a" * 21})
         self.assertEqual(list(e.exception.block_errors.keys()), ["title"])
+
+
+class USSDBlockTests(TestCase):
+    def create_message_value(
+        self,
+        message="",
+    ):
+        return {
+            "message": message,
+        }
+
+    def test_clean_text_char_limit(self):
+        """Text messages should be limited to 160 characters"""
+        USSDBlock().clean(self.create_message_value(message="a" * 160))
+
+        with self.assertRaises(StructBlockValidationError) as e:
+            USSDBlock().clean(self.create_message_value(message="a" * 161))
+        self.assertEqual(list(e.exception.block_errors.keys()), ["message"])