LAMBDA: use arbitrary docker image for lambda tests via MOTO_DOCKER_L…

…AM… (#8256)
getmoto · Oct 24, 2024 · fd4bd3b · fd4bd3b
1 parent 6e705a8
commit fd4bd3b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/moto/awslambda/models.py b/moto/awslambda/models.py
@@ -1027,7 +1027,11 @@ def _invoke_lambda(self, event: Optional[str] = None) -> Tuple[str, bool, str]:
                         ]
                     )
                     for image_repo in image_repos:
-                        image_ref = f"{image_repo}:{self.run_time}"
+                        image_ref = (
+                            image_repo
+                            if ":" in image_repo
+                            else f"{image_repo}:{self.run_time}"
+                        )
                         try:
                             self.ensure_image_exists(image_ref)
                             break