mirror of
https://github.com/langchain-ai/langgraph.git
synced 2026-09-20 16:47:55 +02:00
67 lines
2.0 KiB
Python
67 lines
2.0 KiB
Python
import pytest
|
|
|
|
from langgraph_cli.image_reference import ImageReference
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("reference", "repository", "tag"),
|
|
[
|
|
pytest.param(
|
|
"registry.example.com/team/app:v1",
|
|
"registry.example.com/team/app",
|
|
"v1",
|
|
id="tag_after_last_slash",
|
|
),
|
|
pytest.param(
|
|
"registry.example.com/team/app",
|
|
"registry.example.com/team/app",
|
|
None,
|
|
id="no_tag",
|
|
),
|
|
pytest.param(
|
|
"localhost:5000/app",
|
|
"localhost:5000/app",
|
|
None,
|
|
id="registry_port_is_not_a_tag",
|
|
),
|
|
pytest.param(
|
|
"localhost:5000/app:latest",
|
|
"localhost:5000/app",
|
|
"latest",
|
|
id="registry_port_with_tag",
|
|
),
|
|
pytest.param("app:dev", "app", "dev", id="bare_name_with_tag"),
|
|
],
|
|
)
|
|
def test_parse_splits_repository_and_tag(reference, repository, tag):
|
|
assert ImageReference.parse(reference) == ImageReference(repository, tag)
|
|
|
|
|
|
def test_with_tag_replaces_the_tag():
|
|
assert ImageReference("r/app", "v1").with_tag("v2") == ImageReference("r/app", "v2")
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("reference", "expected"),
|
|
[
|
|
pytest.param(ImageReference("r/app", "v1"), "r/app:v1", id="tagged"),
|
|
pytest.param(ImageReference("r/app"), "r/app", id="untagged"),
|
|
],
|
|
)
|
|
def test_str_renders_the_docker_reference(reference, expected):
|
|
assert str(reference) == expected
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
("repo_digest", "expected"),
|
|
[
|
|
pytest.param("localhost:5000/app@sha256:abc", True, id="same_repository"),
|
|
pytest.param("localhost:5000/app-2@sha256:abc", False, id="other_repository"),
|
|
pytest.param("mirror.example.com/app@sha256:abc", False, id="other_registry"),
|
|
],
|
|
)
|
|
def test_matches_digest_only_for_the_same_repository(repo_digest, expected):
|
|
assert ImageReference("localhost:5000/app", "v1").matches_digest(repo_digest) is (
|
|
expected
|
|
)
|