def test_unique_channel_publishing(
redis_client: redis.StrictRedis, redis_connection_pool: redis.ConnectionPool
):
channel_one = "ch:channel_01"
channel_two = "ch:channel_02"
channel_three = "ch:channel_03"
subscriber_one = generate_subscriber_on_channel(redis_client, channel_one)
subscriber_two = generate_subscriber_on_channel(redis_client, channel_two)
subscriber_three = generate_subscriber_on_channel(redis_client, channel_three)
logger_one = generate_logger(
"first_logger",
"INFO",
RedisChannelHandler(channel_one, connection_pool=redis_connection_pool),
)
logger_two = generate_logger(
"second_logger",
"INFO",
RedisChannelHandler(channel_two, connection_pool=redis_connection_pool),
)
all_messages = (
subscriber_one.get_message(ignore_subscribe_messages=True, timeout=1),
subscriber_two.get_message(ignore_subscribe_messages=True, timeout=1),
subscriber_three.get_message(ignore_subscribe_messages=True, timeout=1),
)
for msg in all_messages:
assert is_subscriber_message_empty(msg) is True
# Publish message to first channel
logger_one.info("Test message on channel 1.")
empty_messages = (
subscriber_two.get_message(ignore_subscribe_messages=True, timeout=1),
subscriber_three.get_message(ignore_subscribe_messages=True, timeout=1),
)
for msg in empty_messages:
assert is_subscriber_message_empty(msg) is True
message_one = subscriber_one.get_message(ignore_subscribe_messages=True, timeout=1)
assert is_subscriber_message_empty(message_one) is False
assert isinstance(message_one.get("data").decode("utf-8"), str)
assert "Test message on channel 1." in message_one.get("data").decode("utf-8")
# Publish message to 2nd channel
logger_two.info("Test message on channel 2.")
empty_messages = (
subscriber_one.get_message(ignore_subscribe_messages=True, timeout=1),
subscriber_three.get_message(ignore_subscribe_messages=True, timeout=1),
)
for msg in empty_messages:
assert is_subscriber_message_empty(msg) is True
message_two = subscriber_two.get_message(ignore_subscribe_messages=True, timeout=1)
assert is_subscriber_message_empty(message_two) is False
assert isinstance(message_two.get("data").decode("utf-8"), str)
assert "Test message on channel 2." in message_two.get("data").decode("utf-8")
close_connection_to_channel(subscriber_one, subscriber_two, subscriber_three)
def test_unique_channel_per_handler(redis_connection_pool: redis.ConnectionPool):
handler_one = RedisChannelHandler(
"ch:channel_01", connection_pool=redis_connection_pool
)
handler_two = RedisChannelHandler(
"ch:channel_02", connection_pool=redis_connection_pool
)
handler_three = RedisChannelHandler(
"ch:channel_03", connection_pool=redis_connection_pool
)
assert handler_one.channel != handler_two.channel != handler_three.channel
def test_create_handler_by_connection_pool(redis_connection_pool: redis.ConnectionPool):
handler = RedisChannelHandler("ch:channel", connection_pool=redis_connection_pool)
assert handler is not None
try:
assert handler.redis_client.ping() is not None
except redis.ConnectionError:
pytest.fail("Creating handler by connection pool failed.")
def test_connecting_channel_handler_by_arguments(redis_host):
handler = RedisChannelHandler("ch:channel", host=redis_host, port=6379)
assert handler is not None
try:
assert handler.redis_client.ping() is not None
except redis.ConnectionError:
pytest.fail("Creating handler by connection pool failed.")
def test_create_handler_for_existing_redis_connection(
redis_connection_pool: redis.ConnectionPool,
):
try:
handler_with_real_host = RedisChannelHandler(
"channel", connection_pool=redis_connection_pool
)
generate_subscriber_on_channel(
handler_with_real_host.redis_client, handler_with_real_host.channel
)
except redis.ConnectionError:
pytest.fail("Handler could not be created for existing redis connection.")
def test_send_message_to_redis(redis_connection_pool: redis.ConnectionPool):
handler = RedisChannelHandler(
"ch:test_channel", connection_pool=redis_connection_pool
)
logger = generate_logger("logger", "INFO", handler)
subscriber = generate_subscriber_on_channel(handler.redis_client, handler.channel)
message = subscriber.get_message(ignore_subscribe_messages=True, timeout=1)
assert is_subscriber_message_empty(message) is True
logger.info("Test logger")
message = subscriber.get_message(ignore_subscribe_messages=True, timeout=1)
assert is_subscriber_message_empty(message) is False
assert "Test logger" in message.get("data").decode("utf-8")
close_connection_to_channel(subscriber)
def test_create_handler_for_non_existing_redis_connection():
with pytest.raises(redis.ConnectionError):
handler_with_fake_host = RedisChannelHandler("fake", host="le_cool_host")
generate_subscriber_on_channel(
handler_with_fake_host.redis_client, handler_with_fake_host.channel
)
def test_handler_uses_environment_var_by_default():
host = os.getenv("REDIS_HOST")
handler = RedisChannelHandler("ch:channel", host=host)
assert host is handler.redis_client.connection_pool.connection_kwargs.get("host")