bitplane
diff --git a/‎localstack/testcontainers/localstack/__init__.py‎
Lines changed: 14 additions & 10 deletions b/‎localstack/testcontainers/localstack/__init__.py‎
Lines changed: 14 additions & 10 deletions
@@ -36,31 +36,35 @@ class LocalStackContainer(DockerContainer):
         scan_result = dynamo_client.scan(TableName='foo')
         # Do something with the scan result
     """
-    EDGE_PORT = 4566
-    IMAGE = 'localstack/localstack:0.11.4'
-
-    def __init__(self, image=IMAGE, **kwargs):
+    def __init__(self, image: str = 'localstack/localstack:0.11.4', edge_port: int = 4566,
+                 **kwargs) -> None:
         super(LocalStackContainer, self).__init__(image, **kwargs)
-        self.with_exposed_ports(LocalStackContainer.EDGE_PORT)
+        self.edge_port = edge_port
+        self.with_exposed_ports(self.edge_port)
 
-    def with_services(self, *services):
+    def with_services(self, *services) -> "LocalStackContainer":
         """
         Restrict what services to run. By default all localstack services are launched.
-        :return: the DockerContainer to allow chaining of 'with_*' calls.
+
+        Args:
+            services: Sequency of services to launch.
+
+        Returns:
+            self: Container to allow chaining of 'with_*' calls.
         """
         return self.with_env('SERVICES', ','.join(services))
 
-    def get_url(self):
+    def get_url(self) -> str:
         """
         Use this to call localstack instead of real AWS services.
         ex: boto3.client('lambda', endpoint_url=localstack.get_url())
         :return: the endpoint where localstack is reachable.
         """
         host = self.get_container_host_ip()
-        port = self.get_exposed_port(LocalStackContainer.EDGE_PORT)
+        port = self.get_exposed_port(self.edge_port)
         return 'http://{}:{}'.format(host, port)
 
-    def start(self, timeout=60):
+    def start(self, timeout: float = 60) -> "LocalStackContainer":
         super().start()
         wait_for_logs(self, r'Ready\.\n', timeout=timeout)
         return self