fix key error assessment_reminder_type

praekeltfoundation · Jun 29, 2023 · 401e718 · 401e718
1 parent 597aace
commit 401e718
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/yal/assessments.py b/yal/assessments.py
@@ -412,10 +412,12 @@ async def state_handle_assessment_reminder_response(self):
 
         if inbound == "remind me in 1 hour":
             self.save_metadata("assessment_reminder_hours", "1hour")
+            self.save_metadata("assessment_reminder_type", "later 1hour")
             return await self.go_to_state("state_reschedule_assessment_reminder")
 
         if inbound == "remind me tomorrow":
             self.save_metadata("assessment_reminder_hours", "23hours")
+            self.save_metadata("assessment_reminder_type", "later 23hours")
             return await self.go_to_state("state_reschedule_assessment_reminder")
 
         if inbound == "i m not interested":

diff --git a/yal/tests/test_assessments.py b/yal/tests/test_assessments.py
@@ -293,7 +293,7 @@ async def test_state_handle_assessment_reminder_response_1h(
         "assessment_reminder_hours": "1hour",
         "assessment_reminder_name": "sexual_health_literacy",
         "assessment_reminder_sent": "",
-        "assessment_reminder_type": "reengagement 1hour",
+        "assessment_reminder_type": "later 1hour",
     }