post refactor fix

Author: Tranquility2

src/testcontainers/community/keycloak/__init__.py

@@ -15,6 +15,7 @@
 
 import requests
 from keycloak import KeycloakAdmin
+from typing_extensions import Self
 
 from testcontainers.core.container import DockerContainer
 from testcontainers.core.waiting_utils import wait_container_is_ready, wait_for_logs
@@ -109,7 +110,7 @@ def _readiness_probe(self) -> None:
             wait_for_logs(self, "started in \\d+\\.\\d+s")
             wait_for_logs(self, "Created temporary admin user|Added user '")
 
-    def start(self) -> "KeycloakContainer":
+    def start(self) -> Self:
         super().start()
         self._readiness_probe()
         self._disable_ssl_required()
@@ -137,15 +138,15 @@ def _disable_ssl_required(self) -> None:
         if result.exit_code != 0:
             raise RuntimeError(f"Failed to disable SSL for master realm: {result.output.decode()}")
 
-    def with_realm_import_file(self, realm_import_file: str) -> "KeycloakContainer":
+    def with_realm_import_file(self, realm_import_file: str) -> Self:
         file = os.path.abspath(realm_import_file)
         if not os.path.exists(file):
             raise FileNotFoundError(f"Realm file {file} does not exist")
         self.with_volume_mapping(file, "/opt/keycloak/data/import/realm.json")
         self.has_realm_imports = True
         return self
 
-    def with_realm_import_folder(self, realm_import_folder: str) -> "KeycloakContainer":
+    def with_realm_import_folder(self, realm_import_folder: str) -> Self:
         folder = os.path.abspath(realm_import_folder)
         if not os.path.exists(folder):
             raise FileNotFoundError(f"Realm folder {folder} does not exist")

src/testcontainers/community/selenium/__init__.py

@@ -85,7 +85,7 @@ def get_connection_url(self) -> str:
         port = self.get_exposed_port(self.port)
         return f"http://{ip}:{port}/wd/hub"
 
-    def with_options(self, options: Optional[ArgOptions]):
+    def with_options(self, options: Optional[ArgOptions]) -> Self:
         self.options = options
         return self
 

src/testcontainers/socat/socat.py

@@ -47,7 +47,7 @@ def __init__(
 
         super().__init__(image=image, **kwargs)
 
-    def with_target(self, exposed_port: int, host: str, internal_port: Optional[int] = None) -> "SocatContainer":
+    def with_target(self, exposed_port: int, host: str, internal_port: Optional[int] = None) -> Self:
         """
         Add a target to forward connections from the exposed port to the given host and port.