add unit tests for image_distro config

This commit is contained in:
David Asamu
2025-05-29 23:30:27 +01:00
parent c46f7a4c3d
commit 574a9246a6
2 changed files with 135 additions and 5 deletions
+6 -3
View File
@@ -1325,7 +1325,10 @@ def docker_tag(
base_image: Optional[str] = None,
) -> str:
base_image = base_image or default_base_image(config)
wolfi_tag = "-wolfi" if config.get("image_distro") == "wolfi" else ""
image_distro = config.get("image_distro")
distro_tag = "" if image_distro == DEFAULT_IMAGE_DISTRO else f"-{image_distro}"
if config.get("_INTERNAL_docker_tag"):
return f"{base_image}:{config['_INTERNAL_docker_tag']}"
@@ -1333,8 +1336,8 @@ def docker_tag(
return f"{base_image}-py{config['python_version']}"
if config.get("node_version") and not config.get("python_version"):
return f"{base_image}:{config['node_version']}{wolfi_tag}"
return f"{base_image}:{config['python_version']}{wolfi_tag}"
return f"{base_image}:{config['node_version']}{distro_tag}"
return f"{base_image}:{config['python_version']}{distro_tag}"
def config_to_docker(