4208412eca
Add temporary functiontal tests, that must be removed soon. Refactor tests: Remove outdated resource manager tests and add functional tests with Docker and Kafka integration - Deleted existing unit tests for ResourceManager. - Introduced new functional tests for single node operations with Redis and Kafka. - Added Docker Compose setup for test environment. - Implemented fixtures for Redis and Kafka consumers. - Created comprehensive tests for data publishing and slot management. - Added unit tests for DataManager and IngestorManager with mocked dependencies. - Included tests for OpcManager covering connection, subscription, and data change notifications.
69 lines
1.7 KiB
Python
69 lines
1.7 KiB
Python
from unittest.mock import MagicMock, patch
|
|
from pytest import fixture
|
|
|
|
from ingestor.managers.data_manager import DataManager
|
|
|
|
|
|
@fixture
|
|
@patch("ingestor.managers.data_manager.KafkaProducer")
|
|
def data_manager(kafka):
|
|
return DataManager(
|
|
kafka_servers="localhost:9092",
|
|
logger=MagicMock()
|
|
)
|
|
|
|
|
|
def test___del__(data_manager):
|
|
flush_mock = MagicMock()
|
|
close_mock = MagicMock()
|
|
|
|
data_manager.kafka_producer.flush = flush_mock
|
|
data_manager.kafka_producer.close = close_mock
|
|
|
|
data_manager.__del__()
|
|
flush_mock.assert_called_once()
|
|
close_mock.assert_called_once()
|
|
|
|
|
|
def test_delivery_report(data_manager):
|
|
msg = MagicMock()
|
|
msg.topic = "test_topic"
|
|
msg.partition = 0
|
|
msg.offset = 1
|
|
|
|
data_manager.delivery_report(msg)
|
|
|
|
data_manager.logger.info.assert_called_once_with(
|
|
f"Record successfully produced to {msg.topic} [{msg.partition}] at offset {msg.offset}"
|
|
)
|
|
|
|
|
|
def test_delivery_error(data_manager):
|
|
err = "Test error"
|
|
data_manager.delivery_error(err)
|
|
|
|
data_manager.logger.error.assert_called_once_with(
|
|
f"Delivery failed for record : {err}"
|
|
)
|
|
|
|
|
|
def test_publish(data_manager):
|
|
topic = "test_topic"
|
|
data = {"key": "value"}
|
|
|
|
# Mock the send method of the Kafka producer
|
|
send_mock = MagicMock()
|
|
data_manager.kafka_producer.send = send_mock
|
|
|
|
# Call the publish method
|
|
data_manager.publish(topic, data)
|
|
|
|
# Check if the send method was called with the correct arguments
|
|
send_mock.assert_called_once_with(
|
|
topic=topic, value=data
|
|
)
|
|
|
|
send_mock.return_value.add_callback.assert_called_once()
|
|
|
|
data_manager.kafka_producer.flush.assert_called_once()
|