feat(main): enable typing for complete package

- fix all typing errors and asserts in tests
- improved type hints for main methods like docker.run.

Author: CarliJoy

pyproject.toml

@@ -139,6 +139,8 @@ lint = [
   "types-paramiko>=4",
   "ruff",
   "pre-commit>=4",
+  "types-docker>=7.1.0.20260518",
+  "boto3-stubs-lite[boto3]",
 ]
 docs = [
   "sphinx>=8; python_version < '3.11'",
@@ -277,15 +279,15 @@ check_untyped_defs = true
 disable_error_code = ["no-untyped-def"]
 
 [[tool.mypy.overrides]]
-module = ["docker.*"]
+module = ["pika.*"]
 ignore_missing_imports = true
 
 [[tool.mypy.overrides]]
-module = ["wrapt.*"]
+module = ["influxdb.*"]
 ignore_missing_imports = true
 
 [[tool.mypy.overrides]]
-module = ["requests.*"]
+module = ["cassandra.*"]
 ignore_missing_imports = true
 
 [[tool.mypy.overrides]]

…estcontainers/community/azurite/py.typed src/testcontainers/community/__init__.pysrc/testcontainers/community/azurite/py.typed renamed to src/testcontainers/community/__init__.py src/testcontainers/community/azurite/py.typed renamed to src/testcontainers/community/__init__.py

@@ -2,6 +2,7 @@
 from typing import Union
 
 import httpx
+
 from testcontainers.community.generic.server import ServerContainer
 from testcontainers.core.image import DockerImage
 

src/testcontainers/community/aws/aws_lambda.py

@@ -131,17 +131,17 @@ def __get_local_connection_string(self) -> str:
             f"DefaultEndpointsProtocol=http;AccountName={self.account_name};AccountKey={self.account_key};"
         )
 
-        if self.blob_service_port in self.ports:
+        if str(self.blob_service_port) in self.ports:
             connection_string += (
                 f"BlobEndpoint=http://{host_ip}:{self.get_exposed_port(self.blob_service_port)}/{self.account_name};"
             )
 
-        if self.queue_service_port in self.ports:
+        if str(self.queue_service_port) in self.ports:
             connection_string += (
                 f"QueueEndpoint=http://{host_ip}:{self.get_exposed_port(self.queue_service_port)}/{self.account_name};"
             )
 
-        if self.table_service_port in self.ports:
+        if str(self.table_service_port) in self.ports:
             connection_string += (
                 f"TableEndpoint=http://{host_ip}:{self.get_exposed_port(self.table_service_port)}/{self.account_name};"
             )
@@ -206,13 +206,13 @@ def __get_external_connection_string(self) -> str:
             f"DefaultEndpointsProtocol=http;AccountName={self.account_name};AccountKey={self.account_key};"
         )
 
-        if self.blob_service_port in self.ports:
+        if str(self.blob_service_port) in self.ports:
             connection_string += f"BlobEndpoint=http://{host_ip}:{blob_port}/{self.account_name};"
 
-        if self.queue_service_port in self.ports:
+        if str(self.queue_service_port) in self.ports:
             connection_string += f"QueueEndpoint=http://{host_ip}:{queue_port}/{self.account_name};"
 
-        if self.table_service_port in self.ports:
+        if str(self.table_service_port) in self.ports:
             connection_string += f"TableEndpoint=http://{host_ip}:{table_port}/{self.account_name};"
 
         return connection_string

src/testcontainers/community/azurite/__init__.py

@@ -1,6 +1,7 @@
 from typing import TYPE_CHECKING
 
 from requests import ConnectionError, get
+
 from testcontainers.core.container import DockerContainer
 from testcontainers.core.utils import raise_for_deprecated_parameter
 from testcontainers.core.waiting_utils import wait_container_is_ready

src/testcontainers/community/chroma/__init__.py

@@ -40,7 +40,7 @@ def __init__(
         bind_ports: bool = os.getenv("AZURE_COSMOS_EMULATOR_BIND_PORTS", "true").strip().lower() in _ENV_TRUTHY,
         endpoint_ports: Iterable[int] = [],
         **other_kwargs,
-    ):
+    ) -> None:
         super().__init__(image=image, **other_kwargs)
         self.endpoint_ports = endpoint_ports
         self.partition_count = partition_count

src/testcontainers/community/cosmosdb/_emulator.py

@@ -30,7 +30,7 @@ def __init__(
             "AZURE_COSMOS_EMULATOR_IMAGE", "mcr.microsoft.com/cosmosdb/linux/azure-cosmos-emulator:latest"
         ),
         **other_kwargs,
-    ):
+    ) -> None:
         super().__init__(image=image, endpoint_ports=[ENDPOINT_PORT], **other_kwargs)
         assert mongodb_version is not None, "A MongoDB version is required to use the MongoDB Endpoint"
         self.mongodb_version = mongodb_version

src/testcontainers/community/cosmosdb/mongodb.py

@@ -1,6 +1,7 @@
 from azure.core.exceptions import ServiceRequestError
 from azure.cosmos import CosmosClient as SyncCosmosClient
 from azure.cosmos.aio import CosmosClient as AsyncCosmosClient
+
 from testcontainers.core.waiting_utils import wait_container_is_ready
 
 from ._emulator import CosmosDBEmulatorContainer
@@ -33,7 +34,7 @@ class CosmosDBNoSQLEndpointContainer(CosmosDBEmulatorContainer):
 
     """
 
-    def __init__(self, **kwargs):
+    def __init__(self, **kwargs: object) -> None:
         super().__init__(endpoint_ports=[NOSQL_PORT], **kwargs)
 
     @property

src/testcontainers/community/cosmosdb/nosql.py

@@ -72,7 +72,7 @@ class ElasticSearchContainer(DockerContainer):
             '8.3.3'
     """
 
-    def __init__(self, image: str = "elasticsearch", port: int = 9200, **kwargs) -> None:
+    def __init__(self, image: str = "elasticsearch", port: int = 9200, **kwargs: object) -> None:
         raise_for_deprecated_parameter(kwargs, "port_to_expose", "port")
         super().__init__(image, _wait_strategy=HttpWaitStrategy(port), **kwargs)
         self.port = port

src/testcontainers/community/elasticsearch/__init__.py

@@ -19,7 +19,7 @@
 class SqlAlchemyConnectWaitStrategy(WaitStrategy):
     """Wait strategy for database connectivity testing using SQLAlchemy."""
 
-    def __init__(self):
+    def __init__(self) -> None:
         super().__init__()
         self.with_transient_exceptions(TimeoutError, ConnectionError, *ADDITIONAL_TRANSIENT_ERRORS)