3f538865b5
* lift node version restraint to allow more recent version if already installed * add node test for latest version * change python version * use purple for debug logs * update workflow * add playwright dev dependency * update workflow * change test * oops * improve test * update test * fix tests * mv units tests to a subfolder * reorganize tests * fix install * update test_state * revert node changes and only keep new tests organization * move integration tests in tests/integration * fix integration workflow * fix dockerfile workflow * fix dockerfile workflow 2 * fix shared_state
107 lines
3.6 KiB
Python
107 lines
3.6 KiB
Python
import json
|
|
from unittest.mock import MagicMock, Mock
|
|
|
|
import pytest
|
|
import sqlalchemy
|
|
from redis.exceptions import RedisError
|
|
|
|
from reflex.app import health
|
|
from reflex.model import get_db_status
|
|
from reflex.utils.prerequisites import get_redis_status
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
"mock_redis_client, expected_status",
|
|
[
|
|
# Case 1: Redis client is available and responds to ping
|
|
(Mock(ping=lambda: None), True),
|
|
# Case 2: Redis client raises RedisError
|
|
(Mock(ping=lambda: (_ for _ in ()).throw(RedisError)), False),
|
|
# Case 3: Redis client is not used
|
|
(None, None),
|
|
],
|
|
)
|
|
async def test_get_redis_status(mock_redis_client, expected_status, mocker):
|
|
# Mock the `get_redis_sync` function to return the mock Redis client
|
|
mock_get_redis_sync = mocker.patch(
|
|
"reflex.utils.prerequisites.get_redis_sync", return_value=mock_redis_client
|
|
)
|
|
|
|
# Call the function
|
|
status = await get_redis_status()
|
|
|
|
# Verify the result
|
|
assert status == expected_status
|
|
mock_get_redis_sync.assert_called_once()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
"mock_engine, execute_side_effect, expected_status",
|
|
[
|
|
# Case 1: Database is accessible
|
|
(MagicMock(), None, True),
|
|
# Case 2: Database connection error (OperationalError)
|
|
(
|
|
MagicMock(),
|
|
sqlalchemy.exc.OperationalError("error", "error", "error"),
|
|
False,
|
|
),
|
|
],
|
|
)
|
|
async def test_get_db_status(mock_engine, execute_side_effect, expected_status, mocker):
|
|
# Mock get_engine to return the mock_engine
|
|
mock_get_engine = mocker.patch("reflex.model.get_engine", return_value=mock_engine)
|
|
|
|
# Mock the connection and its execute method
|
|
if mock_engine:
|
|
mock_connection = mock_engine.connect.return_value.__enter__.return_value
|
|
if execute_side_effect:
|
|
# Simulate execute method raising an exception
|
|
mock_connection.execute.side_effect = execute_side_effect
|
|
else:
|
|
# Simulate successful execute call
|
|
mock_connection.execute.return_value = None
|
|
|
|
# Call the function
|
|
status = await get_db_status()
|
|
|
|
# Verify the result
|
|
assert status == expected_status
|
|
mock_get_engine.assert_called_once()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
@pytest.mark.parametrize(
|
|
"db_status, redis_status, expected_status, expected_code",
|
|
[
|
|
# Case 1: Both services are connected
|
|
(True, True, {"status": True, "db": True, "redis": True}, 200),
|
|
# Case 2: Database not connected, Redis connected
|
|
(False, True, {"status": False, "db": False, "redis": True}, 503),
|
|
# Case 3: Database connected, Redis not connected
|
|
(True, False, {"status": False, "db": True, "redis": False}, 503),
|
|
# Case 4: Both services not connected
|
|
(False, False, {"status": False, "db": False, "redis": False}, 503),
|
|
# Case 5: Database Connected, Redis not used
|
|
(True, None, {"status": True, "db": True, "redis": False}, 200),
|
|
],
|
|
)
|
|
async def test_health(db_status, redis_status, expected_status, expected_code, mocker):
|
|
# Mock get_db_status and get_redis_status
|
|
mocker.patch("reflex.app.get_db_status", return_value=db_status)
|
|
mocker.patch(
|
|
"reflex.utils.prerequisites.get_redis_status", return_value=redis_status
|
|
)
|
|
|
|
# Call the async health function
|
|
response = await health()
|
|
|
|
print(json.loads(response.body))
|
|
print(expected_status)
|
|
|
|
# Verify the response content and status code
|
|
assert response.status_code == expected_code
|
|
assert json.loads(response.body) == expected_status
|