SIENTIAPDE-1110

refactor: remove unused test_prepare_activity function from test_activities.py
This commit is contained in:
vitor-aignosi
2025-07-08 16:39:35 -03:00
parent 5d4faab557
commit e1f1d3e3f4

View File

@@ -95,56 +95,3 @@ def test___init__(mock_formatters_init, mock_slot_manager_init,
logger=logger, logger=logger,
notification_handler=notification_handler 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']