fix(postgres): Convert postgres to new RunFunctionWaitStrategy

This gets rid of a annoying depcreation warning.

Author: jankatins

modules/postgres/testcontainers/postgres/__init__.py

@@ -15,7 +15,8 @@
 
 from testcontainers.core.generic import DbContainer
 from testcontainers.core.utils import raise_for_deprecated_parameter
-from testcontainers.core.waiting_utils import wait_container_is_ready
+from testcontainers.core.wait_strategies import RunFunctionWaitStrategy
+from testcontainers.core.waiting_utils import WaitStrategyTarget
 
 _UNSET = object()
 
@@ -64,6 +65,7 @@ def __init__(
         self.driver = f"+{driver}" if driver else ""
 
         self.with_exposed_ports(self.port)
+        self.waiting_for(RunFunctionWaitStrategy(_check_postgres_ready))
 
     def _configure(self) -> None:
         self.with_env("POSTGRES_USER", self.username)
@@ -87,15 +89,18 @@ def get_connection_url(self, host: Optional[str] = None, driver: Optional[str] =
             port=self.port,
         )
 
-    @wait_container_is_ready()
-    def _connect(self) -> None:
-        escaped_single_password = self.password.replace("'", "'\"'\"'")
-        result = self.exec(
-            [
-                "sh",
-                "-c",
-                f"PGPASSWORD='{escaped_single_password}' psql --username {self.username} --dbname {self.dbname} --host 127.0.0.1 -c 'select version();'",
-            ]
-        )
-        if result.exit_code:
-            raise ConnectionError("pg_isready is not ready yet")
+
+def _check_postgres_ready(container: WaitStrategyTarget) -> bool:
+    if not isinstance(container, PostgresContainer):
+        raise AssertionError("This check can only wait for postgres containers to start up")
+    escaped_single_password = container.password.replace("'", "'\"'\"'")
+    result = container.exec(
+        [
+            "sh",
+            "-c",
+            f"PGPASSWORD='{escaped_single_password}' psql --username {container.username} --dbname {container.dbname} --host 127.0.0.1 -c 'select version();'",
+        ]
+    )
+    if result.exit_code:
+        raise ConnectionError("pg is not ready yet")
+    return True