import json import logging import time from datetime import datetime from unittest.mock import MagicMock, patch import pytest from app.storage_manager import StorageManager, get_storage_manager class TestStorageManagerSkeleton: """StorageManager 骨架与降级检测测试。""" def test_init_holds_client_and_engine_references(self): """__init__ 应持有 redis_client / mysql_engine 引用。""" redis_client = MagicMock() mysql_engine = MagicMock() manager = StorageManager(redis_client=redis_client, mysql_engine=mysql_engine) assert manager.redis_client is redis_client assert manager.mysql_engine is mysql_engine def test_init_sets_default_flags(self): """初始化时应将可用性标志设为 False。""" manager = StorageManager(redis_client=None, mysql_engine=None) assert manager.redis_available is False assert manager.mysql_available is False def test_check_redis_returns_cached_result_within_interval(self): """30 秒间隔内应返回缓存的 Redis 可用性结果，不重新探测。""" redis_client = MagicMock() redis_client.ping.side_effect = Exception("redis down") manager = StorageManager(redis_client=redis_client, mysql_engine=None) first = manager.check_redis() redis_client.ping.side_effect = None second = manager.check_redis() assert first is False assert second is False assert redis_client.ping.call_count == 1 def test_check_redis_recovers_after_interval(self): """超过检测间隔后，Redis 恢复应返回 True。""" redis_client = MagicMock() redis_client.ping.side_effect = Exception("redis down") manager = StorageManager( redis_client=redis_client, mysql_engine=None, check_interval=0.1 ) assert manager.check_redis() is False redis_client.ping.side_effect = None time.sleep(0.15) assert manager.check_redis() is True def test_check_redis_false_when_client_is_none(self): """redis_client 为 None 时 check_redis 返回 False。""" manager = StorageManager(redis_client=None, mysql_engine=None) assert manager.check_redis() is False def test_check_mysql_returns_cached_result_within_interval(self): """30 秒间隔内应返回缓存的 MySQL 可用性结果，不重新探测。""" mysql_engine = MagicMock() mysql_engine.connect.side_effect = Exception("mysql down") manager = StorageManager(redis_client=None, mysql_engine=mysql_engine) first = manager.check_mysql() mysql_engine.connect.side_effect = None second = manager.check_mysql() assert first is False assert second is False assert mysql_engine.connect.call_count == 1 def test_check_mysql_recovers_after_interval(self): """超过检测间隔后，MySQL 恢复应返回 True。""" mysql_engine = MagicMock() mysql_engine.connect.side_effect = Exception("mysql down") manager = StorageManager( redis_client=None, mysql_engine=mysql_engine, check_interval=0.1 ) assert manager.check_mysql() is False mysql_engine.connect.side_effect = None time.sleep(0.15) assert manager.check_mysql() is True def test_check_mysql_false_when_engine_is_none(self): """mysql_engine 为 None 时 check_mysql 返回 False。""" manager = StorageManager(redis_client=None, mysql_engine=None) assert manager.check_mysql() is False def test_is_available_true_when_redis_available(self): """Redis 可用时 is_available 返回 True。""" redis_client = MagicMock() manager = StorageManager(redis_client=redis_client, mysql_engine=None) assert manager.is_available() is True def test_is_available_true_when_mysql_available(self): """MySQL 可用时 is_available 返回 True。""" mysql_engine = MagicMock() manager = StorageManager(redis_client=None, mysql_engine=mysql_engine) assert manager.is_available() is True def test_is_available_false_when_both_unavailable(self): """Redis 与 MySQL 均不可用时 is_available 返回 False。""" redis_client = MagicMock() redis_client.ping.side_effect = Exception("redis down") mysql_engine = MagicMock() mysql_engine.connect.side_effect = Exception("mysql down") manager = StorageManager(redis_client=redis_client, mysql_engine=mysql_engine) assert manager.is_available() is False def test_initialize_checks_backends_and_logs(self, caplog): """initialize 应检测后端可用性并输出日志。""" redis_client = MagicMock() mysql_engine = MagicMock() manager = StorageManager(redis_client=redis_client, mysql_engine=mysql_engine) with caplog.at_level(logging.INFO): manager.initialize() assert "Redis 可用" in caplog.text assert "MySQL 可用" in caplog.text def test_initialize_logs_fallback_when_unavailable(self, caplog): """后端不可用时 initialize 应输出降级日志。""" manager = StorageManager(redis_client=None, mysql_engine=None) with caplog.at_level(logging.WARNING): manager.initialize() assert "Redis 不可用" in caplog.text assert "MySQL 不可用" in caplog.text class TestStorageManagerSingleton: """全局单例工厂函数测试。""" def test_get_storage_manager_returns_same_instance(self): """多次调用 get_storage_manager 应返回同一实例。""" first = get_storage_manager() second = get_storage_manager() assert first is second def test_get_storage_manager_returns_storage_manager(self): """get_storage_manager 应返回 StorageManager 实例。""" manager = get_storage_manager() assert isinstance(manager, StorageManager) class TestStorageManagerReadLogic: """StorageManager Redis → MySQL 读取逻辑测试。""" def test_get_market_data_returns_redis_value_on_cache_hit(self): """Redis 命中时直接返回反序列化数据，不访问 MySQL。""" redis_client = MagicMock() mysql_engine = MagicMock() expected = { "current_price": 123.45, "timestamp": "2026-07-04T10:00:00", "candles": [{"open": 100, "close": 110}], } redis_client.get.return_value = json.dumps(expected) manager = StorageManager(redis_client=redis_client, mysql_engine=mysql_engine) result = manager.get_market_data("AG2606", "futures", "5min") assert result == expected redis_client.get.assert_called_once_with("market_data:AG2606:5min") mysql_engine.connect.assert_not_called() def test_get_market_data_falls_back_to_mysql_and_backfills_redis(self): """Redis 未命中时从 MySQL 读取并回填 Redis（TTL 30 天）。""" redis_client = MagicMock() redis_client.get.return_value = None mysql_engine = MagicMock() manager = StorageManager(redis_client=redis_client, mysql_engine=mysql_engine) fetched_at = datetime(2026, 7, 4, 10, 0, 0) record = MagicMock() record.current_price = 123.45 record.fetched_at = fetched_at record.candles_json = json.dumps([{"open": 100, "close": 110}]) mock_query = MagicMock() mock_query.filter_by.return_value.first.return_value = record mock_session = MagicMock() mock_session.return_value = mock_session mock_session.query.return_value = mock_query mock_session.__enter__ = MagicMock(return_value=mock_session) mock_session.__exit__ = MagicMock(return_value=False) mock_sessionmaker = MagicMock(return_value=mock_session) with patch("app.storage_manager.sessionmaker", mock_sessionmaker): result = manager.get_market_data("AG2606", "futures", "5min") expected = { "current_price": 123.45, "timestamp": "2026-07-04T10:00:00", "candles": [{"open": 100, "close": 110}], } assert result == expected mock_sessionmaker.assert_called_once_with(bind=mysql_engine) mock_query.filter_by.assert_called_once_with( symbol="AG2606", data_type="futures", period="5min" ) redis_client.setex.assert_called_once_with( "market_data:AG2606:5min", 2592000, json.dumps(expected, ensure_ascii=False) ) def test_get_market_data_returns_none_when_mysql_unavailable(self): """Redis 和 MySQL 均不可用时返回 None，由调用方降级到 SQLite。""" redis_client = MagicMock() redis_client.ping.side_effect = Exception("redis down") mysql_engine = MagicMock() mysql_engine.connect.side_effect = Exception("mysql down") manager = StorageManager(redis_client=redis_client, mysql_engine=mysql_engine) result = manager.get_market_data("AG2606", "futures", "5min") assert result is None def test_get_symbol_timestamp_returns_redis_value_on_cache_hit(self): """Redis 命中时直接返回合约时间戳数据，不访问 MySQL。""" redis_client = MagicMock() mysql_engine = MagicMock() expected = { "last_refresh_at": "2026-07-04T10:00:00", "refresh_count": 42, } redis_client.get.return_value = json.dumps(expected) manager = StorageManager(redis_client=redis_client, mysql_engine=mysql_engine) result = manager.get_symbol_timestamp("AG2606", "futures") assert result == expected redis_client.get.assert_called_once_with("symbol_timestamps:AG2606") mysql_engine.connect.assert_not_called() def test_get_symbol_timestamp_falls_back_to_mysql_and_backfills_redis(self): """Redis 未命中时从 MySQL 读取时间戳并回填 Redis（TTL 30 天）。""" redis_client = MagicMock() redis_client.get.return_value = None mysql_engine = MagicMock() manager = StorageManager(redis_client=redis_client, mysql_engine=mysql_engine) last_refresh = datetime(2026, 7, 4, 10, 0, 0) record = MagicMock() record.last_refresh_at = last_refresh record.refresh_count = 42 mock_query = MagicMock() mock_query.filter_by.return_value.first.return_value = record mock_session = MagicMock() mock_session.return_value = mock_session mock_session.query.return_value = mock_query mock_session.__enter__ = MagicMock(return_value=mock_session) mock_session.__exit__ = MagicMock(return_value=False) mock_sessionmaker = MagicMock(return_value=mock_session) with patch("app.storage_manager.sessionmaker", mock_sessionmaker): result = manager.get_symbol_timestamp("AG2606", "futures") expected = { "last_refresh_at": "2026-07-04T10:00:00", "refresh_count": 42, } assert result == expected mock_sessionmaker.assert_called_once_with(bind=mysql_engine) mock_query.filter_by.assert_called_once_with(symbol="AG2606", data_type="futures") redis_client.setex.assert_called_once_with( "symbol_timestamps:AG2606", 2592000, json.dumps(expected, ensure_ascii=False) )