SIENTIAPDE-1005

Unit test fixes
This commit is contained in:
vitor-aignosi
2025-05-19 15:52:39 -03:00
parent 03d62cb5f9
commit cefb5e48ff
4 changed files with 17 additions and 9 deletions

View File

@@ -4,6 +4,7 @@ from scouter.activities.postgres import Postgres
from scouter.activities.redis import Redis
from scouter.activities.kafka import Kafka
from scouter.activities.gates import Gates
from pytest import mark
@patch('scouter.activities.activities.Postgres.__init__')
@@ -91,11 +92,12 @@ def test___init__(mock_gates_init, mock_kafka_init, mock_redis_init, mock_postgr
)
@mark.asyncio
@patch('scouter.activities.activities.Postgres.__init__')
@patch('scouter.activities.activities.Redis.__init__')
@patch('scouter.activities.activities.Kafka.__init__')
def test_prepare_activity(_mock_kafka_init,
_mock_redis_init, _mock_postgres_init):
async def test_prepare_activity(_mock_kafka_init,
_mock_redis_init, _mock_postgres_init):
postgres_config = {
'host': 'localhost',
'port': 5432,
@@ -137,7 +139,7 @@ def test_prepare_activity(_mock_kafka_init,
'model_id': 'test-model-id'
}
activities.prepare_activity(input_data)
await activities.prepare_activity(input_data)
assert activities.notification_handler.base_notification.pipeline_name == input_data[
'workflow_name']