refactor: improve code formatting and readability in test files
CI / lint-test-build (push) Failing after 1m13s
CI / lint-test-build (push) Failing after 1m13s
This commit is contained in:
@@ -1,10 +1,9 @@
|
||||
"""End-to-end test for configuration management system."""
|
||||
|
||||
from unittest.mock import Mock, patch
|
||||
from unittest.mock import MagicMock, Mock, patch
|
||||
|
||||
from arbitrade.config.service import ConfigurationService
|
||||
from arbitrade.config.settings import Settings
|
||||
from arbitrade.storage.db import DuckDBStore
|
||||
from arbitrade.storage.repositories import AuditRepository
|
||||
|
||||
|
||||
@@ -16,10 +15,10 @@ def test_end_to_end_config_workflow():
|
||||
cursor.fetchone.return_value = None
|
||||
cursor.fetchall.return_value = []
|
||||
cursor.execute.return_value = cursor
|
||||
context = Mock()
|
||||
context.__enter__.return_value = cursor
|
||||
store = Mock(spec=DuckDBStore)
|
||||
store.connect.return_value = context
|
||||
cntx = MagicMock()
|
||||
cntx.__enter__.return_value = cursor
|
||||
store = Mock()
|
||||
store.connect.return_value = cntx
|
||||
audit_repo = Mock(spec=AuditRepository)
|
||||
|
||||
# Create service
|
||||
@@ -30,7 +29,7 @@ def test_end_to_end_config_workflow():
|
||||
assert service.get_last_updated_at() is None
|
||||
|
||||
# Test setting a value
|
||||
with patch("arbitrade.config.service.ConfigSettingRepository") as mock_repo_class:
|
||||
with patch("arbitrade.storage.repositories.ConfigSettingRepository") as mock_repo_class:
|
||||
mock_repo_instance = Mock()
|
||||
mock_repo_class.return_value = mock_repo_instance
|
||||
|
||||
@@ -38,24 +37,20 @@ def test_end_to_end_config_workflow():
|
||||
mock_created_setting = Mock()
|
||||
mock_created_setting.updated_at = "2023-01-01T00:00:00"
|
||||
mock_repo_instance.create_setting.return_value = mock_created_setting
|
||||
mock_repo_instance.get_setting.return_value = None
|
||||
mock_repo_instance.get_latest_updated_at.return_value = None
|
||||
mock_repo_instance.list_settings.return_value = []
|
||||
|
||||
# Set a setting
|
||||
service.set_setting("test_key", "test_value", "test_user")
|
||||
|
||||
# Verify version incremented
|
||||
assert service.get_config_version() == 1
|
||||
|
||||
# Verify setting was retrieved
|
||||
result = service.get_setting("test_key", "default")
|
||||
assert result == "test_value"
|
||||
|
||||
# Verify hot-reload detection works
|
||||
mock_repo_instance.get_latest_updated_at.return_value = "2023-01-01T00:00:00"
|
||||
assert service.is_config_outdated() is True
|
||||
|
||||
# Verify reload works
|
||||
assert service.reload_if_changed() is True
|
||||
assert service.get_config_version() == 2
|
||||
# Verify version incremented
|
||||
assert service.get_config_version() == 1
|
||||
assert service.get_last_updated_at() is not None
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
Reference in New Issue
Block a user