diff --git a/tests/orchestrator/activities/test_activities.py b/tests/orchestrator/activities/test_activities.py index 7c0fbb3..4ebbc18 100644 --- a/tests/orchestrator/activities/test_activities.py +++ b/tests/orchestrator/activities/test_activities.py @@ -95,56 +95,3 @@ def test___init__(mock_formatters_init, mock_slot_manager_init, logger=logger, notification_handler=notification_handler ) - - -@mark.asyncio -@patch('orchestrator.activities.mongo_db.MongoClient') -async def test_prepare_activity(_mock_cluster): - couchbase_config = { - 'connection_string': 'couchbase://localhost', - 'username': 'admin', - 'password': 'password' - } - - mongo_db_config = { - 'connection_string': 'mongodb://localhost:27017', - 'database_name': 'test_db' - } - - redis_config = { - 'host': 'localhost', - 'port': 6379, - 'username': 'admin', - 'password': 'password' - } - - temporal_client = MagicMock() - logger = MagicMock() - notification_handler = MagicMock() - - activities = Activities( - temporal_client=temporal_client, - # couchbase_config=couchbase_config, - redis_config=redis_config, - mongodb_config=mongo_db_config, - logger=logger, - notification_handler=notification_handler - ) - - input_data = { - 'workflow_name': 'test-workflow-name', - 'schedule_name': 'test-schedule-name', - 'model_name': 'test-model-name', - 'model_id': 'test-model-id' - } - - await activities.prepare_activity(input_data) - - assert activities.notification_handler.base_notification.pipeline == input_data[ - 'workflow_name'] - assert activities.notification_handler.base_notification.trigger == input_data[ - 'schedule_name'] - assert activities.notification_handler.base_notification.model_name == input_data[ - 'model_name'] - assert activities.notification_handler.base_notification.model_id == input_data[ - 'model_id']