Merge branch 'feature/SIENTIAPDE-1972-gateway-write-read' into feature/SIENTIAPDE-1972-gateway-wiring
# Conflicts: # README.md # requirements.txt
This commit is contained in:
@@ -3,7 +3,6 @@ from unittest.mock import ANY, AsyncMock, MagicMock, patch
|
||||
from pytest import fixture, mark
|
||||
from sientia_do.notifications.models import NotificationLevel
|
||||
|
||||
import ingestor.metrics as metrics
|
||||
from ingestor.managers.data_manager import DataManager
|
||||
|
||||
metadata = {
|
||||
@@ -48,21 +47,11 @@ def test___init___success(mongodb_repository):
|
||||
metrics_controller=MagicMock(),
|
||||
)
|
||||
|
||||
mongodb_repository.assert_called_once_with(
|
||||
connection_string='mongodb://localhost:27017',
|
||||
database_name='sientia',
|
||||
logger=logger_mock,
|
||||
notification_handler=ANY,
|
||||
metrics_controller=ANY,
|
||||
)
|
||||
logger_mock.info.assert_any_call(
|
||||
'Trying to initializing DataManager with MongoDB servers: mongodb://localhost:27017'
|
||||
)
|
||||
logger_mock.info.assert_any_call(
|
||||
'DataManager initialized with MongoDB servers: mongodb://localhost:27017'
|
||||
)
|
||||
logger_mock.error.assert_not_called()
|
||||
assert data_manager.mongo_repository is not None
|
||||
assert data_manager.connection_string == 'mongodb://localhost:27017'
|
||||
|
||||
|
||||
def test_shutdown(data_manager):
|
||||
@@ -70,7 +59,6 @@ def test_shutdown(data_manager):
|
||||
data_manager.mongo_repository.close = close_mock
|
||||
|
||||
data_manager.shutdown()
|
||||
|
||||
close_mock.assert_called_once()
|
||||
|
||||
|
||||
@@ -102,14 +90,6 @@ async def test_publish(data_manager):
|
||||
document={**data, 'inserted_at': ANY},
|
||||
metadata=metadata['metadata'],
|
||||
)
|
||||
data_manager.emit_metric.assert_called_once_with(
|
||||
metric_object=metrics.TAG_WRITTEN_COUNT,
|
||||
tags={
|
||||
'pod_id': data_manager.pod_id,
|
||||
'tag_name': data['name'],
|
||||
'collection_name': topic,
|
||||
},
|
||||
)
|
||||
|
||||
|
||||
@mark.asyncio
|
||||
|
||||
@@ -183,13 +183,6 @@ async def test_shutdown(ingestor_manager):
|
||||
ingestor_manager.data_manager.shutdown.assert_called_once()
|
||||
|
||||
|
||||
@patch('ingestor.managers.ingestor_manager.asyncio')
|
||||
def test___del__(asyncio_mock, ingestor_manager):
|
||||
ingestor_manager.shutdown = MagicMock()
|
||||
ingestor_manager.__del__()
|
||||
asyncio_mock.run.assert_called_once_with(ingestor_manager.shutdown.return_value)
|
||||
|
||||
|
||||
@mark.asyncio
|
||||
async def test_remove_server(ingestor_manager):
|
||||
server1 = AsyncMock()
|
||||
|
||||
Reference in New Issue
Block a user