68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Tests for scrap_reason_exclusion_cache service."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
from unittest.mock import MagicMock
|
|
|
|
import pandas as pd
|
|
|
|
from mes_dashboard.services import scrap_reason_exclusion_cache as cache
|
|
|
|
|
|
def _reset_cache_state():
|
|
with cache._CACHE_LOCK:
|
|
cache._CACHE["reasons"] = set()
|
|
cache._CACHE["updated_at"] = None
|
|
cache._CACHE["loaded"] = False
|
|
cache._CACHE["source"] = None
|
|
|
|
|
|
def test_refresh_cache_loads_enabled_reason_codes(monkeypatch):
|
|
_reset_cache_state()
|
|
|
|
monkeypatch.setattr(cache, "try_acquire_lock", lambda *_args, **_kwargs: True)
|
|
monkeypatch.setattr(cache, "release_lock", lambda *_args, **_kwargs: None)
|
|
monkeypatch.setattr(cache, "get_redis_client", lambda: None)
|
|
monkeypatch.setattr(
|
|
cache,
|
|
"read_sql_df",
|
|
lambda _sql: pd.DataFrame({"REASON_NAME": ["358", " 160 ", "bonus_adjust"]}),
|
|
)
|
|
|
|
assert cache.refresh_cache(force=True) is True
|
|
assert cache.get_excluded_reasons() == {"358", "160", "BONUS_ADJUST"}
|
|
|
|
|
|
def test_refresh_cache_falls_back_to_redis_when_oracle_fails(monkeypatch):
|
|
_reset_cache_state()
|
|
|
|
redis_client = MagicMock()
|
|
redis_client.get.side_effect = [json.dumps(["A01", "b02"]), "2026-02-13T00:00:00"]
|
|
|
|
monkeypatch.setattr(cache, "try_acquire_lock", lambda *_args, **_kwargs: True)
|
|
monkeypatch.setattr(cache, "release_lock", lambda *_args, **_kwargs: None)
|
|
monkeypatch.setattr(cache, "get_redis_client", lambda: redis_client)
|
|
|
|
def _raise(_sql):
|
|
raise RuntimeError("oracle unavailable")
|
|
|
|
monkeypatch.setattr(cache, "read_sql_df", _raise)
|
|
|
|
assert cache.refresh_cache(force=True) is True
|
|
assert cache.get_excluded_reasons() == {"A01", "B02"}
|
|
|
|
|
|
def test_get_excluded_reasons_uses_redis_for_lazy_bootstrap(monkeypatch):
|
|
_reset_cache_state()
|
|
|
|
redis_client = MagicMock()
|
|
redis_client.get.side_effect = [json.dumps(["X1", "x2"]), "2026-02-13T12:00:00"]
|
|
monkeypatch.setattr(cache, "get_redis_client", lambda: redis_client)
|
|
monkeypatch.setattr(cache, "refresh_cache", lambda force=False: True)
|
|
|
|
reasons = cache.get_excluded_reasons(force_refresh=False)
|
|
|
|
assert reasons == {"X1", "X2"}
|