SIENTIAPDE-1169

feat: add shutdown tests for Activities and Email classes

- Implemented unit tests for the shutdown methods in the Activities and Email classes to ensure proper resource cleanup.
- Verified that the shutdown processes for MongoDB, PostgreSQL, and Email are called as expected during the shutdown sequence.
This commit is contained in:
vitor-aignosi
2025-08-19 08:54:18 -03:00
parent d065f7f700
commit e2fbc20af0
2 changed files with 43 additions and 0 deletions

View File

@@ -111,3 +111,40 @@ def test___init__(mock_postgres_init,
logger=logger,
notification_handler=notification_handler
)
@patch('orchestrator.activities.mongo_db.MongoDB.__init__')
@patch('orchestrator.activities.temporal_manager.TemporalManager.__init__')
@patch('orchestrator.activities.slot_manager.SlotManager.__init__')
@patch('orchestrator.activities.formatters.Formatters.__init__')
@patch('orchestrator.activities.email.Email.__init__')
@patch('sientia_do.temporal.activities.postgres.Postgres.__init__')
@patch('orchestrator.activities.email.Email.shutdown')
@patch('sientia_do.temporal.activities.postgres.Postgres.close')
@patch('orchestrator.activities.mongo_db.MongoDB.shutdown')
def test_shutdown(mock_mongodb_close,
mock_postgres_shutdown,
mock_email_close,
mock_postgres_init,
mock_email_init,
mock_formatters_init,
mock_slot_manager_init,
mock_temporal_manager_init,
mock_mongodb_init,
):
activities = Activities(
temporal_config=MagicMock(),
redis_config=MagicMock(),
mongodb_config=MagicMock(),
email_config=MagicMock(),
postgres_config=MagicMock(),
logger=MagicMock(),
notification_handler=MagicMock()
)
activities.shutdown()
mock_mongodb_close.assert_called()
mock_postgres_shutdown.assert_called()
mock_email_close.assert_called()

View File

@@ -76,6 +76,12 @@ metadata = {
}
def test_shutdown(email):
email.shutdown()
email.server.quit.assert_called_once()
@mark.asyncio
async def test_build_email_html(email):
email.email_builder.build_email = MagicMock(