def run_efi_nolink_test(self, unlink_mock, islink_mock, exec_mock,
listdir_mock, isdir_mock, storage_mock):
"""Test OSTree bootloader copy task run() with EFI, efi dir, and no links"""
bootloader_mock = storage_mock.get_proxy()
bootloader_mock.IsEFI.return_value = True
isdir_mock.side_effect = [True, False, True, True,
True] # boot source, 3x listdir, efi
listdir_mock.return_value = ["some_file", "directory", "efi"]
islink_mock.return_value = False
task = CopyBootloaderDataTask("/sysroot", "/physroot")
task.run()
exec_mock.assert_has_calls([
call("cp", [
"-r", "-p", "/sysroot/usr/lib/ostree-boot/directory",
"/physroot/boot"
]),
call("cp", [
"-r", "-p", "/sysroot/usr/lib/ostree-boot/efi",
"/physroot/boot"
])
])
unlink_mock.assert_not_called()
def test_run_failed(self, listdir_mock, isdir_mock, storage_mock):
"""Test OSTree bootloader copy task run() with an exception."""
bootloader_mock = storage_mock.get_proxy()
bootloader_mock.IsEFI.return_value = False
isdir_mock.return_value = False
listdir_mock.side_effect = OSError("Fake!")
task = CopyBootloaderDataTask("/sysroot", "/physroot")
with pytest.raises(PayloadInstallationError) as cm:
task.run()
assert str(cm.value) == "Failed to copy bootloader data: Fake!"
def run_noefi_notadir_test(self, unlink_mock, islink_mock, exec_mock,
listdir_mock, isdir_mock, storage_mock):
"""Test OSTree bootloader copy task run() with non-directory source of data"""
bootloader_mock = storage_mock.get_proxy()
bootloader_mock.IsEFI.return_value = False
isdir_mock.side_effect = [False, False,
True] # boot source, 2x listdir
listdir_mock.return_value = ["some_file", "directory"]
islink_mock.return_value = False
task = CopyBootloaderDataTask("/sysroot", "/physroot")
task.run()
exec_mock.assert_called_once_with(
"cp", ["-r", "-p", "/sysroot/boot/directory", "/physroot/boot"])
unlink_mock.assert_not_called()
def run_noefi_efidir_link_test(self, unlink_mock, islink_mock, exec_mock,
listdir_mock, isdir_mock, storage_mock):
"""Test OSTree bootloader copy task run() with no EFI but efi dir and link"""
bootloader_mock = storage_mock.get_proxy()
bootloader_mock.IsEFI.return_value = False
isdir_mock.side_effect = [True, False, True, True,
True] # boot source, 3x listdir, efi
listdir_mock.return_value = ["some_file", "directory", "efi"]
islink_mock.return_value = True
task = CopyBootloaderDataTask("/sysroot", "/physroot")
task.run()
exec_mock.assert_called_once_with("cp", [
"-r", "-p", "/sysroot/usr/lib/ostree-boot/directory",
"/physroot/boot"
])
unlink_mock.assert_called_with("/physroot/boot/grub2/grubenv")
def test_run_noefi_noefidir_nolink(
self, unlink_mock, islink_mock, exec_mock, listdir_mock, isdir_mock, storage_mock):
"""Test OSTree bootloader copy task run() with no EFI, no efi dir, and no links"""
exec_mock.return_value = 0
bootloader_mock = storage_mock.get_proxy()
bootloader_mock.IsEFI.return_value = False
isdir_mock.side_effect = [True, False, True] # boot source, 2x listdir
listdir_mock.return_value = ["some_file", "directory"]
islink_mock.return_value = False
task = CopyBootloaderDataTask("/sysroot", "/physroot")
task.run()
exec_mock.assert_called_once_with(
"cp", ["-r", "-p", "/sysroot/usr/lib/ostree-boot/directory", "/physroot/boot"]
)
unlink_mock.assert_not_called()