chore: reinitialize project with vite architecture
This commit is contained in:
beabigegg
313
tests/test_cache.py
Normal file
313
tests/test_cache.py
Normal file
@@ -0,0 +1,313 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""Unit tests for cache module.
|
||||
|
||||
Tests cache read/write functionality and fallback mechanism.
|
||||
"""
|
||||
|
||||
import pytest
|
||||
from unittest.mock import patch, MagicMock
|
||||
import pandas as pd
|
||||
import json
|
||||
|
||||
|
||||
class TestGetCachedWipData:
|
||||
"""Test get_cached_wip_data function."""
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
def reset_redis(self):
|
||||
"""Reset Redis client state and process-level cache."""
|
||||
import mes_dashboard.core.redis_client as rc
|
||||
import mes_dashboard.core.cache as cache
|
||||
rc._REDIS_CLIENT = None
|
||||
# Clear process-level cache to avoid test interference
|
||||
cache._wip_df_cache.clear()
|
||||
yield
|
||||
rc._REDIS_CLIENT = None
|
||||
cache._wip_df_cache.clear()
|
||||
|
||||
def test_returns_none_when_redis_disabled(self):
|
||||
"""Test returns None when Redis is disabled."""
|
||||
import mes_dashboard.core.cache as cache
|
||||
|
||||
with patch.object(cache, 'REDIS_ENABLED', False):
|
||||
result = cache.get_cached_wip_data()
|
||||
assert result is None
|
||||
|
||||
def test_returns_none_when_client_unavailable(self):
|
||||
"""Test returns None when Redis client is unavailable."""
|
||||
import mes_dashboard.core.cache as cache
|
||||
|
||||
with patch.object(cache, 'REDIS_ENABLED', True):
|
||||
with patch.object(cache, 'get_redis_client', return_value=None):
|
||||
result = cache.get_cached_wip_data()
|
||||
assert result is None
|
||||
|
||||
def test_returns_none_when_cache_miss(self, reset_redis):
|
||||
"""Test returns None when cache key doesn't exist."""
|
||||
import mes_dashboard.core.cache as cache
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.get.return_value = None
|
||||
|
||||
with patch.object(cache, 'REDIS_ENABLED', True):
|
||||
with patch.object(cache, 'get_redis_client', return_value=mock_client):
|
||||
result = cache.get_cached_wip_data()
|
||||
assert result is None
|
||||
|
||||
def test_returns_dataframe_from_cache(self, reset_redis):
|
||||
"""Test returns DataFrame when cache hit."""
|
||||
import mes_dashboard.core.cache as cache
|
||||
|
||||
# Create test data as JSON string (what Redis returns with decode_responses=True)
|
||||
test_data = [
|
||||
{'LOTID': 'LOT001', 'QTY': 100, 'WORKORDER': 'WO001'},
|
||||
{'LOTID': 'LOT002', 'QTY': 200, 'WORKORDER': 'WO002'}
|
||||
]
|
||||
cached_json = json.dumps(test_data)
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.get.return_value = cached_json # String, not bytes
|
||||
|
||||
with patch.object(cache, 'REDIS_ENABLED', True):
|
||||
with patch.object(cache, 'get_redis_client', return_value=mock_client):
|
||||
with patch.object(cache, 'get_key', return_value='mes_wip:data'):
|
||||
result = cache.get_cached_wip_data()
|
||||
|
||||
assert result is not None
|
||||
assert isinstance(result, pd.DataFrame)
|
||||
assert len(result) == 2
|
||||
assert 'LOTID' in result.columns
|
||||
|
||||
def test_handles_invalid_json(self, reset_redis):
|
||||
"""Test handles invalid JSON gracefully."""
|
||||
import mes_dashboard.core.cache as cache
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.get.return_value = 'invalid json {'
|
||||
|
||||
with patch.object(cache, 'REDIS_ENABLED', True):
|
||||
with patch.object(cache, 'get_redis_client', return_value=mock_client):
|
||||
with patch.object(cache, 'get_key', return_value='mes_wip:data'):
|
||||
result = cache.get_cached_wip_data()
|
||||
assert result is None
|
||||
|
||||
|
||||
class TestGetCachedSysDate:
|
||||
"""Test get_cached_sys_date function."""
|
||||
|
||||
def test_returns_none_when_redis_disabled(self):
|
||||
"""Test returns None when Redis is disabled."""
|
||||
import mes_dashboard.core.cache as cache
|
||||
|
||||
with patch.object(cache, 'REDIS_ENABLED', False):
|
||||
result = cache.get_cached_sys_date()
|
||||
assert result is None
|
||||
|
||||
def test_returns_sys_date_from_cache(self):
|
||||
"""Test returns SYS_DATE when cache hit."""
|
||||
import mes_dashboard.core.cache as cache
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.get.return_value = '2024-01-15 10:30:00' # String, not bytes
|
||||
|
||||
with patch.object(cache, 'REDIS_ENABLED', True):
|
||||
with patch.object(cache, 'get_redis_client', return_value=mock_client):
|
||||
with patch.object(cache, 'get_key', return_value='mes_wip:meta:sys_date'):
|
||||
result = cache.get_cached_sys_date()
|
||||
assert result == '2024-01-15 10:30:00'
|
||||
|
||||
|
||||
class TestGetCacheUpdatedAt:
|
||||
"""Test get_cache_updated_at function."""
|
||||
|
||||
def test_returns_none_when_redis_disabled(self):
|
||||
"""Test returns None when Redis is disabled."""
|
||||
import mes_dashboard.core.cache as cache
|
||||
|
||||
with patch.object(cache, 'REDIS_ENABLED', False):
|
||||
result = cache.get_cache_updated_at()
|
||||
assert result is None
|
||||
|
||||
def test_returns_updated_at_from_cache(self):
|
||||
"""Test returns updated_at timestamp when cache hit."""
|
||||
import mes_dashboard.core.cache as cache
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.get.return_value = '2024-01-15T10:30:00' # String, not bytes
|
||||
|
||||
with patch.object(cache, 'REDIS_ENABLED', True):
|
||||
with patch.object(cache, 'get_redis_client', return_value=mock_client):
|
||||
with patch.object(cache, 'get_key', return_value='mes_wip:meta:updated_at'):
|
||||
result = cache.get_cache_updated_at()
|
||||
assert result == '2024-01-15T10:30:00'
|
||||
|
||||
|
||||
class TestWipDataWithFallback:
|
||||
"""Test get_wip_data_with_fallback function."""
|
||||
|
||||
def test_uses_cache_when_available(self):
|
||||
"""Test uses cache when data is available."""
|
||||
import mes_dashboard.core.cache as cache
|
||||
|
||||
cached_df = pd.DataFrame({
|
||||
'LOTID': ['LOT001'],
|
||||
'QTY': [100]
|
||||
})
|
||||
|
||||
mock_fallback = MagicMock()
|
||||
|
||||
with patch.object(cache, 'get_cached_wip_data', return_value=cached_df):
|
||||
result = cache.get_wip_data_with_fallback(mock_fallback)
|
||||
|
||||
assert result is not None
|
||||
assert len(result) == 1
|
||||
# Fallback should NOT be called
|
||||
mock_fallback.assert_not_called()
|
||||
|
||||
def test_fallback_when_cache_unavailable(self):
|
||||
"""Test falls back when cache is unavailable."""
|
||||
import mes_dashboard.core.cache as cache
|
||||
|
||||
oracle_df = pd.DataFrame({
|
||||
'LOTID': ['LOT001', 'LOT002'],
|
||||
'QTY': [100, 200]
|
||||
})
|
||||
|
||||
mock_fallback = MagicMock(return_value=oracle_df)
|
||||
|
||||
with patch.object(cache, 'get_cached_wip_data', return_value=None):
|
||||
result = cache.get_wip_data_with_fallback(mock_fallback)
|
||||
|
||||
assert result is not None
|
||||
assert len(result) == 2
|
||||
mock_fallback.assert_called_once()
|
||||
|
||||
|
||||
class TestNoOpCache:
|
||||
"""Test NoOpCache fallback class."""
|
||||
|
||||
def test_noop_cache_get(self):
|
||||
"""Test NoOpCache.get returns None."""
|
||||
from mes_dashboard.core.cache import NoOpCache
|
||||
cache = NoOpCache()
|
||||
result = cache.get('any_key')
|
||||
assert result is None
|
||||
|
||||
def test_noop_cache_set(self):
|
||||
"""Test NoOpCache.set returns None."""
|
||||
from mes_dashboard.core.cache import NoOpCache
|
||||
cache = NoOpCache()
|
||||
result = cache.set('any_key', 'any_value', 300)
|
||||
assert result is None
|
||||
|
||||
|
||||
class TestMemoryTTLCache:
|
||||
"""Test in-memory TTL cache backend."""
|
||||
|
||||
def test_set_and_get_value(self):
|
||||
from mes_dashboard.core.cache import MemoryTTLCache
|
||||
|
||||
cache = MemoryTTLCache()
|
||||
cache.set('k1', {'v': 1}, 10)
|
||||
assert cache.get('k1') == {'v': 1}
|
||||
|
||||
def test_expired_value_returns_none(self):
|
||||
from mes_dashboard.core.cache import MemoryTTLCache
|
||||
|
||||
cache = MemoryTTLCache()
|
||||
cache.set('k2', {'v': 2}, 1)
|
||||
|
||||
with patch('mes_dashboard.core.cache.time.time', return_value=10_000):
|
||||
cache._store['k2'] = ({'v': 2}, 9_999)
|
||||
assert cache.get('k2') is None
|
||||
|
||||
|
||||
class TestCreateDefaultCacheBackend:
|
||||
"""Test default cache backend factory."""
|
||||
|
||||
def test_returns_layered_cache_without_redis(self):
|
||||
import mes_dashboard.core.cache as cache
|
||||
|
||||
with patch.object(cache, 'redis_available', return_value=False):
|
||||
backend = cache.create_default_cache_backend()
|
||||
backend.set('factory-key', {'x': 1}, 30)
|
||||
assert backend.get('factory-key') == {'x': 1}
|
||||
|
||||
|
||||
class TestLayeredCacheTelemetry:
|
||||
"""Telemetry behavior for layered route cache."""
|
||||
|
||||
def test_l1_only_degraded_mode_visibility(self):
|
||||
from mes_dashboard.core.cache import MemoryTTLCache, LayeredCache
|
||||
|
||||
backend = LayeredCache(l1=MemoryTTLCache(), l2=None, redis_expected=True)
|
||||
backend.set('k1', {'v': 1}, 30)
|
||||
assert backend.get('k1') == {'v': 1} # L1 hit
|
||||
assert backend.get('missing') is None # miss
|
||||
|
||||
telemetry = backend.telemetry()
|
||||
assert telemetry['mode'] == 'l1-only'
|
||||
assert telemetry['degraded'] is True
|
||||
assert telemetry['l1_hits'] >= 1
|
||||
assert telemetry['misses'] >= 1
|
||||
|
||||
def test_l1_l2_hit_and_rates(self):
|
||||
from mes_dashboard.core.cache import MemoryTTLCache, LayeredCache
|
||||
|
||||
class FakeL2:
|
||||
def __init__(self):
|
||||
self.store = {'cold': {'from': 'l2'}}
|
||||
|
||||
def get(self, key):
|
||||
return self.store.get(key)
|
||||
|
||||
def set(self, key, value, ttl):
|
||||
self.store[key] = value
|
||||
|
||||
def telemetry(self):
|
||||
return {'error_count': 0}
|
||||
|
||||
backend = LayeredCache(l1=MemoryTTLCache(), l2=FakeL2(), redis_expected=True)
|
||||
assert backend.get('cold') == {'from': 'l2'} # L2 hit then warm L1
|
||||
assert backend.get('cold') == {'from': 'l2'} # L1 hit
|
||||
|
||||
telemetry = backend.telemetry()
|
||||
assert telemetry['mode'] == 'l1+l2'
|
||||
assert telemetry['degraded'] is False
|
||||
assert telemetry['l2_hits'] >= 1
|
||||
assert telemetry['l1_hits'] >= 1
|
||||
assert telemetry['reads_total'] >= 2
|
||||
|
||||
|
||||
class TestIsCacheAvailable:
|
||||
"""Test is_cache_available function."""
|
||||
|
||||
def test_returns_false_when_disabled(self):
|
||||
"""Test returns False when Redis is disabled."""
|
||||
import mes_dashboard.core.cache as cache
|
||||
|
||||
with patch.object(cache, 'REDIS_ENABLED', False):
|
||||
result = cache.is_cache_available()
|
||||
assert result is False
|
||||
|
||||
def test_returns_false_when_no_client(self):
|
||||
"""Test returns False when no Redis client."""
|
||||
import mes_dashboard.core.cache as cache
|
||||
|
||||
with patch.object(cache, 'REDIS_ENABLED', True):
|
||||
with patch.object(cache, 'get_redis_client', return_value=None):
|
||||
result = cache.is_cache_available()
|
||||
assert result is False
|
||||
|
||||
def test_returns_true_when_data_exists(self):
|
||||
"""Test returns True when data exists in Redis."""
|
||||
import mes_dashboard.core.cache as cache
|
||||
|
||||
mock_client = MagicMock()
|
||||
mock_client.exists.return_value = 1
|
||||
|
||||
with patch.object(cache, 'REDIS_ENABLED', True):
|
||||
with patch.object(cache, 'get_redis_client', return_value=mock_client):
|
||||
with patch.object(cache, 'get_key', return_value='mes_wip:data'):
|
||||
result = cache.is_cache_available()
|
||||
assert result is True
|
||||
Reference in New Issue
Block a user