linting

Author: alexanderankin

core/testcontainers/compose/compose.py

@@ -247,7 +247,9 @@ def docker_compose_command(self) -> list[str]:
 
     @cached_property
     def compose_command_property(self) -> list[str]:
-        docker_compose_cmd = [self.docker_command_path, "compose"] if self.docker_command_path else ["docker", "compose"]
+        docker_compose_cmd = (
+            [self.docker_command_path, "compose"] if self.docker_command_path else ["docker", "compose"]
+        )
         if self.compose_file_name:
             for file in self.compose_file_name:
                 docker_compose_cmd += ["-f", file]

core/testcontainers/core/container.py

@@ -20,7 +20,7 @@
 from testcontainers.core.network import Network
 from testcontainers.core.utils import is_arm, setup_logger
 from testcontainers.core.wait_strategies import LogMessageWaitStrategy
-from testcontainers.core.waiting_utils import WaitStrategy, wait_container_is_ready
+from testcontainers.core.waiting_utils import WaitStrategy
 
 if TYPE_CHECKING:
     from docker.models.containers import Container

core/testcontainers/core/wait_strategies.py

@@ -31,14 +31,15 @@
 import time
 from datetime import timedelta
 from pathlib import Path
-from typing import Any, Callable, Optional, TYPE_CHECKING, Union
+from typing import TYPE_CHECKING, Any, Callable, Optional, Union, cast
 from urllib.error import HTTPError, URLError
 from urllib.request import Request, urlopen
 
 from typing_extensions import Self
 
 from testcontainers.compose import DockerCompose
 from testcontainers.core.utils import setup_logger
+
 # Import base classes from waiting_utils to make them available for tests
 from testcontainers.core.waiting_utils import WaitStrategy, WaitStrategyTarget
 
@@ -655,9 +656,10 @@ class ContainerStatusWaitStrategy(WaitStrategy):
         dead
     https://docs.docker.com/reference/cli/docker/container/ls/#status
     """
-    CONTINUE_STATUSES = {"created", "restarting"}
 
-    def __init__(self):
+    CONTINUE_STATUSES = frozenset(("created", "restarting"))
+
+    def __init__(self) -> None:
         super().__init__()
 
     def wait_until_ready(self, container: WaitStrategyTarget) -> None:
@@ -671,7 +673,11 @@ def running(status: str) -> bool:
             logger.debug("status is now running")
             return True
         if status in ContainerStatusWaitStrategy.CONTINUE_STATUSES:
-            logger.debug("status is %s, which is valid for continuing (%s)", status, ContainerStatusWaitStrategy.CONTINUE_STATUSES)
+            logger.debug(
+                "status is %s, which is valid for continuing (%s)",
+                status,
+                ContainerStatusWaitStrategy.CONTINUE_STATUSES,
+            )
             return False
         raise StopIteration(f"container status not valid for continuing: {status}")
 
@@ -689,7 +695,7 @@ def _get_status_tc_container(container: "DockerContainer") -> str:
         logger.debug("fetching status of container %s", container)
         wrapped = container.get_wrapped_container()
         wrapped.reload()
-        return wrapped.status
+        return cast("str", wrapped.status)
 
     @staticmethod
     def _get_status_compose_container(container: DockerCompose) -> str:

core/testcontainers/core/waiting_utils.py

@@ -103,7 +103,7 @@ def wait_until_ready(self, container: WaitStrategyTarget) -> None:
         """Wait until the container is ready."""
         pass
 
-    def _poll(self, check: Callable[[], bool], transient_exceptions: list[type[Exception]] = None) -> bool:
+    def _poll(self, check: Callable[[], bool], transient_exceptions: Optional[list[type[Exception]]] = None) -> bool:
         if not transient_exceptions:
             all_te_types = self._transient_exceptions
         else: