Recursive broadcast test fix (#1557)

* Recursive broadcast fix * Recursive broadcast test case fix * Recursive broadcast test case fix * Recursive broadcast test case fix * Recursive broadcast test case fix * Recursive broadcast test case fix * Recursive broadcast test case fix
digiteinfotech · Oct 1, 2024 · 9579f4f · 9579f4f
1 parent 1ae9ba3
commit 9579f4f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/kairon/shared/chat/models.py b/kairon/shared/chat/models.py
@@ -81,6 +81,8 @@ class MessageBroadcastRequest(BaseModel):
     scheduler_config: SchedulerConfiguration = None
     recipients_config: RecipientsConfiguration = None
     template_config: List[TemplateConfiguration] = None
+    template_name: str = None
+    language_code: str = None
     pyscript: str = None
 
     @root_validator

diff --git a/tests/unit_test/events/scheduler_test.py b/tests/unit_test/events/scheduler_test.py
@@ -60,9 +60,7 @@ def test_add_schedule(self, mock_add_job, mock_scheduler, mock_collection):
             assert args[2]
 
             assert args[3][0] == 'message_broadcast'
-            assert args[3][1] == 'Event'
-            assert args[3][2] == {'bot': 'test_bot', 'user': 'test_user', 'event_id': event_id}
-
+            assert args[3][1] == {'bot': 'test_bot', 'user': 'test_user', 'event_id': event_id}
 
     @patch("apscheduler.schedulers.background.BackgroundScheduler.get_jobs", autospec=True)
     @patch("apscheduler.schedulers.background.BackgroundScheduler.start", autospec=True)