Skip to content

Commit

Permalink
Add unit test for image name interpolation
Browse files Browse the repository at this point in the history
  • Loading branch information
bimtauer committed Oct 12, 2021
1 parent 2ace67b commit 9aaf53b
Showing 1 changed file with 16 additions and 0 deletions.
16 changes: 16 additions & 0 deletions tests/flytekit/unit/core/test_python_auto_container.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
import pytest

from flytekit.core.context_manager import ImageConfig, Image
from flytekit.core.python_auto_container import get_registerable_container_image


@pytest.fixture
def default_image_config():
default_image = Image(name="default", fqn="docker.io/xyz", tag="some-git-hash")
return ImageConfig(default_image=default_image)


def test_image_name_interpolation(default_image_config):
img_to_interpolate = "{{.image.default.fqn}}:{{.image.default.version}}-special"
img = get_registerable_container_image(img=img_to_interpolate, cfg=default_image_config)
assert img == "docker.io/xyz:some-git-hash-special"

0 comments on commit 9aaf53b

Please sign in to comment.