mirror of
https://gitlab.com/allianceauth/allianceauth.git
synced 2026-02-12 18:16:24 +01:00
30 lines
1013 B
Python
30 lines
1013 B
Python
from unittest import TestCase
|
|
from unittest.mock import patch
|
|
|
|
from redis import RedisError
|
|
|
|
from allianceauth.authentication.task_statistics.helpers import (
|
|
_RedisStub,
|
|
get_redis_client_or_stub,
|
|
)
|
|
|
|
MODULE_PATH = "allianceauth.authentication.task_statistics.helpers"
|
|
|
|
|
|
class TestGetRedisClient(TestCase):
|
|
def test_should_return_mock_if_redis_not_available_1(self):
|
|
# when
|
|
with patch(MODULE_PATH + ".get_redis_client") as mock_get_master_client:
|
|
mock_get_master_client.return_value.ping.side_effect = RedisError
|
|
result = get_redis_client_or_stub()
|
|
# then
|
|
self.assertIsInstance(result, _RedisStub)
|
|
|
|
def test_should_return_mock_if_redis_not_available_2(self):
|
|
# when
|
|
with patch(MODULE_PATH + ".get_redis_client") as mock_get_master_client:
|
|
mock_get_master_client.return_value.ping.return_value = False
|
|
result = get_redis_client_or_stub()
|
|
# then
|
|
self.assertIsInstance(result, _RedisStub)
|