115 lines
3.9 KiB
Python
115 lines
3.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Unit tests for EventFetcher."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import patch
|
|
|
|
import pandas as pd
|
|
|
|
from mes_dashboard.services.event_fetcher import EventFetcher
|
|
|
|
|
|
def test_cache_key_is_stable_for_sorted_ids():
|
|
key1 = EventFetcher._cache_key("history", ["CID-B", "CID-A", "CID-A"])
|
|
key2 = EventFetcher._cache_key("history", ["CID-A", "CID-B"])
|
|
|
|
assert key1 == key2
|
|
assert key1.startswith("evt:history:")
|
|
|
|
|
|
def test_get_rate_limit_config_supports_env_override(monkeypatch):
|
|
monkeypatch.setenv("EVT_HISTORY_RATE_MAX_REQUESTS", "33")
|
|
monkeypatch.setenv("EVT_HISTORY_RATE_WINDOW_SECONDS", "77")
|
|
|
|
config = EventFetcher._get_rate_limit_config("history")
|
|
|
|
assert config["bucket"] == "event-history"
|
|
assert config["max_attempts"] == 33
|
|
assert config["window_seconds"] == 77
|
|
|
|
|
|
@patch("mes_dashboard.services.event_fetcher.read_sql_df")
|
|
@patch("mes_dashboard.services.event_fetcher.cache_get")
|
|
def test_fetch_events_cache_hit_skips_db(mock_cache_get, mock_read_sql_df):
|
|
mock_cache_get.return_value = {"CID-1": [{"CONTAINERID": "CID-1"}]}
|
|
|
|
result = EventFetcher.fetch_events(["CID-1"], "materials")
|
|
|
|
assert result["CID-1"][0]["CONTAINERID"] == "CID-1"
|
|
mock_read_sql_df.assert_not_called()
|
|
|
|
|
|
@patch("mes_dashboard.services.event_fetcher.cache_set")
|
|
@patch("mes_dashboard.services.event_fetcher.cache_get", return_value=None)
|
|
@patch("mes_dashboard.services.event_fetcher.read_sql_df")
|
|
@patch("mes_dashboard.services.event_fetcher.SQLLoader.load_with_params")
|
|
def test_fetch_events_upstream_history_branch(
|
|
mock_sql_load,
|
|
mock_read_sql_df,
|
|
_mock_cache_get,
|
|
mock_cache_set,
|
|
):
|
|
mock_sql_load.return_value = "SELECT * FROM UPSTREAM"
|
|
mock_read_sql_df.return_value = pd.DataFrame(
|
|
[
|
|
{"CONTAINERID": "CID-1", "WORKCENTER_GROUP": "DB"},
|
|
{"CONTAINERID": "CID-2", "WORKCENTER_GROUP": "WB"},
|
|
]
|
|
)
|
|
|
|
result = EventFetcher.fetch_events(["CID-1", "CID-2"], "upstream_history")
|
|
|
|
assert sorted(result.keys()) == ["CID-1", "CID-2"]
|
|
assert mock_sql_load.call_args.args[0] == "mid_section_defect/upstream_history"
|
|
_, params = mock_read_sql_df.call_args.args
|
|
assert len(params) == 2
|
|
mock_cache_set.assert_called_once()
|
|
assert mock_cache_set.call_args.args[0].startswith("evt:upstream_history:")
|
|
|
|
|
|
@patch("mes_dashboard.services.event_fetcher.cache_set")
|
|
@patch("mes_dashboard.services.event_fetcher.cache_get", return_value=None)
|
|
@patch("mes_dashboard.services.event_fetcher.read_sql_df")
|
|
@patch("mes_dashboard.services.event_fetcher.SQLLoader.load")
|
|
def test_fetch_events_history_branch_replaces_container_filter(
|
|
mock_sql_load,
|
|
mock_read_sql_df,
|
|
_mock_cache_get,
|
|
_mock_cache_set,
|
|
):
|
|
mock_sql_load.return_value = (
|
|
"SELECT * FROM t WHERE h.CONTAINERID = :container_id {{ WORKCENTER_FILTER }}"
|
|
)
|
|
mock_read_sql_df.return_value = pd.DataFrame([])
|
|
|
|
EventFetcher.fetch_events(["CID-1"], "history")
|
|
|
|
sql, params = mock_read_sql_df.call_args.args
|
|
assert "h.CONTAINERID = :container_id" not in sql
|
|
assert "{{ WORKCENTER_FILTER }}" not in sql
|
|
assert params == {"p0": "CID-1"}
|
|
|
|
|
|
@patch("mes_dashboard.services.event_fetcher.cache_set")
|
|
@patch("mes_dashboard.services.event_fetcher.cache_get", return_value=None)
|
|
@patch("mes_dashboard.services.event_fetcher.read_sql_df")
|
|
@patch("mes_dashboard.services.event_fetcher.SQLLoader.load")
|
|
def test_fetch_events_materials_branch_replaces_aliased_container_filter(
|
|
mock_sql_load,
|
|
mock_read_sql_df,
|
|
_mock_cache_get,
|
|
_mock_cache_set,
|
|
):
|
|
mock_sql_load.return_value = (
|
|
"SELECT * FROM t m WHERE m.CONTAINERID = :container_id ORDER BY TXNDATE"
|
|
)
|
|
mock_read_sql_df.return_value = pd.DataFrame([])
|
|
|
|
EventFetcher.fetch_events(["CID-1", "CID-2"], "materials")
|
|
|
|
sql, params = mock_read_sql_df.call_args.args
|
|
assert "m.CONTAINERID = :container_id" not in sql
|
|
assert "IN" in sql.upper()
|
|
assert params == {"p0": "CID-1", "p1": "CID-2"}
|