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()