SIENTIAPDE-1110
refactor: remove unused test_prepare_activity function from test_activities.py
This commit is contained in:
@@ -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']
|
|
||||||
|
|||||||
Reference in New Issue
Block a user