Merge pull request #768 from praekeltfoundation/demand_gen

Added demand gen keyword
praekeltfoundation · Jan 13, 2025 · 6ca6bf4 · 6ca6bf4
2 parents f545158 + 33f5fc1
commit 6ca6bf4
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/yal/main.py b/yal/main.py
@@ -64,6 +64,7 @@
     "i saw this on facebook",
 }
 TRACKING_KEYWORDS_ROUND_3 = {"youth", "yth", "yuth", "yut", "yoth", "yot", "yoh"}
+TRACKING_KEYWORDS_ROUND_4 = {"registerme"}
 TRACKING_KEYWORDS_TIKTOK = {"sho"}
 OPTOUT_KEYWORDS = {"stop", "opt out", "cancel", "quit"}
 ONBOARDING_REMINDER_KEYWORDS = {
@@ -151,6 +152,7 @@ async def process_message(self, message):
                 or keyword in TRACKING_KEYWORDS
                 or keyword in TRACKING_KEYWORDS_ROUND_2
                 or keyword in TRACKING_KEYWORDS_ROUND_3
+                or keyword in TRACKING_KEYWORDS_ROUND_4
                 or keyword in TRACKING_KEYWORDS_TIKTOK
             ):
                 self.user.session_id = None
@@ -310,6 +312,7 @@ async def state_start(self):
             inbound in TRACKING_KEYWORDS
             or inbound in TRACKING_KEYWORDS_ROUND_2
             or inbound in TRACKING_KEYWORDS_ROUND_3
+            or inbound in TRACKING_KEYWORDS_ROUND_4
             or inbound in TRACKING_KEYWORDS_TIKTOK
         ):
             self.save_answer("state_source_tracking", inbound)
@@ -321,6 +324,7 @@ async def state_start(self):
             or inbound in TRACKING_KEYWORDS
             or inbound in TRACKING_KEYWORDS_ROUND_2
             or inbound in TRACKING_KEYWORDS_ROUND_3
+            or inbound in TRACKING_KEYWORDS_ROUND_4
             or inbound in TRACKING_KEYWORDS_TIKTOK
         ):
             if terms_accepted and onboarding_completed:

diff --git a/yal/tests/test_main.py b/yal/tests/test_main.py
@@ -939,6 +939,20 @@ async def test_tracked_keywords_saved_for_new_user_ads_round_3(
     tester.assert_answer("state_source_tracking", "yoth")
 
 
+@pytest.mark.asyncio
+async def test_tracked_keywords_demand_gen(
+    tester: AppTester, rapidpro_mock, contentrepo_api_mock
+):
+    rapidpro_mock.tstate.contact_fields["onboarding_completed"] = ""
+    rapidpro_mock.tstate.contact_fields["terms_accepted"] = ""
+    await tester.user_input("registerme")
+    tester.assert_state("state_welcome")
+
+    tester.assert_num_messages(1)
+
+    tester.assert_answer("state_source_tracking", "registerme")
+
+
 @pytest.mark.asyncio
 async def test_tracked_keywords_tiktok(
     tester: AppTester, rapidpro_mock, contentrepo_api_mock