fix: rabbitmq readiness probe

Add pika AMQPConnectionError to transient errors

Author: jonathandannenberg

rabbitmq/testcontainers/rabbitmq/__init__.py

@@ -23,8 +23,15 @@ class RabbitMqContainer(DockerContainer):
             ...    connection = pika.BlockingConnection(rabbitmq.get_connection_params())
             ...    channel = connection.channel()
     """
-    def __init__(self, image: str = "rabbitmq:latest", port: Optional[int] = None,
-                 username: Optional[str] = None, password: Optional[str] = None, **kwargs) -> None:
+
+    def __init__(
+        self,
+        image: str = "rabbitmq:latest",
+        port: Optional[int] = None,
+        username: Optional[str] = None,
+        password: Optional[str] = None,
+        **kwargs
+    ) -> None:
         """Initialize the RabbitMQ test container.
 
         Args:
@@ -43,7 +50,9 @@ def __init__(self, image: str = "rabbitmq:latest", port: Optional[int] = None,
         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())
@@ -58,7 +67,9 @@ def get_connection_params(self) -> pika.ConnectionParameters:
         For more details see:
         https://pika.readthedocs.io/en/latest/modules/parameters.html
         """
-        credentials = pika.PlainCredentials(username=self.username, password=self.password)
+        credentials = pika.PlainCredentials(
+            username=self.username, password=self.password
+        )
 
         return pika.ConnectionParameters(
             host=self.get_container_host_ip(),