particle-os/debian-forge
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

testutil: make_container context manager

Browse source 
Make make_container a context manager so we can reliably clean up
containers that were created in tests.
This commit is contained in:
Achilleas Koutsou 2024-02-13 18:43:35 +01:00 • committed by Ondřej Budai
parent d6cd4b93ba
commit 591593ea00
3 changed files with 35 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

5
osbuild/testutil/__init__.py
View file

@ -80,6 +80,7 @@ def mock_command(cmd_name: str, script: str):
os.environ["PATH"] = original_path
@contextlib.contextmanager
def make_container(tmp_path, tag, fake_content, base="scratch"):
fake_container_src = tmp_path / "fake-container-src"
make_fake_tree(fake_container_src, fake_content)
@ -95,3 +96,7 @@ def make_container(tmp_path, tag, fake_content, base="scratch"):
"-f", os.fspath(fake_containerfile_path),
"-t", tag,
])
try:
yield
finally:
subprocess.check_call(["podman", "image", "rm", tag])