diff --git a/tests/activities/test_gates.py b/tests/activities/test_gates.py index dec6b9d..f0057fd 100644 --- a/tests/activities/test_gates.py +++ b/tests/activities/test_gates.py @@ -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'] + ) diff --git a/tests/activities/test_redis.py b/tests/activities/test_redis.py index ce3215e..bed4c10 100644 --- a/tests/activities/test_redis.py +++ b/tests/activities/test_redis.py @@ -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