fix: rabbitmq readiness probe (#375)

Add pika AMQPConnectionError to transient errors fixes #348 --------- Co-authored-by: David Ankin <[email protected]>
testcontainers · Mar 5, 2024 · 71cb75b · 71cb75b
1 parent 750e12a
commit 71cb75b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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())