102 lines
2.8 KiB
Python
102 lines
2.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""Unit tests for isolated slow-query pool path."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import mes_dashboard.core.database as db
|
|
|
|
|
|
@patch.object(db, "oracledb")
|
|
@patch.object(db, "get_slow_engine")
|
|
@patch.object(db, "_get_slow_query_semaphore")
|
|
@patch.object(db, "get_db_runtime_config")
|
|
def test_read_sql_df_slow_uses_slow_pool_when_enabled(
|
|
mock_runtime,
|
|
mock_sem_fn,
|
|
mock_get_slow_engine,
|
|
mock_oracledb,
|
|
):
|
|
"""Slow query should checkout connection from isolated slow pool."""
|
|
mock_runtime.return_value = {
|
|
"slow_pool_enabled": True,
|
|
"slow_call_timeout_ms": 60000,
|
|
"slow_fetchmany_size": 5000,
|
|
"tcp_connect_timeout": 10,
|
|
"retry_count": 1,
|
|
"retry_delay": 1.0,
|
|
}
|
|
|
|
sem = MagicMock()
|
|
sem.acquire.return_value = True
|
|
mock_sem_fn.return_value = sem
|
|
|
|
cursor = MagicMock()
|
|
cursor.description = [("COL_A",), ("COL_B",)]
|
|
cursor.fetchall.return_value = [("v1", "v2")]
|
|
|
|
conn = MagicMock()
|
|
conn.cursor.return_value = cursor
|
|
|
|
engine = MagicMock()
|
|
engine.raw_connection.return_value = conn
|
|
mock_get_slow_engine.return_value = engine
|
|
|
|
df = db.read_sql_df_slow("SELECT 1", {"p0": "x"})
|
|
|
|
assert list(df.columns) == ["COL_A", "COL_B"]
|
|
assert len(df) == 1
|
|
mock_get_slow_engine.assert_called_once()
|
|
mock_oracledb.connect.assert_not_called()
|
|
conn.close.assert_called_once()
|
|
sem.release.assert_called_once()
|
|
|
|
|
|
@patch.object(db, "oracledb")
|
|
@patch.object(db, "get_slow_engine")
|
|
@patch.object(db, "_get_slow_query_semaphore")
|
|
@patch.object(db, "get_db_runtime_config")
|
|
def test_read_sql_df_slow_iter_uses_slow_pool_when_enabled(
|
|
mock_runtime,
|
|
mock_sem_fn,
|
|
mock_get_slow_engine,
|
|
mock_oracledb,
|
|
):
|
|
"""Slow iterator query should checkout connection from isolated slow pool."""
|
|
mock_runtime.return_value = {
|
|
"slow_pool_enabled": True,
|
|
"slow_call_timeout_ms": 60000,
|
|
"slow_fetchmany_size": 2,
|
|
"tcp_connect_timeout": 10,
|
|
"retry_count": 1,
|
|
"retry_delay": 1.0,
|
|
}
|
|
|
|
sem = MagicMock()
|
|
sem.acquire.return_value = True
|
|
mock_sem_fn.return_value = sem
|
|
|
|
cursor = MagicMock()
|
|
cursor.description = [("COL_A",), ("COL_B",)]
|
|
cursor.fetchmany.side_effect = [
|
|
[("r1a", "r1b")],
|
|
[],
|
|
]
|
|
|
|
conn = MagicMock()
|
|
conn.cursor.return_value = cursor
|
|
|
|
engine = MagicMock()
|
|
engine.raw_connection.return_value = conn
|
|
mock_get_slow_engine.return_value = engine
|
|
|
|
batches = list(db.read_sql_df_slow_iter("SELECT 1", {"p0": "x"}, batch_size=2))
|
|
|
|
assert batches == [(["COL_A", "COL_B"], [("r1a", "r1b")])]
|
|
mock_get_slow_engine.assert_called_once()
|
|
mock_oracledb.connect.assert_not_called()
|
|
conn.close.assert_called_once()
|
|
sem.release.assert_called_once()
|
|
|