SIENTIAPDE-1005
Full coverage Refactor code structure and improve logging configuration; update tests for activities and connectors
This commit is contained in:
@@ -63,3 +63,20 @@ async def test_load_from_kafka(kafka):
|
||||
kafka.kafka_connector.subscribe.assert_called_once_with(["test-topic"])
|
||||
|
||||
kafka.kafka_connector.poll.assert_called_once_with(timeout_ms=1000)
|
||||
|
||||
|
||||
@mark.asyncio
|
||||
async def test_load_from_kafka_empty(kafka):
|
||||
input_data = {"topic": "test-topic"}
|
||||
|
||||
kafka.kafka_connector.poll.return_value = MagicMock(
|
||||
items=MagicMock(return_value=[])
|
||||
)
|
||||
|
||||
result = await kafka.load_from_kafka(input_data)
|
||||
|
||||
assert result == {}
|
||||
|
||||
kafka.kafka_connector.subscribe.assert_called_once_with(["test-topic"])
|
||||
|
||||
kafka.kafka_connector.poll.assert_called_once_with(timeout_ms=1000)
|
||||
|
||||
Reference in New Issue
Block a user