Add S3 object storage protocol support (#1780)

* Initial plan

* Add S3 object storage support with provider selection

Co-authored-by: RockChinQ <45992437+RockChinQ@users.noreply.github.com>

* Fix lint issue: remove unused MagicMock import

Co-authored-by: RockChinQ <45992437+RockChinQ@users.noreply.github.com>

---------

Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com>
Co-authored-by: RockChinQ <45992437+RockChinQ@users.noreply.github.com>
This commit is contained in:
Copilot
2025-11-14 10:09:26 +08:00
committed by GitHub
parent 99f649c6b7
commit 0f10cc62ec
6 changed files with 266 additions and 3 deletions

View File

View File

@@ -0,0 +1,100 @@
"""
Tests for storage manager and provider selection
"""
import pytest
from unittest.mock import Mock, AsyncMock, patch
from pkg.storage.mgr import StorageMgr
from pkg.storage.providers.localstorage import LocalStorageProvider
from pkg.storage.providers.s3storage import S3StorageProvider
class TestStorageProviderSelection:
"""Test storage provider selection based on configuration"""
@pytest.mark.asyncio
async def test_default_to_local_storage(self):
"""Test that local storage is used by default when no config is provided"""
# Mock application
mock_app = Mock()
mock_app.instance_config = Mock()
mock_app.instance_config.data = {}
mock_app.logger = Mock()
storage_mgr = StorageMgr(mock_app)
with patch.object(LocalStorageProvider, 'initialize', new_callable=AsyncMock) as mock_init:
await storage_mgr.initialize()
assert isinstance(storage_mgr.storage_provider, LocalStorageProvider)
mock_init.assert_called_once()
@pytest.mark.asyncio
async def test_explicit_local_storage(self):
"""Test that local storage is used when explicitly configured"""
# Mock application
mock_app = Mock()
mock_app.instance_config = Mock()
mock_app.instance_config.data = {
'storage': {
'use': 'local'
}
}
mock_app.logger = Mock()
storage_mgr = StorageMgr(mock_app)
with patch.object(LocalStorageProvider, 'initialize', new_callable=AsyncMock) as mock_init:
await storage_mgr.initialize()
assert isinstance(storage_mgr.storage_provider, LocalStorageProvider)
mock_init.assert_called_once()
@pytest.mark.asyncio
async def test_s3_storage_provider_selection(self):
"""Test that S3 storage is used when configured"""
# Mock application
mock_app = Mock()
mock_app.instance_config = Mock()
mock_app.instance_config.data = {
'storage': {
'use': 's3',
's3': {
'endpoint_url': 'https://s3.amazonaws.com',
'access_key_id': 'test_key',
'secret_access_key': 'test_secret',
'region': 'us-east-1',
'bucket': 'test-bucket'
}
}
}
mock_app.logger = Mock()
storage_mgr = StorageMgr(mock_app)
with patch.object(S3StorageProvider, 'initialize', new_callable=AsyncMock) as mock_init:
await storage_mgr.initialize()
assert isinstance(storage_mgr.storage_provider, S3StorageProvider)
mock_init.assert_called_once()
@pytest.mark.asyncio
async def test_invalid_storage_type_defaults_to_local(self):
"""Test that invalid storage type defaults to local storage"""
# Mock application
mock_app = Mock()
mock_app.instance_config = Mock()
mock_app.instance_config.data = {
'storage': {
'use': 'invalid_type'
}
}
mock_app.logger = Mock()
storage_mgr = StorageMgr(mock_app)
with patch.object(LocalStorageProvider, 'initialize', new_callable=AsyncMock) as mock_init:
await storage_mgr.initialize()
assert isinstance(storage_mgr.storage_provider, LocalStorageProvider)
mock_init.assert_called_once()
if __name__ == '__main__':
pytest.main([__file__, '-v'])