refactor
This commit is contained in:
@@ -15,7 +15,7 @@ class TestParseImageRef:
|
||||
def test_simple_name(self):
|
||||
ref = parse_image_ref("devbox")
|
||||
assert ref.registry == "registry.tomastm.com"
|
||||
assert ref.name == "devbox"
|
||||
assert ref.repo == "devbox"
|
||||
assert ref.tag == "latest"
|
||||
|
||||
def test_with_tag(self):
|
||||
@@ -25,7 +25,7 @@ class TestParseImageRef:
|
||||
def test_full_ref(self):
|
||||
ref = parse_image_ref("ghcr.io/user/image:main")
|
||||
assert ref.registry == "ghcr.io"
|
||||
assert ref.name == "user/image"
|
||||
assert ref.repo == "user/image"
|
||||
assert ref.tag == "main"
|
||||
|
||||
def test_full_image_string(self):
|
||||
@@ -35,38 +35,35 @@ class TestParseImageRef:
|
||||
|
||||
class TestContainerName:
|
||||
def test_basic(self):
|
||||
assert container_name("personal", "devbox") == "flow-personal-devbox"
|
||||
assert container_name("devbox") == "dev-devbox"
|
||||
|
||||
|
||||
class TestResolveMounts:
|
||||
def test_projects_mount(self, tmp_path):
|
||||
projects = tmp_path / "projects"
|
||||
projects.mkdir()
|
||||
mounts = resolve_mounts(tmp_path, str(projects))
|
||||
project_mounts = [m for m in mounts if m.target == "/home/user/projects"]
|
||||
mounts = resolve_mounts(tmp_path, project_path=str(projects))
|
||||
project_mounts = [m for m in mounts if m.target == "/workspace"]
|
||||
assert len(project_mounts) == 1
|
||||
|
||||
def test_extra_mounts(self, tmp_path):
|
||||
mounts = resolve_mounts(
|
||||
tmp_path, str(tmp_path),
|
||||
extra_mounts=[{"source": str(tmp_path), "target": "/data"}],
|
||||
)
|
||||
extra = [m for m in mounts if m.target == "/data"]
|
||||
assert len(extra) == 1
|
||||
def test_dotfiles_mount(self, tmp_path):
|
||||
dotfiles = tmp_path / "dotfiles"
|
||||
dotfiles.mkdir()
|
||||
mounts = resolve_mounts(tmp_path, dotfiles_dir=dotfiles)
|
||||
assert any(m.target.endswith("/flow/dotfiles") for m in mounts)
|
||||
|
||||
|
||||
class TestBuildContainerSpec:
|
||||
def test_basic(self):
|
||||
image = ImageRef(registry="reg", name="img", tag="v1")
|
||||
spec = build_container_spec("personal", image, [])
|
||||
assert spec.name == "flow-personal-img"
|
||||
assert spec.env["DF_NAMESPACE"] == "personal"
|
||||
assert spec.env["DF_PLATFORM"] == "container"
|
||||
image = ImageRef(registry="reg", repo="img", tag="v1", label="reg/img")
|
||||
spec = build_container_spec("api", image, [])
|
||||
assert spec.name == "dev-api"
|
||||
assert spec.labels["dev.name"] == "api"
|
||||
|
||||
def test_with_mounts(self):
|
||||
image = ImageRef(registry="reg", name="img", tag="v1")
|
||||
image = ImageRef(registry="reg", repo="img", tag="v1", label="reg/img")
|
||||
mounts = [Mount(source=Path("/a"), target="/b")]
|
||||
spec = build_container_spec("ns", image, mounts)
|
||||
spec = build_container_spec("api", image, mounts)
|
||||
assert len(spec.mounts) == 1
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user