SIENTIAPDE-1005
Unit test fixes
This commit is contained in:
@@ -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']
|
||||
|
||||
Reference in New Issue
Block a user