From cb674566626b9e9aed68e5cd325f345d720d30ef Mon Sep 17 00:00:00 2001
From: Hlamalani <hlamalani@praekelt.org>
Date: Thu, 27 Jul 2023 15:25:35 +0200
Subject: [PATCH] survey start

---
 yal/main.py                       | 11 +++++++----
 yal/tests/surveys/test_endline.py |  8 ++++++--
 2 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/yal/main.py b/yal/main.py
index 89acfe6a..eb6ca6b9 100644
--- a/yal/main.py
+++ b/yal/main.py
@@ -84,7 +84,12 @@
 QA_RESET_FEEDBACK_TIMESTAMP_KEYWORDS = {"resetfeedbacktimestampobzvmp"}
 EMERGENCY_KEYWORDS = utils.get_keywords("emergency")
 AAQ_KEYWORDS = {"ask a question"}
-EJAF_ENDLINE_SURVEY_KEYWORDS = {"answer", "yes i want to answer", "remind me tomorrow", "i m not interested"}
+EJAF_ENDLINE_SURVEY_KEYWORDS = {
+    "answer",
+    "yes i want to answer",
+    "remind me tomorrow",
+    "i m not interested",
+}
 
 
 class Application(
@@ -180,7 +185,7 @@ async def process_message(self, message):
                 if self.user.metadata.get("onboarding_reminder_sent"):
                     self.user.session_id = None
                     self.state_name = OnboardingApplication.REMINDER_STATE
-    
+
             if keyword in ASSESSMENT_REENGAGEMENT_KEYWORDS:
                 if self.user.metadata.get("assessment_reminder_sent"):
                     self.user.session_id = None
@@ -190,7 +195,6 @@ async def process_message(self, message):
                 self.user.session_id = None
                 self.state_name = "state_baseline_start"
 
-
             baseline_survey_completed = self.user.metadata.get(
                 "baseline_survey_completed"
             )
@@ -206,7 +210,6 @@ async def process_message(self, message):
                 self.save_metadata("assessment_reminder_sent", False)
                 self.save_metadata("assessment_reminder_name", "")
 
-
                 if keyword == "remind me tomorrow":
                     self.user.session_id = None
                     self.state_name = AssessmentApplication.REMINDER_STATE
diff --git a/yal/tests/surveys/test_endline.py b/yal/tests/surveys/test_endline.py
index 57d2d7c2..58e7c6a2 100644
--- a/yal/tests/surveys/test_endline.py
+++ b/yal/tests/surveys/test_endline.py
@@ -92,9 +92,11 @@ async def test_endline_invitation_i_want_to_answer(tester: AppTester, rapidpro_m
 
     assert user.state.name == "state_start_terms"
 
+
 @pytest.mark.asyncio
 async def test_endline_invitation_remind_me_tomorrow(
-    tester: AppTester, rapidpro_mock,
+    tester: AppTester,
+    rapidpro_mock,
 ):
     user = User(
         addr="278201234567",
@@ -115,9 +117,11 @@ async def test_endline_invitation_remind_me_tomorrow(
 
     assert user.state.name == "state_remind_tomorrow"
 
+
 @pytest.mark.asyncio
 async def test_endline_invitation_not_interested(
-    tester: AppTester, rapidpro_mock,
+    tester: AppTester,
+    rapidpro_mock,
 ):
     user = User(
         addr="278201234567",