From bfaf1e5fc27d7186bd3e4ec46475c17fa8b0916a Mon Sep 17 00:00:00 2001 From: Nicholas Tindle Date: Thu, 13 Feb 2025 06:48:24 -0600 Subject: [PATCH] fix(backend): manually start the notification service in testing? --- autogpt_platform/backend/backend/util/test.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/autogpt_platform/backend/backend/util/test.py b/autogpt_platform/backend/backend/util/test.py index ed4a343ab3..bc60ce2d26 100644 --- a/autogpt_platform/backend/backend/util/test.py +++ b/autogpt_platform/backend/backend/util/test.py @@ -9,6 +9,7 @@ from backend.data.execution import ExecutionResult, ExecutionStatus from backend.data.model import _BaseCredentials from backend.data.user import create_default_user from backend.executor import DatabaseManager, ExecutionManager, ExecutionScheduler +from backend.notifications.notifications import NotificationManager from backend.server.rest_api import AgentServer from backend.server.utils import get_user_id @@ -21,6 +22,7 @@ class SpinTestServer: self.exec_manager = ExecutionManager() self.agent_server = AgentServer() self.scheduler = ExecutionScheduler() + self.notifications = NotificationManager() @staticmethod def test_get_user_id(): @@ -32,6 +34,7 @@ class SpinTestServer: self.agent_server.__enter__() self.exec_manager.__enter__() self.scheduler.__enter__() + self.notifications.__enter__() await db.connect() await initialize_blocks() @@ -46,6 +49,7 @@ class SpinTestServer: self.exec_manager.__exit__(exc_type, exc_val, exc_tb) self.agent_server.__exit__(exc_type, exc_val, exc_tb) self.db_api.__exit__(exc_type, exc_val, exc_tb) + self.notifications.__exit__(exc_type, exc_val, exc_tb) def setup_dependency_overrides(self): # Override get_user_id for testing