mirror of
https://github.com/Significant-Gravitas/AutoGPT.git
synced 2026-01-21 04:57:58 -05:00
fix(backend): manually start the notification service in testing?
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user