Add type annotations for arangodb.

Author: tillahoffmann

arangodb/testcontainers/arangodb/__init__.py

@@ -40,7 +40,7 @@ def __init__(self,
                  arango_root_password: str = "passwd",
                  arango_no_auth: typing.Optional[bool] = None,
                  arango_random_root_password: typing.Optional[bool] = None,
-                 **kwargs):
+                 **kwargs) -> None:
         """
         Args:
             image: Actual docker image/tag to pull.
@@ -69,23 +69,16 @@ def __init__(self,
             else arango_random_root_password
         ))
 
-    def _configure(self):
+    def _configure(self) -> None:
         self.with_env("ARANGO_ROOT_PASSWORD", self.arango_root_password)
         if self.arango_no_auth:
             self.with_env("ARANGO_NO_AUTH", "1")
         if self.arango_random_root_password:
             self.with_env("ARANGO_RANDOM_ROOT_PASSWORD", "1")
 
-    def get_connection_url(self):
-        # for now, single host over HTTP
-        scheme = "http"
+    def get_connection_url(self) -> str:
         port = self.get_exposed_port(self.port_to_expose)
-        url = f"{scheme}://{self.get_container_host_ip()}:{port}"
+        return f"http://{self.get_container_host_ip()}:{port}"
 
-        return url
-
-    def _connect(self):
-        wait_for_logs(
-            self,
-            predicate="is ready for business",
-            timeout=MAX_TRIES)
+    def _connect(self) -> None:
+        wait_for_logs(self, predicate="is ready for business", timeout=MAX_TRIES)