have wiremock api key passed to the extension

Author: HarshCasper

wiremock/localstack_wiremock/extension.py

@@ -25,21 +25,20 @@ class WireMockExtension(ProxiedDockerContainerExtension):
     def __init__(self):
         env_vars = {}
         image_name = self.DOCKER_IMAGE
-        kwargs = {}
+        volumes = None
         if api_token := os.getenv(ENV_WIREMOCK_API_TOKEN):
             env_vars["WMC_ADMIN_PORT"] = str(PORT)
-            # TODO remove?
-            # env_vars["WMC_DEFAULT_MODE"] = "record-many"
             env_vars["WMC_API_TOKEN"] = api_token
             env_vars["WMC_RUNNER_ENABLED"] = "true"
             image_name = self.DOCKER_IMAGE_RUNNER
             settings_file = Util.mountable_tmp_file()
             # TODO: set configs in YAML file
-            kwargs["volumes"] = ([(settings_file, "/work/.wiremock/wiremock.yaml")],)
+            volumes = [(settings_file, "/work/.wiremock/wiremock.yaml")]
         super().__init__(
             image_name=image_name,
             container_ports=[PORT],
             container_name=self.CONTAINER_NAME,
             host=self.HOST,
-            **kwargs,
+            volumes=volumes,
+            env_vars=env_vars if env_vars else None,
         )

wiremock/localstack_wiremock/utils/docker.py

@@ -47,9 +47,12 @@ class ProxiedDockerContainerExtension(Extension):
     http2_ports: list[int] | None
     """List of ports for which HTTP2 proxy forwarding into the container should be enabled."""
 
-    volumes: list[SimpleVolumeBind] | None = (None,)
+    volumes: list[SimpleVolumeBind] | None = None
     """Optional volumes to mount into the container host."""
 
+    env_vars: dict[str, str] | None = None
+    """Optional environment variables to pass to the container."""
+
     def __init__(
         self,
         image_name: str,
@@ -61,6 +64,7 @@ def __init__(
         request_to_port_router: Callable[[Request], int] | None = None,
         http2_ports: list[int] | None = None,
         volumes: list[SimpleVolumeBind] | None = None,
+        env_vars: dict[str, str] | None = None,
     ):
         self.image_name = image_name
         self.container_ports = container_ports
@@ -71,6 +75,7 @@ def __init__(
         self.request_to_port_router = request_to_port_router
         self.http2_ports = http2_ports
         self.volumes = volumes
+        self.env_vars = env_vars
 
     def update_gateway_routes(self, router: http.Router[http.RouteHandler]):
         if self.path:
@@ -106,6 +111,8 @@ def start_container(self) -> None:
         kwargs = {}
         if self.command:
             kwargs["command"] = self.command
+        if self.env_vars:
+            kwargs["env_vars"] = self.env_vars
 
         try:
             DOCKER_CLIENT.run_container(

wiremock/sample-app/main.tf

@@ -34,7 +34,7 @@ resource "null_resource" "package_lambda" {
       rm -rf "${path.module}/build"
       mkdir -p "${path.module}/build"
       cp "${path.module}/src/handler.py" "${path.module}/build/"
-      pip install -r "${path.module}/src/requirements.txt" -t "${path.module}/build"
+      pip3 install -r "${path.module}/src/requirements.txt" -t "${path.module}/build"
     EOT
     interpreter = ["/bin/bash", "-c"]
   }