SIENTIAPDE-1174

Implement async test for write_metrics method in Gates activity, enhancing test coverage for metric logging functionality.
This commit is contained in:
vitor-aignosi
2025-08-01 07:40:06 -03:00
parent fd740ca451
commit 40c57ed23d
2 changed files with 16 additions and 0 deletions

View File

@@ -402,3 +402,18 @@ async def test_aggregate_data_raise_exception(gates_fixture):
)
else:
assert False
@pytest.mark.asyncio
@patch('scouter.activities.gates.metrics')
async def test_write_metrics(mock_metrics, gates_fixture):
"""Test write_metrics method."""
input_data = {
'metadata': metadata['metadata']
}
await gates_fixture.write_metrics(input_data)
mock_metrics.LABORIOUS_DATA_WRITTEN_COUNT.labels.assert_called_once_with(
pod_id=gates_fixture.pod_id,
model_name=metadata['metadata']['model_name'],
pipeline_name=metadata['metadata']['workflow_name']
)

View File

@@ -20,6 +20,7 @@ def redis_activity(_mock_redis_init):
activity.redis_client = MagicMock()
activity.logger = logger
activity.notification_handler = notification_handler
activity.pod_id = 'test_pod_id'
return activity