Remove static variables from keycloak.

Author: tillahoffmann

keycloak/testcontainers/keycloak/__init__.py

@@ -17,6 +17,7 @@
 
 from testcontainers.core.container import DockerContainer
 from testcontainers.core.waiting_utils import wait_container_is_ready
+from typing import Optional
 
 
 class KeycloakContainer(DockerContainer):
@@ -32,17 +33,17 @@ class KeycloakContainer(DockerContainer):
             >>> with KeycloakContainer() as kc:
             ...    keycloak = kc.get_client()
     """
-    KEYCLOAK_USER = os.environ.get("KEYCLOAK_USER", "test")
-    KEYCLOAK_PASSWORD = os.environ.get("KEYCLOAK_PASSWORD", "test")
-
-    def __init__(self, image="jboss/keycloak:latest") -> None:
+    def __init__(self, image="jboss/keycloak:latest", username: Optional[str] = None,
+                 password: Optional[str] = None, port: int = 8080) -> None:
         super(KeycloakContainer, self).__init__(image=image)
-        self.port_to_expose = 8080
+        self.username = username or os.environ.get("KEYCLOAK_USER", "test")
+        self.password = password or os.environ.get("KEYCLOAK_PASSWORD", "test")
+        self.port_to_expose = port
         self.with_exposed_ports(self.port_to_expose)
 
     def _configure(self) -> None:
-        self.with_env("KEYCLOAK_USER", self.KEYCLOAK_USER)
-        self.with_env("KEYCLOAK_PASSWORD", self.KEYCLOAK_PASSWORD)
+        self.with_env("KEYCLOAK_USER", self.username)
+        self.with_env("KEYCLOAK_PASSWORD", self.password)
 
     def get_url(self) -> str:
         host = self.get_container_host_ip()
@@ -64,8 +65,8 @@ def start(self) -> "KeycloakContainer":
     def get_client(self, **kwargs) -> KeycloakAdmin:
         default_kwargs = dict(
             server_url="{}/auth/".format(self.get_url()),
-            username=self.KEYCLOAK_USER,
-            password=self.KEYCLOAK_PASSWORD,
+            username=self.username,
+            password=self.password,
             realm_name="master",
             verify=True,
         )