vitor-aignosi
ad00661516
Refactor tests for Postgres activities and improve error handling - Updated test_postgres.py to enhance the testing of load_custom_query method, including cases for None data and date conversion. - Refactored repeat_last_prediction tests to use mocks for SQLAlchemy session execution. - Added tests for export_data_to_postgres method, covering both success and error scenarios. - Improved the initialization tests for Activities class to ensure proper instantiation of dependencies. - Enhanced test coverage for OPC repository connection validation. - Updated tests for prediction workflows to streamline input handling and improve clarity. - Introduced tests for connectors configuration to validate environment variable handling for MLFlow, OPC, and Postgres. - Added tests for logger utility to ensure default settings are correctly applied.
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
import os
|
|
from unittest.mock import patch
|
|
import logging
|
|
import pytest
|
|
from laborious.utils.logger import get_logger
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_env_vars():
|
|
with patch.dict(os.environ, {}, clear=True):
|
|
yield
|
|
|
|
|
|
@pytest.mark.usefixtures("mock_env_vars")
|
|
@patch('laborious.utils.logger.logging.Formatter')
|
|
@patch('laborious.utils.logger.logging.StreamHandler')
|
|
def test_get_logger_defaults(mock_stream_handler, mock_formatter):
|
|
"""Test logger creation with default settings"""
|
|
# Mock the StreamHandler and Formatter
|
|
|
|
logger = get_logger('test_logger')
|
|
|
|
# Verify logger settings
|
|
assert logger.name == 'test_logger'
|
|
assert logger.level == logging.INFO
|
|
|
|
# Verify handler configuration
|
|
mock_stream_handler.return_value.setLevel.assert_called_once_with('INFO')
|
|
mock_stream_handler.return_value.setFormatter.assert_called_once()
|
|
|
|
# Verify formatter configuration
|
|
mock_formatter.assert_called_once_with(
|
|
'%(asctime)s - %(name)s - %(levelname)s - %(message)s'
|
|
)
|
|
|
|
# Verify handler was added to logger
|
|
assert len(logger.handlers) == 1
|