debian-forge/test/mod/test_util_mnt.py
2024-01-02 19:31:31 +01:00

47 lines
1.5 KiB
Python

import os
import pytest
from osbuild.mounts import FileSystemMountService
from osbuild.util.mnt import MountGuard, mount
@pytest.mark.skipif(os.getuid() != 0, reason="root only")
def test_mount_failure_msg(tmp_path):
with pytest.raises(RuntimeError) as e:
mount("/dev/invalid-src", tmp_path)
assert "special device /dev/invalid-src does not exist" in str(e.value)
@pytest.mark.skipif(os.getuid() != 0, reason="root only")
def test_mount_guard_failure_msg(tmp_path):
with pytest.raises(RuntimeError) as e:
with MountGuard() as mg:
mg.mount("/dev/invalid-src", tmp_path)
assert "special device /dev/invalid-src does not exist" in str(e.value)
# This needs a proper refactor so that FileSystemMountService just uses
# a common mount helper.
class FakeFileSystemMountService(FileSystemMountService):
def __init__(self, args=None): # pylint: disable=super-init-not-called
# override __init__ to make it testable
pass
def translate_options(self, options):
return options
@pytest.mark.skipif(os.getuid() != 0, reason="root only")
def test_osbuild_mount_failure_msg(tmp_path):
mnt_service = FakeFileSystemMountService()
# yes, we have a third way of mounting things
with pytest.raises(RuntimeError) as e:
args = {
"source": "/dev/invalid-src",
"target": os.fspath(tmp_path),
"root": "/",
"options": [],
}
mnt_service.mount(args)
assert "special device /dev/invalid-src does not exist" in str(e.value)