Add text only error to FreeText state

vaccine/states.py

@@ -216,6 +216,7 @@ def __init__(
         override_answer_name: Optional[str] = None,
         footer: Optional[str] = None,
         header: Optional[str] = None,
+        text_only_error: Optional[str] = None,
     ):
         self.app = app
         self.question = question
@@ -225,8 +226,12 @@ def __init__(
         self.override_answer_name = override_answer_name
         self.footer = footer
         self.header = header
+        self.text_only_error = text_only_error
 
     async def process_message(self, message: Message):
+        if self.text_only_error is not None and message.transport_metadata:
+            if message.transport_metadata.get("message", {}).get("type") != "text":
+                return self.app.send_message(self.text_only_error)
         if self.check is not None:
             if not isinstance(self.check, list):
                 self.check = [self.check]

yal/surveys/location.py

@@ -263,6 +263,7 @@ async def state_location_name_city(self):
             question=question,
             next="state_location_area_type",
             check=nonempty_validator(question),
+            text_only_error=question,
         )
 
     async def state_location_area_type(self):

yal/tests/surveys/test_location.py

@@ -308,6 +308,32 @@ async def test_state_location_name_city(tester: AppTester):
     )
 
 
+@pytest.mark.asyncio
+async def test_state_location_name_city_invalid(tester: AppTester):
+    tester.setup_state("state_location_name_city")
+    await tester.user_input(
+        "",
+        transport_metadata={
+            "message": {
+                "type": "image",
+                "image": {"id": "img1", "mime_type": "image/jpeg"},
+            }
+        },
+    )
+
+    tester.assert_state("state_location_name_city")
+    tester.assert_message(
+        "\n".join(
+            [
+                "*What is the name of the city or town you live in or live closest "
+                "to?*",
+                "",
+                "Please *TYPE* in the name of the city or town.",
+            ]
+        )
+    )
+
+
 @pytest.mark.asyncio
 async def test_state_location_area_type(tester: AppTester):
     tester.setup_state("state_location_name_city")