fix(postgres): Convert postgres to new RunFunctionWaitStrategy

This gets rid of a annoying depcreation warning.

Author: jankatins

modules/postgres/testcontainers/postgres/__init__.py

@@ -11,11 +11,12 @@
 #    License for the specific language governing permissions and limitations
 #    under the License.
 import os
-from typing import Optional
+from typing import Optional, cast
 
 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,17 @@ 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:
+    wrapped = cast("PostgresContainer", container.get_wrapped_container())
+    escaped_single_password = wrapped.password.replace("'", "'\"'\"'")
+    result = wrapped.exec(
+        [
+            "sh",
+            "-c",
+            f"PGPASSWORD='{escaped_single_password}' psql --username {wrapped.username} --dbname {wrapped.dbname} --host 127.0.0.1 -c 'select version();'",
+        ]
+    )
+    if result.exit_code:
+        raise ConnectionError("pg is not ready yet")
+    return True