SIENTIAPDE-1325
Update environment configuration and refactor activities to include metrics controller. Remove Kafka settings and adjust Redis and MongoDB initialization. Update tests to reflect changes in initialization and metrics tracking.
This commit is contained in:
@@ -1,5 +1,5 @@
|
||||
from datetime import datetime
|
||||
from unittest.mock import ANY, MagicMock, patch
|
||||
from unittest.mock import ANY, AsyncMock, MagicMock, patch
|
||||
|
||||
from pytest import fixture, mark
|
||||
from sientia_do.notifications.models import NotificationLevel
|
||||
@@ -14,12 +14,13 @@ def test_mongodb___init__(mock_mongodb_repository):
|
||||
|
||||
logger = MagicMock()
|
||||
notification_handler = MagicMock()
|
||||
|
||||
metrics_controller = MagicMock()
|
||||
mongo = MongoDB(
|
||||
connection_string='mongodb://localhost:27017',
|
||||
database_name='test_db',
|
||||
logger=logger,
|
||||
notification_handler=notification_handler,
|
||||
metrics_controller=metrics_controller,
|
||||
)
|
||||
|
||||
mock_mongodb_repository.assert_called_once_with(
|
||||
@@ -27,6 +28,7 @@ def test_mongodb___init__(mock_mongodb_repository):
|
||||
database_name='test_db',
|
||||
logger=logger,
|
||||
notification_handler=notification_handler,
|
||||
metrics_controller=metrics_controller,
|
||||
)
|
||||
|
||||
assert mongo.mongodb_repository is not None
|
||||
@@ -42,11 +44,8 @@ def mongodb_activity(mock_mongodb_repository):
|
||||
database_name='test_db',
|
||||
logger=MagicMock(),
|
||||
notification_handler=MagicMock(),
|
||||
metrics_controller=AsyncMock(),
|
||||
)
|
||||
|
||||
mongo.logger = MagicMock()
|
||||
mongo.notification_handler = MagicMock()
|
||||
mongo.pod_id = 'localhost'
|
||||
return mongo
|
||||
|
||||
|
||||
@@ -69,7 +68,7 @@ def test_del(mongodb_activity):
|
||||
async def test_load_latest_data_none_last_data_timestamp(mongodb_activity):
|
||||
"""Test load_latest_data"""
|
||||
|
||||
mongodb_activity.mongodb_repository.find.return_value = [
|
||||
mongodb_activity.mongodb_repository.find = AsyncMock(return_value = [
|
||||
{
|
||||
'name': 'test1',
|
||||
'value': 1,
|
||||
@@ -77,7 +76,7 @@ async def test_load_latest_data_none_last_data_timestamp(mongodb_activity):
|
||||
'2023-01-01 12:00:00.000000+0000', DATETIME_FORMAT_MS_WITH_TZ
|
||||
),
|
||||
}
|
||||
]
|
||||
])
|
||||
|
||||
result = await mongodb_activity.load_latest_data(
|
||||
{
|
||||
@@ -106,7 +105,7 @@ async def test_load_latest_data_none_last_data_timestamp(mongodb_activity):
|
||||
async def test_load_latest_data_not_none_last_data_timestamp(mongodb_activity):
|
||||
"""Test load_latest_data"""
|
||||
|
||||
mongodb_activity.mongodb_repository.find.return_value = [
|
||||
mongodb_activity.mongodb_repository.find = AsyncMock(return_value = [
|
||||
{
|
||||
'name': 'test1',
|
||||
'value': 1,
|
||||
@@ -114,7 +113,7 @@ async def test_load_latest_data_not_none_last_data_timestamp(mongodb_activity):
|
||||
'2023-01-01 12:00:00.000000+0000', DATETIME_FORMAT_MS_WITH_TZ
|
||||
),
|
||||
}
|
||||
]
|
||||
])
|
||||
|
||||
result = await mongodb_activity.load_latest_data(
|
||||
{
|
||||
|
||||
Reference in New Issue
Block a user