From 71cb75b281df55ece4d5caf5d487059a7f38c34f Mon Sep 17 00:00:00 2001 From: jondan <48798545+jonathandannenberg@users.noreply.github.com> Date: Tue, 5 Mar 2024 23:08:49 +0100 Subject: [PATCH] fix: rabbitmq readiness probe (#375) Add pika AMQPConnectionError to transient errors fixes #348 --------- Co-authored-by: David Ankin --- modules/rabbitmq/testcontainers/rabbitmq/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/rabbitmq/testcontainers/rabbitmq/__init__.py b/modules/rabbitmq/testcontainers/rabbitmq/__init__.py index 6c26518c..0a548602 100644 --- a/modules/rabbitmq/testcontainers/rabbitmq/__init__.py +++ b/modules/rabbitmq/testcontainers/rabbitmq/__init__.py @@ -51,7 +51,7 @@ def __init__( self.with_env("RABBITMQ_DEFAULT_USER", self.username) self.with_env("RABBITMQ_DEFAULT_PASS", self.password) - @wait_container_is_ready(pika.exceptions.IncompatibleProtocolError) + @wait_container_is_ready(pika.exceptions.IncompatibleProtocolError, pika.exceptions.AMQPConnectionError) def readiness_probe(self) -> bool: """Test if the RabbitMQ broker is ready.""" connection = pika.BlockingConnection(self.get_connection_params())