2015-02-19 13:42:39 +00:00
|
|
|
# Copyright 2015 Red Hat, Inc.
|
|
|
|
|
# All Rights Reserved.
|
|
|
|
|
#
|
|
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
|
|
|
# not use this file except in compliance with the License. You may obtain
|
|
|
|
|
# a copy of the License at
|
|
|
|
|
#
|
|
|
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
|
#
|
|
|
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
|
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
|
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
|
|
|
# License for the specific language governing permissions and limitations
|
|
|
|
|
# under the License.
|
|
|
|
|
|
2015-03-12 09:15:49 +00:00
|
|
|
import os
|
2015-02-19 13:42:39 +00:00
|
|
|
import shutil
|
|
|
|
|
import tempfile
|
2020-04-06 12:28:30 +02:00
|
|
|
from unittest import mock
|
2015-02-19 13:42:39 +00:00
|
|
|
|
2019-08-06 13:52:13 +02:00
|
|
|
from ironic_lib import utils as ilib_utils
|
2015-02-19 13:42:39 +00:00
|
|
|
from oslo_concurrency import processutils
|
|
|
|
|
|
|
|
|
|
from ironic_python_agent import errors
|
|
|
|
|
from ironic_python_agent.extensions import image
|
2015-12-15 13:47:40 +01:00
|
|
|
from ironic_python_agent.extensions import iscsi
|
2015-02-19 13:42:39 +00:00
|
|
|
from ironic_python_agent import hardware
|
2017-10-18 11:06:30 -07:00
|
|
|
from ironic_python_agent.tests.unit import base
|
2015-02-19 13:42:39 +00:00
|
|
|
from ironic_python_agent import utils
|
|
|
|
|
|
|
|
|
|
|
2017-03-19 08:30:25 -07:00
|
|
|
@mock.patch.object(hardware, 'dispatch_to_managers', autospec=True)
|
|
|
|
|
@mock.patch.object(utils, 'execute', autospec=True)
|
2015-02-19 13:42:39 +00:00
|
|
|
@mock.patch.object(tempfile, 'mkdtemp', lambda *_: '/tmp/fake-dir')
|
|
|
|
|
@mock.patch.object(shutil, 'rmtree', lambda *_: None)
|
2017-10-18 11:06:30 -07:00
|
|
|
class TestImageExtension(base.IronicAgentTest):
|
2015-02-19 13:42:39 +00:00
|
|
|
|
|
|
|
|
def setUp(self):
|
|
|
|
|
super(TestImageExtension, self).setUp()
|
|
|
|
|
self.agent_extension = image.ImageExtension()
|
|
|
|
|
self.fake_dev = '/dev/fake'
|
2015-03-12 09:15:49 +00:00
|
|
|
self.fake_efi_system_part = '/dev/fake1'
|
2015-02-19 13:42:39 +00:00
|
|
|
self.fake_root_part = '/dev/fake2'
|
2018-07-11 13:35:54 +00:00
|
|
|
self.fake_prep_boot_part = '/dev/fake3'
|
2015-02-19 13:42:39 +00:00
|
|
|
self.fake_root_uuid = '11111111-2222-3333-4444-555555555555'
|
2015-03-12 09:15:49 +00:00
|
|
|
self.fake_efi_system_part_uuid = '45AB-2312'
|
2018-07-11 13:35:54 +00:00
|
|
|
self.fake_prep_boot_part_uuid = '76937797-3253-8843-999999999999'
|
2015-02-19 13:42:39 +00:00
|
|
|
self.fake_dir = '/tmp/fake-dir'
|
|
|
|
|
|
2017-03-19 08:30:25 -07:00
|
|
|
@mock.patch.object(iscsi, 'clean_up', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_install_grub2', autospec=True)
|
2019-12-02 16:56:42 +01:00
|
|
|
def test__install_bootloader_bios(self, mock_grub2, mock_iscsi_clean,
|
|
|
|
|
mock_execute, mock_dispatch):
|
|
|
|
|
mock_dispatch.side_effect = [
|
|
|
|
|
self.fake_dev, hardware.BootInfo(current_boot_mode='bios')
|
|
|
|
|
]
|
2015-02-19 13:42:39 +00:00
|
|
|
self.agent_extension.install_bootloader(root_uuid=self.fake_root_uuid)
|
2019-12-02 16:56:42 +01:00
|
|
|
mock_dispatch.assert_any_call('get_os_install_device')
|
|
|
|
|
mock_dispatch.assert_any_call('get_boot_info')
|
|
|
|
|
self.assertEqual(2, mock_dispatch.call_count)
|
2015-03-12 09:15:49 +00:00
|
|
|
mock_grub2.assert_called_once_with(
|
|
|
|
|
self.fake_dev, root_uuid=self.fake_root_uuid,
|
2019-08-06 13:52:13 +02:00
|
|
|
efi_system_part_uuid=None, prep_boot_part_uuid=None,
|
|
|
|
|
target_boot_mode='bios'
|
|
|
|
|
)
|
2015-12-15 13:47:40 +01:00
|
|
|
mock_iscsi_clean.assert_called_once_with(self.fake_dev)
|
2015-02-19 13:42:39 +00:00
|
|
|
|
2017-03-19 08:30:25 -07:00
|
|
|
@mock.patch.object(iscsi, 'clean_up', autospec=True)
|
2019-12-02 16:56:42 +01:00
|
|
|
@mock.patch.object(image, '_manage_uefi', autospec=True)
|
2017-03-19 08:30:25 -07:00
|
|
|
@mock.patch.object(image, '_install_grub2', autospec=True)
|
2019-12-02 16:56:42 +01:00
|
|
|
def test__install_bootloader_uefi(self, mock_grub2, mock_uefi,
|
|
|
|
|
mock_iscsi_clean,
|
|
|
|
|
mock_execute, mock_dispatch):
|
|
|
|
|
mock_dispatch.side_effect = [
|
|
|
|
|
self.fake_dev, hardware.BootInfo(current_boot_mode='uefi')
|
|
|
|
|
]
|
|
|
|
|
mock_uefi.return_value = False
|
2015-03-12 09:15:49 +00:00
|
|
|
self.agent_extension.install_bootloader(
|
|
|
|
|
root_uuid=self.fake_root_uuid,
|
2019-08-06 13:52:13 +02:00
|
|
|
efi_system_part_uuid=self.fake_efi_system_part_uuid,
|
|
|
|
|
target_boot_mode='uefi'
|
|
|
|
|
)
|
2019-12-02 16:56:42 +01:00
|
|
|
mock_dispatch.assert_any_call('get_os_install_device')
|
|
|
|
|
mock_dispatch.assert_any_call('get_boot_info')
|
|
|
|
|
self.assertEqual(2, mock_dispatch.call_count)
|
2015-03-12 09:15:49 +00:00
|
|
|
mock_grub2.assert_called_once_with(
|
|
|
|
|
self.fake_dev,
|
|
|
|
|
root_uuid=self.fake_root_uuid,
|
2018-07-11 13:35:54 +00:00
|
|
|
efi_system_part_uuid=self.fake_efi_system_part_uuid,
|
2019-08-06 13:52:13 +02:00
|
|
|
prep_boot_part_uuid=None,
|
|
|
|
|
target_boot_mode='uefi'
|
|
|
|
|
)
|
2018-07-11 13:35:54 +00:00
|
|
|
mock_iscsi_clean.assert_called_once_with(self.fake_dev)
|
|
|
|
|
|
2019-08-06 13:52:13 +02:00
|
|
|
@mock.patch.object(hardware, 'is_md_device', lambda *_: False)
|
2019-12-02 16:56:42 +01:00
|
|
|
@mock.patch.object(os.path, 'exists', lambda *_: False)
|
|
|
|
|
@mock.patch.object(iscsi, 'clean_up', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_efi_bootloaders', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_partition', autospec=True)
|
|
|
|
|
@mock.patch.object(utils, 'get_efi_part_on_device', autospec=False)
|
|
|
|
|
@mock.patch.object(os, 'makedirs', autospec=True)
|
|
|
|
|
def test__uefi_bootloader_given_partition(
|
|
|
|
|
self, mkdir_mock, mock_utils_efi_part, mock_partition,
|
|
|
|
|
mock_efi_bl, mock_iscsi_clean, mock_execute, mock_dispatch):
|
|
|
|
|
mock_dispatch.side_effect = [
|
|
|
|
|
self.fake_dev, hardware.BootInfo(current_boot_mode='uefi')
|
|
|
|
|
]
|
|
|
|
|
mock_partition.side_effect = [self.fake_dev, self.fake_efi_system_part]
|
|
|
|
|
mock_efi_bl.return_value = ['\\EFI\\BOOT\\BOOTX64.EFI']
|
|
|
|
|
mock_utils_efi_part.return_value = '1'
|
|
|
|
|
|
|
|
|
|
mock_execute.side_effect = iter([('', ''), ('', ''),
|
2020-02-08 07:18:12 -08:00
|
|
|
('', ''), ('', ''),
|
2019-12-02 16:56:42 +01:00
|
|
|
('', ''), ('', ''),
|
|
|
|
|
('', ''), ('', '')])
|
|
|
|
|
|
|
|
|
|
expected = [mock.call('efibootmgr', '--version'),
|
2020-02-08 07:18:12 -08:00
|
|
|
mock.call('partx', '-u', '/dev/fake', attempts=3,
|
|
|
|
|
delay_on_retry=True),
|
|
|
|
|
mock.call('udevadm', 'settle'),
|
2019-12-02 16:56:42 +01:00
|
|
|
mock.call('mount', self.fake_efi_system_part,
|
|
|
|
|
self.fake_dir + '/boot/efi'),
|
|
|
|
|
mock.call('efibootmgr'),
|
|
|
|
|
mock.call('efibootmgr', '-c', '-d', self.fake_dev,
|
|
|
|
|
'-p', '1', '-w',
|
|
|
|
|
'-L', 'ironic1', '-l',
|
|
|
|
|
'\\EFI\\BOOT\\BOOTX64.EFI'),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/boot/efi',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('sync')]
|
|
|
|
|
|
|
|
|
|
self.agent_extension.install_bootloader(
|
|
|
|
|
root_uuid=self.fake_root_uuid,
|
|
|
|
|
efi_system_part_uuid=self.fake_efi_system_part_uuid)
|
|
|
|
|
|
|
|
|
|
mock_dispatch.assert_any_call('get_os_install_device')
|
|
|
|
|
mock_dispatch.assert_any_call('get_boot_info')
|
|
|
|
|
mkdir_mock.assert_called_once_with(self.fake_dir + '/boot/efi')
|
|
|
|
|
mock_efi_bl.assert_called_once_with(self.fake_dir + '/boot/efi')
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
|
|
|
|
mock_utils_efi_part.assert_called_once_with(self.fake_dev)
|
2020-02-08 07:18:12 -08:00
|
|
|
self.assertEqual(8, mock_execute.call_count)
|
2019-12-02 16:56:42 +01:00
|
|
|
|
2019-08-06 13:52:13 +02:00
|
|
|
@mock.patch.object(hardware, 'is_md_device', lambda *_: False)
|
2019-12-02 16:56:42 +01:00
|
|
|
@mock.patch.object(os.path, 'exists', lambda *_: False)
|
|
|
|
|
@mock.patch.object(iscsi, 'clean_up', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_efi_bootloaders', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_partition', autospec=True)
|
|
|
|
|
@mock.patch.object(utils, 'get_efi_part_on_device', autospec=True)
|
|
|
|
|
@mock.patch.object(os, 'makedirs', autospec=True)
|
|
|
|
|
def test__uefi_bootloader_find_partition(
|
|
|
|
|
self, mkdir_mock, mock_utils_efi_part, mock_partition,
|
|
|
|
|
mock_efi_bl, mock_iscsi_clean, mock_execute, mock_dispatch):
|
|
|
|
|
mock_dispatch.side_effect = [
|
|
|
|
|
self.fake_dev, hardware.BootInfo(current_boot_mode='uefi')
|
|
|
|
|
]
|
|
|
|
|
mock_partition.return_value = self.fake_dev
|
|
|
|
|
mock_utils_efi_part.return_value = '1'
|
|
|
|
|
mock_efi_bl.return_value = ['\\EFI\\BOOT\\BOOTX64.EFI']
|
|
|
|
|
mock_execute.side_effect = iter([('', ''), ('', ''),
|
2020-02-08 07:18:12 -08:00
|
|
|
('', ''), ('', ''),
|
2019-12-02 16:56:42 +01:00
|
|
|
('', ''), ('', ''),
|
|
|
|
|
('', ''), ('', '')])
|
|
|
|
|
|
|
|
|
|
expected = [mock.call('efibootmgr', '--version'),
|
2020-02-08 07:18:12 -08:00
|
|
|
mock.call('partx', '-u', '/dev/fake', attempts=3,
|
|
|
|
|
delay_on_retry=True),
|
|
|
|
|
mock.call('udevadm', 'settle'),
|
2019-12-02 16:56:42 +01:00
|
|
|
mock.call('mount', self.fake_efi_system_part,
|
|
|
|
|
self.fake_dir + '/boot/efi'),
|
|
|
|
|
mock.call('efibootmgr'),
|
|
|
|
|
mock.call('efibootmgr', '-c', '-d', self.fake_dev,
|
|
|
|
|
'-p', '1', '-w',
|
|
|
|
|
'-L', 'ironic1', '-l',
|
|
|
|
|
'\\EFI\\BOOT\\BOOTX64.EFI'),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/boot/efi',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('sync')]
|
|
|
|
|
|
|
|
|
|
self.agent_extension.install_bootloader(
|
|
|
|
|
root_uuid=self.fake_root_uuid,
|
|
|
|
|
efi_system_part_uuid=None)
|
|
|
|
|
|
|
|
|
|
mock_dispatch.assert_any_call('get_os_install_device')
|
|
|
|
|
mock_dispatch.assert_any_call('get_boot_info')
|
|
|
|
|
mkdir_mock.assert_called_once_with(self.fake_dir + '/boot/efi')
|
|
|
|
|
mock_efi_bl.assert_called_once_with(self.fake_dir + '/boot/efi')
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
|
|
|
|
mock_utils_efi_part.assert_called_once_with(self.fake_dev)
|
2020-02-08 07:18:12 -08:00
|
|
|
self.assertEqual(8, mock_execute.call_count)
|
2019-12-02 16:56:42 +01:00
|
|
|
|
2019-08-06 13:52:13 +02:00
|
|
|
@mock.patch.object(hardware, 'is_md_device', lambda *_: False)
|
2019-12-02 16:56:42 +01:00
|
|
|
@mock.patch.object(os.path, 'exists', lambda *_: False)
|
|
|
|
|
@mock.patch.object(iscsi, 'clean_up', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_efi_bootloaders', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_partition', autospec=True)
|
|
|
|
|
@mock.patch.object(utils, 'get_efi_part_on_device', autospec=True)
|
|
|
|
|
@mock.patch.object(os, 'makedirs', autospec=True)
|
|
|
|
|
def test__uefi_bootloader_with_entry_removal(
|
|
|
|
|
self, mkdir_mock, mock_utils_efi_part, mock_partition,
|
|
|
|
|
mock_efi_bl, mock_iscsi_clean, mock_execute, mock_dispatch):
|
|
|
|
|
mock_dispatch.side_effect = [
|
|
|
|
|
self.fake_dev, hardware.BootInfo(current_boot_mode='uefi')
|
|
|
|
|
]
|
|
|
|
|
mock_partition.return_value = self.fake_dev
|
|
|
|
|
mock_utils_efi_part.return_value = '1'
|
|
|
|
|
mock_efi_bl.return_value = ['\\EFI\\BOOT\\BOOTX64.EFI']
|
|
|
|
|
stdeer_msg = """
|
|
|
|
|
efibootmgr: ** Warning ** : Boot0004 has same label ironic1\n
|
|
|
|
|
efibootmgr: ** Warning ** : Boot0005 has same label ironic1\n
|
|
|
|
|
"""
|
|
|
|
|
mock_execute.side_effect = iter([('', ''), ('', ''),
|
2020-02-08 07:18:12 -08:00
|
|
|
('', ''), ('', ''),
|
2019-12-02 16:56:42 +01:00
|
|
|
('', ''), ('', stdeer_msg),
|
|
|
|
|
('', ''), ('', ''),
|
|
|
|
|
('', ''), ('', '')])
|
|
|
|
|
|
|
|
|
|
expected = [mock.call('efibootmgr', '--version'),
|
2020-02-08 07:18:12 -08:00
|
|
|
mock.call('partx', '-u', '/dev/fake', attempts=3,
|
|
|
|
|
delay_on_retry=True),
|
|
|
|
|
mock.call('udevadm', 'settle'),
|
2019-12-02 16:56:42 +01:00
|
|
|
mock.call('mount', self.fake_efi_system_part,
|
|
|
|
|
self.fake_dir + '/boot/efi'),
|
|
|
|
|
mock.call('efibootmgr'),
|
|
|
|
|
mock.call('efibootmgr', '-c', '-d', self.fake_dev,
|
|
|
|
|
'-p', '1', '-w',
|
|
|
|
|
'-L', 'ironic1', '-l',
|
|
|
|
|
'\\EFI\\BOOT\\BOOTX64.EFI'),
|
|
|
|
|
mock.call('efibootmgr', '-b', '0004', '-B'),
|
|
|
|
|
mock.call('efibootmgr', '-b', '0005', '-B'),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/boot/efi',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('sync')]
|
|
|
|
|
|
|
|
|
|
self.agent_extension.install_bootloader(
|
|
|
|
|
root_uuid=self.fake_root_uuid,
|
|
|
|
|
efi_system_part_uuid=None)
|
|
|
|
|
|
|
|
|
|
mock_dispatch.assert_any_call('get_os_install_device')
|
|
|
|
|
mock_dispatch.assert_any_call('get_boot_info')
|
|
|
|
|
mkdir_mock.assert_called_once_with(self.fake_dir + '/boot/efi')
|
|
|
|
|
mock_efi_bl.assert_called_once_with(self.fake_dir + '/boot/efi')
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
|
|
|
|
mock_utils_efi_part.assert_called_once_with(self.fake_dev)
|
2020-02-08 07:18:12 -08:00
|
|
|
self.assertEqual(10, mock_execute.call_count)
|
2019-12-02 16:56:42 +01:00
|
|
|
|
2019-08-06 13:52:13 +02:00
|
|
|
@mock.patch.object(hardware, 'is_md_device', lambda *_: False)
|
2019-12-02 16:56:42 +01:00
|
|
|
@mock.patch.object(os.path, 'exists', lambda *_: False)
|
|
|
|
|
@mock.patch.object(iscsi, 'clean_up', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_efi_bootloaders', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_partition', autospec=True)
|
|
|
|
|
@mock.patch.object(utils, 'get_efi_part_on_device', autospec=True)
|
|
|
|
|
@mock.patch.object(os, 'makedirs', autospec=True)
|
|
|
|
|
def test__add_multi_bootloaders(
|
|
|
|
|
self, mkdir_mock, mock_utils_efi_part, mock_partition,
|
|
|
|
|
mock_efi_bl, mock_iscsi_clean, mock_execute, mock_dispatch):
|
|
|
|
|
mock_dispatch.side_effect = [
|
|
|
|
|
self.fake_dev, hardware.BootInfo(current_boot_mode='uefi')
|
|
|
|
|
]
|
|
|
|
|
mock_partition.return_value = self.fake_dev
|
|
|
|
|
mock_utils_efi_part.return_value = '1'
|
|
|
|
|
mock_efi_bl.return_value = ['\\EFI\\BOOT\\BOOTX64.EFI',
|
|
|
|
|
'\\WINDOWS\\system32\\winload.efi']
|
|
|
|
|
|
|
|
|
|
mock_execute.side_effect = iter([('', ''), ('', ''),
|
2020-02-08 07:18:12 -08:00
|
|
|
('', ''), ('', ''),
|
2019-12-02 16:56:42 +01:00
|
|
|
('', ''), ('', ''),
|
|
|
|
|
('', ''), ('', ''),
|
|
|
|
|
('', '')])
|
|
|
|
|
|
|
|
|
|
expected = [mock.call('efibootmgr', '--version'),
|
2020-02-08 07:18:12 -08:00
|
|
|
mock.call('partx', '-u', '/dev/fake', attempts=3,
|
|
|
|
|
delay_on_retry=True),
|
|
|
|
|
mock.call('udevadm', 'settle'),
|
2019-12-02 16:56:42 +01:00
|
|
|
mock.call('mount', self.fake_efi_system_part,
|
|
|
|
|
self.fake_dir + '/boot/efi'),
|
|
|
|
|
mock.call('efibootmgr'),
|
|
|
|
|
mock.call('efibootmgr', '-c', '-d', self.fake_dev,
|
|
|
|
|
'-p', '1', '-w',
|
|
|
|
|
'-L', 'ironic1', '-l',
|
|
|
|
|
'\\EFI\\BOOT\\BOOTX64.EFI'),
|
|
|
|
|
mock.call('efibootmgr', '-c', '-d', self.fake_dev,
|
|
|
|
|
'-p', '1', '-w',
|
|
|
|
|
'-L', 'ironic2', '-l',
|
|
|
|
|
'\\WINDOWS\\system32\\winload.efi'),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/boot/efi',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('sync')]
|
|
|
|
|
|
|
|
|
|
self.agent_extension.install_bootloader(
|
|
|
|
|
root_uuid=self.fake_root_uuid,
|
|
|
|
|
efi_system_part_uuid=None)
|
|
|
|
|
|
|
|
|
|
mock_dispatch.assert_any_call('get_os_install_device')
|
|
|
|
|
mock_dispatch.assert_any_call('get_boot_info')
|
|
|
|
|
mkdir_mock.assert_called_once_with(self.fake_dir + '/boot/efi')
|
|
|
|
|
mock_efi_bl.assert_called_once_with(self.fake_dir + '/boot/efi')
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
|
|
|
|
mock_utils_efi_part.assert_called_once_with(self.fake_dev)
|
2020-02-08 07:18:12 -08:00
|
|
|
self.assertEqual(9, mock_execute.call_count)
|
2019-12-02 16:56:42 +01:00
|
|
|
|
2018-07-11 13:35:54 +00:00
|
|
|
@mock.patch.object(iscsi, 'clean_up', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_install_grub2', autospec=True)
|
2019-12-02 16:56:42 +01:00
|
|
|
def test__install_bootloader_prep(self, mock_grub2, mock_iscsi_clean,
|
|
|
|
|
mock_execute, mock_dispatch):
|
|
|
|
|
mock_dispatch.side_effect = [
|
|
|
|
|
self.fake_dev, hardware.BootInfo(current_boot_mode='bios')
|
|
|
|
|
]
|
2018-07-11 13:35:54 +00:00
|
|
|
self.agent_extension.install_bootloader(
|
|
|
|
|
root_uuid=self.fake_root_uuid,
|
|
|
|
|
efi_system_part_uuid=None,
|
|
|
|
|
prep_boot_part_uuid=self.fake_prep_boot_part_uuid)
|
2019-12-02 16:56:42 +01:00
|
|
|
mock_dispatch.assert_any_call('get_os_install_device')
|
|
|
|
|
mock_dispatch.assert_any_call('get_boot_info')
|
|
|
|
|
self.assertEqual(2, mock_dispatch.call_count)
|
2018-07-11 13:35:54 +00:00
|
|
|
mock_grub2.assert_called_once_with(
|
|
|
|
|
self.fake_dev,
|
|
|
|
|
root_uuid=self.fake_root_uuid,
|
|
|
|
|
efi_system_part_uuid=None,
|
2019-08-06 13:52:13 +02:00
|
|
|
prep_boot_part_uuid=self.fake_prep_boot_part_uuid,
|
|
|
|
|
target_boot_mode='bios'
|
|
|
|
|
)
|
2015-12-15 13:47:40 +01:00
|
|
|
mock_iscsi_clean.assert_called_once_with(self.fake_dev)
|
2015-03-12 09:15:49 +00:00
|
|
|
|
2019-08-06 13:52:13 +02:00
|
|
|
@mock.patch.object(hardware, 'is_md_device', lambda *_: False)
|
2020-02-21 05:24:05 -06:00
|
|
|
@mock.patch.object(os.path, 'exists', lambda *_: False)
|
|
|
|
|
@mock.patch.object(iscsi, 'clean_up', autospec=True)
|
|
|
|
|
def test_install_bootloader_failure(self, mock_iscsi_clean, mock_execute,
|
|
|
|
|
mock_dispatch):
|
|
|
|
|
mock_dispatch.side_effect = [
|
|
|
|
|
self.fake_dev, hardware.BootInfo(current_boot_mode='uefi')
|
|
|
|
|
]
|
|
|
|
|
mock_execute.side_effect = FileNotFoundError
|
|
|
|
|
self.assertRaises(FileNotFoundError,
|
|
|
|
|
self.agent_extension.install_bootloader,
|
|
|
|
|
root_uuid=self.fake_root_uuid,
|
|
|
|
|
efi_system_part_uuid=None)
|
|
|
|
|
expected = [mock.call('efibootmgr', '--version')]
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
|
|
|
|
|
2019-08-29 16:22:50 -04:00
|
|
|
@mock.patch.object(image, '_is_bootloader_loaded', lambda *_: False)
|
2019-02-26 17:56:20 +01:00
|
|
|
@mock.patch.object(hardware, 'is_md_device', autospec=True)
|
2019-10-17 11:14:04 +00:00
|
|
|
@mock.patch.object(hardware, 'md_get_raid_devices', autospec=True)
|
2017-03-19 08:30:25 -07:00
|
|
|
@mock.patch.object(os, 'environ', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_partition', autospec=True)
|
2015-03-12 09:15:49 +00:00
|
|
|
def test__install_grub2(self, mock_get_part_uuid, environ_mock,
|
2019-10-17 11:14:04 +00:00
|
|
|
mock_md_get_raid_devices, mock_is_md_device,
|
|
|
|
|
mock_execute, mock_dispatch):
|
2015-03-12 09:15:49 +00:00
|
|
|
mock_get_part_uuid.return_value = self.fake_root_part
|
|
|
|
|
environ_mock.get.return_value = '/sbin'
|
2019-08-29 16:22:50 -04:00
|
|
|
mock_is_md_device.return_value = False
|
2019-10-17 11:14:04 +00:00
|
|
|
mock_md_get_raid_devices.return_value = {}
|
2015-02-19 13:42:39 +00:00
|
|
|
image._install_grub2(self.fake_dev, self.fake_root_uuid)
|
|
|
|
|
|
|
|
|
|
expected = [mock.call('mount', '/dev/fake2', self.fake_dir),
|
|
|
|
|
mock.call('mount', '-o', 'bind', '/dev',
|
|
|
|
|
self.fake_dir + '/dev'),
|
|
|
|
|
mock.call('mount', '-o', 'bind', '/proc',
|
|
|
|
|
self.fake_dir + '/proc'),
|
2019-04-24 23:25:19 +01:00
|
|
|
mock.call('mount', '-o', 'bind', '/run',
|
|
|
|
|
self.fake_dir + '/run'),
|
2015-10-06 14:50:55 -04:00
|
|
|
mock.call('mount', '-t', 'sysfs', 'none',
|
|
|
|
|
self.fake_dir + '/sys'),
|
2017-01-16 21:32:37 +02:00
|
|
|
mock.call(('chroot %s /bin/sh -c '
|
2017-09-14 07:05:25 -06:00
|
|
|
'"grub-install %s"' %
|
2015-10-02 10:01:00 -07:00
|
|
|
(self.fake_dir, self.fake_dev)), shell=True,
|
2019-08-06 13:52:13 +02:00
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
2017-01-16 21:32:37 +02:00
|
|
|
mock.call(('chroot %s /bin/sh -c '
|
2017-09-14 07:05:25 -06:00
|
|
|
'"grub-mkconfig -o '
|
2015-02-19 13:42:39 +00:00
|
|
|
'/boot/grub/grub.cfg"' % self.fake_dir),
|
2015-03-12 09:15:49 +00:00
|
|
|
shell=True,
|
2019-08-06 13:52:13 +02:00
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
2015-02-19 13:42:39 +00:00
|
|
|
mock.call('umount', self.fake_dir + '/dev',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/proc',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
2019-04-24 23:25:19 +01:00
|
|
|
mock.call('umount', self.fake_dir + '/run',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
2015-10-06 14:50:55 -04:00
|
|
|
mock.call('umount', self.fake_dir + '/sys',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
2015-02-19 13:42:39 +00:00
|
|
|
mock.call('umount', self.fake_dir, attempts=3,
|
|
|
|
|
delay_on_retry=True)]
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
2015-03-12 09:15:49 +00:00
|
|
|
mock_get_part_uuid.assert_called_once_with(self.fake_dev,
|
|
|
|
|
uuid=self.fake_root_uuid)
|
|
|
|
|
self.assertFalse(mock_dispatch.called)
|
|
|
|
|
|
2019-08-29 16:22:50 -04:00
|
|
|
@mock.patch.object(image, '_is_bootloader_loaded', lambda *_: False)
|
2019-02-26 17:56:20 +01:00
|
|
|
@mock.patch.object(hardware, 'is_md_device', autospec=True)
|
2019-10-17 11:14:04 +00:00
|
|
|
@mock.patch.object(hardware, 'md_get_raid_devices', autospec=True)
|
2018-07-11 13:35:54 +00:00
|
|
|
@mock.patch.object(os, 'environ', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_partition', autospec=True)
|
|
|
|
|
def test__install_grub2_prep(self, mock_get_part_uuid, environ_mock,
|
2019-10-17 11:14:04 +00:00
|
|
|
mock_md_get_raid_devices, mock_is_md_device,
|
|
|
|
|
mock_execute, mock_dispatch):
|
2018-07-11 13:35:54 +00:00
|
|
|
mock_get_part_uuid.side_effect = [self.fake_root_part,
|
|
|
|
|
self.fake_prep_boot_part]
|
|
|
|
|
environ_mock.get.return_value = '/sbin'
|
2019-08-29 16:22:50 -04:00
|
|
|
mock_is_md_device.return_value = False
|
2019-10-17 11:14:04 +00:00
|
|
|
mock_md_get_raid_devices.return_value = {}
|
2018-07-11 13:35:54 +00:00
|
|
|
image._install_grub2(self.fake_dev, self.fake_root_uuid,
|
|
|
|
|
prep_boot_part_uuid=self.fake_prep_boot_part_uuid)
|
|
|
|
|
|
|
|
|
|
expected = [mock.call('mount', '/dev/fake2', self.fake_dir),
|
|
|
|
|
mock.call('mount', '-o', 'bind', '/dev',
|
|
|
|
|
self.fake_dir + '/dev'),
|
|
|
|
|
mock.call('mount', '-o', 'bind', '/proc',
|
|
|
|
|
self.fake_dir + '/proc'),
|
2019-04-24 23:25:19 +01:00
|
|
|
mock.call('mount', '-o', 'bind', '/run',
|
|
|
|
|
self.fake_dir + '/run'),
|
2018-07-11 13:35:54 +00:00
|
|
|
mock.call('mount', '-t', 'sysfs', 'none',
|
|
|
|
|
self.fake_dir + '/sys'),
|
|
|
|
|
mock.call(('chroot %s /bin/sh -c '
|
|
|
|
|
'"grub-install %s"' %
|
|
|
|
|
(self.fake_dir, self.fake_prep_boot_part)),
|
|
|
|
|
shell=True,
|
2019-08-06 13:52:13 +02:00
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
2018-07-11 13:35:54 +00:00
|
|
|
mock.call(('chroot %s /bin/sh -c '
|
|
|
|
|
'"grub-mkconfig -o '
|
|
|
|
|
'/boot/grub/grub.cfg"' % self.fake_dir),
|
|
|
|
|
shell=True,
|
2019-08-06 13:52:13 +02:00
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
2018-07-11 13:35:54 +00:00
|
|
|
mock.call('umount', self.fake_dir + '/dev',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/proc',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
2019-04-24 23:25:19 +01:00
|
|
|
mock.call('umount', self.fake_dir + '/run',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
2018-07-11 13:35:54 +00:00
|
|
|
mock.call('umount', self.fake_dir + '/sys',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir, attempts=3,
|
|
|
|
|
delay_on_retry=True)]
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
|
|
|
|
mock_get_part_uuid.assert_any_call(self.fake_dev,
|
|
|
|
|
uuid=self.fake_root_uuid)
|
|
|
|
|
mock_get_part_uuid.assert_any_call(self.fake_dev,
|
|
|
|
|
uuid=self.fake_prep_boot_part_uuid)
|
|
|
|
|
self.assertFalse(mock_dispatch.called)
|
|
|
|
|
|
2019-12-02 16:56:42 +01:00
|
|
|
@mock.patch.object(os.path, 'exists', lambda *_: False)
|
2019-08-29 16:22:50 -04:00
|
|
|
@mock.patch.object(image, '_is_bootloader_loaded', lambda *_: True)
|
2019-02-26 17:56:20 +01:00
|
|
|
@mock.patch.object(hardware, 'is_md_device', autospec=True)
|
2019-10-17 11:14:04 +00:00
|
|
|
@mock.patch.object(hardware, 'md_get_raid_devices', autospec=True)
|
2017-03-19 08:30:25 -07:00
|
|
|
@mock.patch.object(os, 'environ', autospec=True)
|
|
|
|
|
@mock.patch.object(os, 'makedirs', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_partition', autospec=True)
|
2015-03-12 09:15:49 +00:00
|
|
|
def test__install_grub2_uefi(self, mock_get_part_uuid, mkdir_mock,
|
2019-10-17 11:14:04 +00:00
|
|
|
environ_mock, mock_md_get_raid_devices,
|
|
|
|
|
mock_is_md_device, mock_execute,
|
|
|
|
|
mock_dispatch):
|
2015-03-12 09:15:49 +00:00
|
|
|
mock_get_part_uuid.side_effect = [self.fake_root_part,
|
|
|
|
|
self.fake_efi_system_part]
|
|
|
|
|
environ_mock.get.return_value = '/sbin'
|
2019-02-26 17:56:20 +01:00
|
|
|
mock_is_md_device.return_value = False
|
2019-10-17 11:14:04 +00:00
|
|
|
mock_md_get_raid_devices.return_value = {}
|
2015-03-12 09:15:49 +00:00
|
|
|
|
|
|
|
|
image._install_grub2(
|
|
|
|
|
self.fake_dev, root_uuid=self.fake_root_uuid,
|
2019-08-06 13:52:13 +02:00
|
|
|
efi_system_part_uuid=self.fake_efi_system_part_uuid,
|
|
|
|
|
target_boot_mode='uefi')
|
2015-03-12 09:15:49 +00:00
|
|
|
|
|
|
|
|
expected = [mock.call('mount', '/dev/fake2', self.fake_dir),
|
|
|
|
|
mock.call('mount', '-o', 'bind', '/dev',
|
|
|
|
|
self.fake_dir + '/dev'),
|
|
|
|
|
mock.call('mount', '-o', 'bind', '/proc',
|
|
|
|
|
self.fake_dir + '/proc'),
|
2019-04-24 23:25:19 +01:00
|
|
|
mock.call('mount', '-o', 'bind', '/run',
|
|
|
|
|
self.fake_dir + '/run'),
|
2015-10-06 14:50:55 -04:00
|
|
|
mock.call('mount', '-t', 'sysfs', 'none',
|
|
|
|
|
self.fake_dir + '/sys'),
|
2015-03-12 09:15:49 +00:00
|
|
|
mock.call('mount', self.fake_efi_system_part,
|
|
|
|
|
self.fake_dir + '/boot/efi'),
|
2019-08-06 13:52:13 +02:00
|
|
|
mock.call(('chroot %s /bin/sh -c "grub-install"' %
|
|
|
|
|
self.fake_dir), shell=True,
|
|
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
2017-01-16 21:32:37 +02:00
|
|
|
mock.call(('chroot %s /bin/sh -c '
|
2019-08-06 13:52:13 +02:00
|
|
|
'"grub-install --removable"' %
|
|
|
|
|
self.fake_dir), shell=True,
|
|
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
|
|
|
|
mock.call(
|
|
|
|
|
'umount', self.fake_dir + '/boot/efi',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('mount', self.fake_efi_system_part,
|
|
|
|
|
'/tmp/fake-dir/boot/efi'),
|
2017-01-16 21:32:37 +02:00
|
|
|
mock.call(('chroot %s /bin/sh -c '
|
2017-09-14 07:05:25 -06:00
|
|
|
'"grub-mkconfig -o '
|
2015-03-12 09:15:49 +00:00
|
|
|
'/boot/grub/grub.cfg"' % self.fake_dir),
|
|
|
|
|
shell=True,
|
2019-08-06 13:52:13 +02:00
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
2015-03-12 09:15:49 +00:00
|
|
|
mock.call('umount', self.fake_dir + '/boot/efi',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/dev',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/proc',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
2019-04-24 23:25:19 +01:00
|
|
|
mock.call('umount', self.fake_dir + '/run',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
2015-10-06 14:50:55 -04:00
|
|
|
mock.call('umount', self.fake_dir + '/sys',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
2015-03-12 09:15:49 +00:00
|
|
|
mock.call('umount', self.fake_dir, attempts=3,
|
|
|
|
|
delay_on_retry=True)]
|
|
|
|
|
mkdir_mock.assert_called_once_with(self.fake_dir + '/boot/efi')
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
|
|
|
|
mock_get_part_uuid.assert_any_call(self.fake_dev,
|
|
|
|
|
uuid=self.fake_root_uuid)
|
|
|
|
|
mock_get_part_uuid.assert_any_call(self.fake_dev,
|
|
|
|
|
uuid=self.fake_efi_system_part_uuid)
|
2015-02-19 13:42:39 +00:00
|
|
|
self.assertFalse(mock_dispatch.called)
|
|
|
|
|
|
2019-08-29 16:22:50 -04:00
|
|
|
@mock.patch.object(image, '_is_bootloader_loaded', lambda *_: False)
|
2019-02-26 17:56:20 +01:00
|
|
|
@mock.patch.object(hardware, 'is_md_device', autospec=True)
|
2019-10-17 11:14:04 +00:00
|
|
|
@mock.patch.object(hardware, 'md_get_raid_devices', autospec=True)
|
2017-03-19 08:30:25 -07:00
|
|
|
@mock.patch.object(os, 'environ', autospec=True)
|
|
|
|
|
@mock.patch.object(os, 'makedirs', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_partition', autospec=True)
|
2015-03-12 09:15:49 +00:00
|
|
|
def test__install_grub2_uefi_umount_fails(
|
2019-02-26 17:56:20 +01:00
|
|
|
self, mock_get_part_uuid, mkdir_mock, environ_mock,
|
2019-10-17 11:14:04 +00:00
|
|
|
mock_md_get_raid_devices, mock_is_md_device, mock_execute,
|
|
|
|
|
mock_dispatch):
|
2015-03-12 09:15:49 +00:00
|
|
|
mock_get_part_uuid.side_effect = [self.fake_root_part,
|
|
|
|
|
self.fake_efi_system_part]
|
2019-02-26 17:56:20 +01:00
|
|
|
mock_is_md_device.return_value = False
|
2019-10-17 11:14:04 +00:00
|
|
|
mock_md_get_raid_devices.return_value = {}
|
2015-03-12 09:15:49 +00:00
|
|
|
|
|
|
|
|
def umount_raise_func(*args, **kwargs):
|
|
|
|
|
if args[0] == 'umount':
|
|
|
|
|
raise processutils.ProcessExecutionError('error')
|
|
|
|
|
|
|
|
|
|
mock_execute.side_effect = umount_raise_func
|
|
|
|
|
environ_mock.get.return_value = '/sbin'
|
|
|
|
|
self.assertRaises(errors.CommandExecutionError,
|
|
|
|
|
image._install_grub2,
|
|
|
|
|
self.fake_dev, root_uuid=self.fake_root_uuid,
|
|
|
|
|
efi_system_part_uuid=self.fake_efi_system_part_uuid)
|
|
|
|
|
|
|
|
|
|
expected = [mock.call('mount', '/dev/fake2', self.fake_dir),
|
|
|
|
|
mock.call('mount', '-o', 'bind', '/dev',
|
|
|
|
|
self.fake_dir + '/dev'),
|
|
|
|
|
mock.call('mount', '-o', 'bind', '/proc',
|
|
|
|
|
self.fake_dir + '/proc'),
|
2019-04-24 23:25:19 +01:00
|
|
|
mock.call('mount', '-o', 'bind', '/run',
|
|
|
|
|
self.fake_dir + '/run'),
|
2015-10-06 14:50:55 -04:00
|
|
|
mock.call('mount', '-t', 'sysfs', 'none',
|
|
|
|
|
self.fake_dir + '/sys'),
|
2015-03-12 09:15:49 +00:00
|
|
|
mock.call('mount', self.fake_efi_system_part,
|
|
|
|
|
self.fake_dir + '/boot/efi'),
|
2019-08-06 13:52:13 +02:00
|
|
|
mock.call(('chroot %s /bin/sh -c "grub-install"' %
|
|
|
|
|
self.fake_dir), shell=True,
|
|
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
2017-01-16 21:32:37 +02:00
|
|
|
mock.call(('chroot %s /bin/sh -c '
|
2019-08-06 13:52:13 +02:00
|
|
|
'"grub-install --removable"' % self.fake_dir),
|
2015-03-12 09:15:49 +00:00
|
|
|
shell=True,
|
2019-08-06 13:52:13 +02:00
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
|
|
|
|
# Call from for loop
|
|
|
|
|
mock.call('umount', self.fake_dir + '/boot/efi',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
# Call from finally
|
2015-03-12 09:15:49 +00:00
|
|
|
mock.call('umount', self.fake_dir + '/boot/efi',
|
2019-08-06 13:52:13 +02:00
|
|
|
attempts=3, delay_on_retry=True)
|
|
|
|
|
]
|
2015-03-12 09:15:49 +00:00
|
|
|
mock_execute.assert_has_calls(expected)
|
|
|
|
|
|
2019-08-29 16:22:50 -04:00
|
|
|
@mock.patch.object(image, '_is_bootloader_loaded', lambda *_: False)
|
2019-02-26 17:56:20 +01:00
|
|
|
@mock.patch.object(hardware, 'is_md_device', autospec=True)
|
2019-10-17 11:14:04 +00:00
|
|
|
@mock.patch.object(hardware, 'md_get_raid_devices', autospec=True)
|
2017-11-17 16:34:24 +01:00
|
|
|
@mock.patch.object(os, 'environ', autospec=True)
|
|
|
|
|
@mock.patch.object(os, 'makedirs', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_partition', autospec=True)
|
|
|
|
|
def test__install_grub2_uefi_mount_fails(
|
2019-02-26 17:56:20 +01:00
|
|
|
self, mock_get_part_uuid, mkdir_mock, environ_mock,
|
2019-10-17 11:14:04 +00:00
|
|
|
mock_is_md_device, mock_md_get_raid_devices, mock_execute,
|
|
|
|
|
mock_dispatch):
|
2017-11-17 16:34:24 +01:00
|
|
|
mock_get_part_uuid.side_effect = [self.fake_root_part,
|
|
|
|
|
self.fake_efi_system_part]
|
2019-10-17 11:14:04 +00:00
|
|
|
mock_is_md_device.side_effect = [False, False]
|
|
|
|
|
mock_md_get_raid_devices.return_value = {}
|
2017-11-17 16:34:24 +01:00
|
|
|
|
|
|
|
|
def mount_raise_func(*args, **kwargs):
|
|
|
|
|
if args[0] == 'mount':
|
|
|
|
|
raise processutils.ProcessExecutionError('error')
|
|
|
|
|
|
|
|
|
|
mock_execute.side_effect = mount_raise_func
|
|
|
|
|
self.assertRaises(errors.CommandExecutionError,
|
|
|
|
|
image._install_grub2,
|
|
|
|
|
self.fake_dev, root_uuid=self.fake_root_uuid,
|
|
|
|
|
efi_system_part_uuid=self.fake_efi_system_part_uuid)
|
|
|
|
|
|
|
|
|
|
expected = [mock.call('mount', '/dev/fake2', self.fake_dir),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/dev',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/proc',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
2019-04-24 23:25:19 +01:00
|
|
|
mock.call('umount', self.fake_dir + '/run',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
2017-11-17 16:34:24 +01:00
|
|
|
mock.call('umount', self.fake_dir + '/sys',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir, attempts=3,
|
|
|
|
|
delay_on_retry=True)]
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
|
|
|
|
|
2019-08-29 16:22:50 -04:00
|
|
|
@mock.patch.object(image, '_is_bootloader_loaded', lambda *_: False)
|
2017-03-19 08:30:25 -07:00
|
|
|
@mock.patch.object(image, '_get_partition', autospec=True)
|
2015-03-12 09:15:49 +00:00
|
|
|
def test__install_grub2_command_fail(self, mock_get_part_uuid,
|
|
|
|
|
mock_execute,
|
2015-02-19 13:42:39 +00:00
|
|
|
mock_dispatch):
|
2015-03-12 09:15:49 +00:00
|
|
|
mock_get_part_uuid.return_value = self.fake_root_part
|
2015-02-19 13:42:39 +00:00
|
|
|
mock_execute.side_effect = processutils.ProcessExecutionError('boom')
|
|
|
|
|
|
|
|
|
|
self.assertRaises(errors.CommandExecutionError, image._install_grub2,
|
|
|
|
|
self.fake_dev, self.fake_root_uuid)
|
|
|
|
|
|
2015-03-12 09:15:49 +00:00
|
|
|
mock_get_part_uuid.assert_called_once_with(self.fake_dev,
|
|
|
|
|
uuid=self.fake_root_uuid)
|
2015-02-19 13:42:39 +00:00
|
|
|
self.assertFalse(mock_dispatch.called)
|
|
|
|
|
|
2019-08-06 13:52:13 +02:00
|
|
|
@mock.patch.object(utils, 'get_efi_part_on_device', autospec=True)
|
|
|
|
|
def test__prepare_boot_partitions_for_softraid_uefi_gpt(
|
|
|
|
|
self, mock_efi_part, mock_execute, mock_dispatch):
|
|
|
|
|
mock_efi_part.return_value = '12'
|
|
|
|
|
mock_execute.side_effect = [
|
|
|
|
|
('451', None), # sgdisk -F
|
|
|
|
|
(None, None), # sgdisk create part
|
|
|
|
|
(None, None), # partprobe
|
|
|
|
|
(None, None), # blkid
|
|
|
|
|
('/dev/sda12: dsfkgsdjfg', None), # blkid
|
|
|
|
|
(None, None), # cp
|
|
|
|
|
('452', None), # sgdisk -F
|
|
|
|
|
(None, None), # sgdisk create part
|
|
|
|
|
(None, None), # partprobe
|
|
|
|
|
(None, None), # blkid
|
|
|
|
|
('/dev/sdb14: whatever', None), # blkid
|
|
|
|
|
(None, None), # cp
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
efi_parts = image._prepare_boot_partitions_for_softraid(
|
|
|
|
|
'/dev/md0', ['/dev/sda', '/dev/sdb'], None,
|
|
|
|
|
target_boot_mode='uefi')
|
|
|
|
|
|
|
|
|
|
mock_efi_part.assert_called_once_with('/dev/md0')
|
|
|
|
|
expected = [
|
|
|
|
|
mock.call('sgdisk', '-F', '/dev/sda'),
|
|
|
|
|
mock.call('sgdisk', '-n', '0:451s:+128MiB', '-t', '0:ef00', '-c',
|
|
|
|
|
'0:uefi-holder-0', '/dev/sda'),
|
|
|
|
|
mock.call('partprobe'),
|
|
|
|
|
mock.call('blkid'),
|
|
|
|
|
mock.call('blkid', '-l', '-t', 'PARTLABEL=uefi-holder-0',
|
|
|
|
|
'/dev/sda'),
|
|
|
|
|
mock.call('cp', '/dev/md0p12', '/dev/sda12'),
|
|
|
|
|
mock.call('sgdisk', '-F', '/dev/sdb'),
|
|
|
|
|
mock.call('sgdisk', '-n', '0:452s:+128MiB', '-t', '0:ef00', '-c',
|
|
|
|
|
'0:uefi-holder-1', '/dev/sdb'),
|
|
|
|
|
mock.call('partprobe'),
|
|
|
|
|
mock.call('blkid'),
|
|
|
|
|
mock.call('blkid', '-l', '-t', 'PARTLABEL=uefi-holder-1',
|
|
|
|
|
'/dev/sdb'),
|
|
|
|
|
mock.call('cp', '/dev/md0p12', '/dev/sdb14')
|
|
|
|
|
]
|
|
|
|
|
mock_execute.assert_has_calls(expected, any_order=False)
|
|
|
|
|
self.assertEqual(efi_parts, ['/dev/sda12', '/dev/sdb14'])
|
|
|
|
|
|
|
|
|
|
@mock.patch.object(utils, 'get_efi_part_on_device', autospec=True)
|
|
|
|
|
@mock.patch.object(ilib_utils, 'mkfs', autospec=True)
|
|
|
|
|
def test__prepare_boot_partitions_for_softraid_uefi_gpt_esp_not_found(
|
|
|
|
|
self, mock_mkfs, mock_efi_part, mock_execute, mock_dispatch):
|
|
|
|
|
mock_efi_part.return_value = None
|
|
|
|
|
mock_execute.side_effect = [
|
|
|
|
|
('451', None), # sgdisk -F
|
|
|
|
|
(None, None), # sgdisk create part
|
|
|
|
|
(None, None), # partprobe
|
|
|
|
|
(None, None), # blkid
|
|
|
|
|
('/dev/sda12: dsfkgsdjfg', None), # blkid
|
|
|
|
|
('452', None), # sgdisk -F
|
|
|
|
|
(None, None), # sgdisk create part
|
|
|
|
|
(None, None), # partprobe
|
|
|
|
|
(None, None), # blkid
|
|
|
|
|
('/dev/sdb14: whatever', None), # blkid
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
efi_parts = image._prepare_boot_partitions_for_softraid(
|
|
|
|
|
'/dev/md0', ['/dev/sda', '/dev/sdb'], None,
|
|
|
|
|
target_boot_mode='uefi')
|
|
|
|
|
|
|
|
|
|
mock_efi_part.assert_called_once_with('/dev/md0')
|
|
|
|
|
expected = [
|
|
|
|
|
mock.call('sgdisk', '-F', '/dev/sda'),
|
|
|
|
|
mock.call('sgdisk', '-n', '0:451s:+128MiB', '-t', '0:ef00', '-c',
|
|
|
|
|
'0:uefi-holder-0', '/dev/sda'),
|
|
|
|
|
mock.call('partprobe'),
|
|
|
|
|
mock.call('blkid'),
|
|
|
|
|
mock.call('blkid', '-l', '-t', 'PARTLABEL=uefi-holder-0',
|
|
|
|
|
'/dev/sda'),
|
|
|
|
|
mock.call('sgdisk', '-F', '/dev/sdb'),
|
|
|
|
|
mock.call('sgdisk', '-n', '0:452s:+128MiB', '-t', '0:ef00', '-c',
|
|
|
|
|
'0:uefi-holder-1', '/dev/sdb'),
|
|
|
|
|
mock.call('partprobe'),
|
|
|
|
|
mock.call('blkid'),
|
|
|
|
|
mock.call('blkid', '-l', '-t', 'PARTLABEL=uefi-holder-1',
|
|
|
|
|
'/dev/sdb'),
|
|
|
|
|
]
|
|
|
|
|
mock_execute.assert_has_calls(expected, any_order=False)
|
|
|
|
|
mock_mkfs.assert_has_calls([
|
|
|
|
|
mock.call(path='/dev/sda12', label='efi-part', fs='vfat'),
|
|
|
|
|
mock.call(path='/dev/sdb14', label='efi-part-b', fs='vfat'),
|
|
|
|
|
], any_order=False)
|
|
|
|
|
self.assertEqual(efi_parts, ['/dev/sda12', '/dev/sdb14'])
|
|
|
|
|
|
|
|
|
|
def test__prepare_boot_partitions_for_softraid_uefi_gpt_efi_provided(
|
|
|
|
|
self, mock_execute, mock_dispatch):
|
|
|
|
|
mock_execute.side_effect = [
|
|
|
|
|
('451', None), # sgdisk -F
|
|
|
|
|
(None, None), # sgdisk create part
|
|
|
|
|
(None, None), # partprobe
|
|
|
|
|
(None, None), # blkid
|
|
|
|
|
('/dev/sda12: dsfkgsdjfg', None), # blkid
|
|
|
|
|
(None, None), # cp
|
|
|
|
|
('452', None), # sgdisk -F
|
|
|
|
|
(None, None), # sgdisk create part
|
|
|
|
|
(None, None), # partprobe
|
|
|
|
|
(None, None), # blkid
|
|
|
|
|
('/dev/sdb14: whatever', None), # blkid
|
|
|
|
|
(None, None), # cp
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
efi_parts = image._prepare_boot_partitions_for_softraid(
|
|
|
|
|
'/dev/md0', ['/dev/sda', '/dev/sdb'], '/dev/md0p15',
|
|
|
|
|
target_boot_mode='uefi')
|
|
|
|
|
|
|
|
|
|
expected = [
|
|
|
|
|
mock.call('sgdisk', '-F', '/dev/sda'),
|
|
|
|
|
mock.call('sgdisk', '-n', '0:451s:+128MiB', '-t', '0:ef00', '-c',
|
|
|
|
|
'0:uefi-holder-0', '/dev/sda'),
|
|
|
|
|
mock.call('partprobe'),
|
|
|
|
|
mock.call('blkid'),
|
|
|
|
|
mock.call('blkid', '-l', '-t', 'PARTLABEL=uefi-holder-0',
|
|
|
|
|
'/dev/sda'),
|
|
|
|
|
mock.call('cp', '/dev/md0p15', '/dev/sda12'),
|
|
|
|
|
mock.call('sgdisk', '-F', '/dev/sdb'),
|
|
|
|
|
mock.call('sgdisk', '-n', '0:452s:+128MiB', '-t', '0:ef00', '-c',
|
|
|
|
|
'0:uefi-holder-1', '/dev/sdb'),
|
|
|
|
|
mock.call('partprobe'),
|
|
|
|
|
mock.call('blkid'),
|
|
|
|
|
mock.call('blkid', '-l', '-t', 'PARTLABEL=uefi-holder-1',
|
|
|
|
|
'/dev/sdb'),
|
|
|
|
|
mock.call('cp', '/dev/md0p15', '/dev/sdb14')
|
|
|
|
|
]
|
|
|
|
|
mock_execute.assert_has_calls(expected, any_order=False)
|
|
|
|
|
self.assertEqual(efi_parts, ['/dev/sda12', '/dev/sdb14'])
|
|
|
|
|
|
|
|
|
|
@mock.patch.object(utils, 'scan_partition_table_type', autospec=True,
|
|
|
|
|
return_value='msdos')
|
|
|
|
|
def test__prepare_boot_partitions_for_softraid_bios_msdos(
|
|
|
|
|
self, mock_label_scan, mock_execute, mock_dispatch):
|
|
|
|
|
|
|
|
|
|
efi_parts = image._prepare_boot_partitions_for_softraid(
|
|
|
|
|
'/dev/md0', ['/dev/sda', '/dev/sdb'], 'notusedanyway',
|
|
|
|
|
target_boot_mode='bios')
|
|
|
|
|
|
|
|
|
|
expected = [
|
|
|
|
|
mock.call('/dev/sda'),
|
|
|
|
|
mock.call('/dev/sdb'),
|
|
|
|
|
]
|
|
|
|
|
mock_label_scan.assert_has_calls(expected, any_order=False)
|
|
|
|
|
self.assertEqual(efi_parts, [])
|
|
|
|
|
|
|
|
|
|
@mock.patch.object(utils, 'scan_partition_table_type', autospec=True,
|
|
|
|
|
return_value='gpt')
|
|
|
|
|
def test__prepare_boot_partitions_for_softraid_bios_gpt(
|
|
|
|
|
self, mock_label_scan, mock_execute, mock_dispatch):
|
|
|
|
|
|
|
|
|
|
mock_execute.side_effect = [
|
|
|
|
|
('whatever\n314', None), # sgdisk -F
|
|
|
|
|
(None, None), # bios boot grub
|
|
|
|
|
('warning message\n914', None), # sgdisk -F
|
|
|
|
|
(None, None), # bios boot grub
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
efi_parts = image._prepare_boot_partitions_for_softraid(
|
|
|
|
|
'/dev/md0', ['/dev/sda', '/dev/sdb'], 'notusedanyway',
|
|
|
|
|
target_boot_mode='bios')
|
|
|
|
|
|
|
|
|
|
expected_scan = [
|
|
|
|
|
mock.call('/dev/sda'),
|
|
|
|
|
mock.call('/dev/sdb'),
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
mock_label_scan.assert_has_calls(expected_scan, any_order=False)
|
|
|
|
|
|
|
|
|
|
expected_exec = [
|
|
|
|
|
mock.call('sgdisk', '-F', '/dev/sda'),
|
|
|
|
|
mock.call('sgdisk', '-n', '0:314s:+2MiB', '-t', '0:ef02', '-c',
|
|
|
|
|
'0:bios-boot-part-0', '/dev/sda'),
|
|
|
|
|
mock.call('sgdisk', '-F', '/dev/sdb'),
|
|
|
|
|
mock.call('sgdisk', '-n', '0:914s:+2MiB', '-t', '0:ef02', '-c',
|
|
|
|
|
'0:bios-boot-part-1', '/dev/sdb'),
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
mock_execute.assert_has_calls(expected_exec, any_order=False)
|
|
|
|
|
self.assertEqual(efi_parts, [])
|
|
|
|
|
|
|
|
|
|
@mock.patch.object(image, '_is_bootloader_loaded', lambda *_: True)
|
|
|
|
|
@mock.patch.object(hardware, 'is_md_device', autospec=True)
|
|
|
|
|
@mock.patch.object(hardware, 'md_restart', autospec=True)
|
|
|
|
|
@mock.patch.object(hardware, 'md_get_raid_devices', autospec=True)
|
|
|
|
|
@mock.patch.object(hardware, 'get_holder_disks', autospec=True,
|
|
|
|
|
return_value=['/dev/sda', '/dev/sdb'])
|
|
|
|
|
@mock.patch.object(os, 'environ', autospec=True)
|
|
|
|
|
@mock.patch.object(os, 'makedirs', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_partition', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_prepare_boot_partitions_for_softraid',
|
|
|
|
|
autospec=True,
|
|
|
|
|
return_value=['/dev/sda1', '/dev/sdb2'])
|
|
|
|
|
@mock.patch.object(image, '_has_dracut',
|
|
|
|
|
autospec=True,
|
|
|
|
|
return_value=False)
|
|
|
|
|
def test__install_grub2_softraid_uefi_gpt(
|
|
|
|
|
self, mock_dracut,
|
|
|
|
|
mock_prepare, mock_get_part_uuid, mkdir_mock, environ_mock,
|
|
|
|
|
mock_holder, mock_md_get_raid_devices, mock_restart,
|
|
|
|
|
mock_is_md_device,
|
|
|
|
|
mock_execute, mock_dispatch):
|
|
|
|
|
|
|
|
|
|
mock_get_part_uuid.side_effect = [self.fake_root_part,
|
|
|
|
|
self.fake_efi_system_part]
|
|
|
|
|
environ_mock.get.return_value = '/sbin'
|
|
|
|
|
mock_is_md_device.return_value = True
|
|
|
|
|
mock_md_get_raid_devices.return_value = {}
|
|
|
|
|
|
|
|
|
|
image._install_grub2(
|
|
|
|
|
self.fake_dev, root_uuid=self.fake_root_uuid,
|
|
|
|
|
efi_system_part_uuid=self.fake_efi_system_part_uuid,
|
|
|
|
|
target_boot_mode='uefi')
|
|
|
|
|
|
|
|
|
|
expected = [mock.call('mount', '/dev/fake2', self.fake_dir),
|
|
|
|
|
mock.call('mount', '-o', 'bind', '/dev',
|
|
|
|
|
self.fake_dir + '/dev'),
|
|
|
|
|
mock.call('mount', '-o', 'bind', '/proc',
|
|
|
|
|
self.fake_dir + '/proc'),
|
|
|
|
|
mock.call('mount', '-o', 'bind', '/run',
|
|
|
|
|
self.fake_dir + '/run'),
|
|
|
|
|
mock.call('mount', '-t', 'sysfs', 'none',
|
|
|
|
|
self.fake_dir + '/sys'),
|
|
|
|
|
mock.call('mount', '/dev/sda1',
|
|
|
|
|
self.fake_dir + '/boot/efi'),
|
|
|
|
|
mock.call(('chroot %s /bin/sh -c "grub-install"' %
|
|
|
|
|
self.fake_dir), shell=True,
|
|
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
|
|
|
|
mock.call(('chroot %s /bin/sh -c '
|
|
|
|
|
'"grub-install --removable"' %
|
|
|
|
|
self.fake_dir), shell=True,
|
|
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
|
|
|
|
mock.call(
|
|
|
|
|
'umount', self.fake_dir + '/boot/efi',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('mount', '/dev/sdb2',
|
|
|
|
|
self.fake_dir + '/boot/efi'),
|
|
|
|
|
mock.call(('chroot %s /bin/sh -c "grub-install"' %
|
|
|
|
|
self.fake_dir), shell=True,
|
|
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
|
|
|
|
mock.call(('chroot %s /bin/sh -c '
|
|
|
|
|
'"grub-install --removable"' %
|
|
|
|
|
self.fake_dir), shell=True,
|
|
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
|
|
|
|
mock.call(
|
|
|
|
|
'umount', self.fake_dir + '/boot/efi',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('mount', '/dev/sda1',
|
|
|
|
|
'/tmp/fake-dir/boot/efi'),
|
|
|
|
|
mock.call(('chroot %s /bin/sh -c '
|
|
|
|
|
'"grub-mkconfig -o '
|
|
|
|
|
'/boot/grub/grub.cfg"' % self.fake_dir),
|
|
|
|
|
shell=True,
|
|
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/boot/efi',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/dev',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/proc',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/run',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/sys',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir, attempts=3,
|
|
|
|
|
delay_on_retry=True)]
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
|
|
|
|
mock_get_part_uuid.assert_any_call(self.fake_dev,
|
|
|
|
|
uuid=self.fake_root_uuid)
|
|
|
|
|
mock_get_part_uuid.assert_any_call(self.fake_dev,
|
|
|
|
|
uuid=self.fake_efi_system_part_uuid)
|
|
|
|
|
self.assertFalse(mock_dispatch.called)
|
|
|
|
|
mock_prepare.assert_called_once_with(self.fake_dev,
|
|
|
|
|
['/dev/sda', '/dev/sdb'],
|
|
|
|
|
self.fake_efi_system_part, 'uefi')
|
|
|
|
|
mock_restart.assert_called_once_with(self.fake_dev)
|
|
|
|
|
mock_holder.assert_called_once_with(self.fake_dev)
|
|
|
|
|
mock_dracut.assert_called_once_with(self.fake_dir)
|
|
|
|
|
|
|
|
|
|
@mock.patch.object(image, '_is_bootloader_loaded', lambda *_: True)
|
|
|
|
|
@mock.patch.object(hardware, 'is_md_device', autospec=True)
|
|
|
|
|
@mock.patch.object(hardware, 'md_restart', autospec=True)
|
|
|
|
|
@mock.patch.object(hardware, 'md_get_raid_devices', autospec=True)
|
|
|
|
|
@mock.patch.object(hardware, 'get_holder_disks', autospec=True,
|
|
|
|
|
return_value=['/dev/sda', '/dev/sdb'])
|
|
|
|
|
@mock.patch.object(os, 'environ', autospec=True)
|
|
|
|
|
@mock.patch.object(os, 'makedirs', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_partition', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_prepare_boot_partitions_for_softraid',
|
|
|
|
|
autospec=True,
|
|
|
|
|
return_value=[])
|
|
|
|
|
@mock.patch.object(image, '_has_dracut',
|
|
|
|
|
autospec=True,
|
|
|
|
|
return_value=False)
|
|
|
|
|
def test__install_grub2_softraid_bios(
|
|
|
|
|
self, mock_dracut,
|
|
|
|
|
mock_prepare, mock_get_part_uuid, mkdir_mock, environ_mock,
|
|
|
|
|
mock_holder, mock_md_get_raid_devices, mock_restart,
|
|
|
|
|
mock_is_md_device,
|
|
|
|
|
mock_execute, mock_dispatch):
|
|
|
|
|
|
|
|
|
|
mock_get_part_uuid.side_effect = [self.fake_root_part,
|
|
|
|
|
self.fake_efi_system_part]
|
|
|
|
|
environ_mock.get.return_value = '/sbin'
|
|
|
|
|
mock_is_md_device.return_value = True
|
|
|
|
|
mock_md_get_raid_devices.return_value = {}
|
|
|
|
|
|
|
|
|
|
image._install_grub2(
|
|
|
|
|
self.fake_dev, root_uuid=self.fake_root_uuid,
|
|
|
|
|
efi_system_part_uuid=None,
|
|
|
|
|
target_boot_mode='bios')
|
|
|
|
|
|
|
|
|
|
expected = [
|
|
|
|
|
mock.call('mount', '/dev/fake2', self.fake_dir),
|
|
|
|
|
mock.call('mount', '-o', 'bind', '/dev',
|
|
|
|
|
self.fake_dir + '/dev'),
|
|
|
|
|
mock.call('mount', '-o', 'bind', '/proc',
|
|
|
|
|
self.fake_dir + '/proc'),
|
|
|
|
|
mock.call('mount', '-o', 'bind', '/run',
|
|
|
|
|
self.fake_dir + '/run'),
|
|
|
|
|
mock.call('mount', '-t', 'sysfs', 'none',
|
|
|
|
|
self.fake_dir + '/sys'),
|
|
|
|
|
mock.call(('chroot %s /bin/sh -c '
|
|
|
|
|
'"grub-install %s"' %
|
|
|
|
|
(self.fake_dir, '/dev/sda')), shell=True,
|
|
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
|
|
|
|
mock.call(('chroot %s /bin/sh -c '
|
|
|
|
|
'"grub-install %s"' %
|
|
|
|
|
(self.fake_dir, '/dev/sdb')), shell=True,
|
|
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
|
|
|
|
mock.call(('chroot %s /bin/sh -c '
|
|
|
|
|
'"grub-mkconfig -o '
|
|
|
|
|
'/boot/grub/grub.cfg"' % self.fake_dir),
|
|
|
|
|
shell=True,
|
|
|
|
|
env_variables={
|
|
|
|
|
'PATH': '/sbin:/bin:/usr/sbin:/sbin'}),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/dev',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/proc',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/run',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/sys',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('umount', self.fake_dir, attempts=3,
|
|
|
|
|
delay_on_retry=True)]
|
|
|
|
|
self.assertFalse(mkdir_mock.called)
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
|
|
|
|
mock_get_part_uuid.assert_any_call(self.fake_dev,
|
|
|
|
|
uuid=self.fake_root_uuid)
|
|
|
|
|
self.assertFalse(mock_dispatch.called)
|
|
|
|
|
mock_prepare.assert_called_once_with(self.fake_dev,
|
|
|
|
|
['/dev/sda', '/dev/sdb'],
|
|
|
|
|
None, 'bios')
|
|
|
|
|
mock_restart.assert_called_once_with(self.fake_dev)
|
|
|
|
|
mock_holder.assert_called_once_with(self.fake_dev)
|
|
|
|
|
mock_dracut.assert_called_once_with(self.fake_dir)
|
|
|
|
|
|
2019-08-29 16:22:50 -04:00
|
|
|
@mock.patch.object(image, '_is_bootloader_loaded', autospec=True)
|
2019-02-26 17:56:20 +01:00
|
|
|
@mock.patch.object(hardware, 'is_md_device', autospec=True)
|
2019-08-29 16:22:50 -04:00
|
|
|
def test__get_partition(self, mock_is_md_device, mock_is_bootloader,
|
|
|
|
|
mock_execute, mock_dispatch):
|
|
|
|
|
mock_is_md_device.side_effect = [False]
|
2019-10-17 11:14:04 +00:00
|
|
|
mock_is_md_device.side_effect = [False, False]
|
2015-02-19 13:42:39 +00:00
|
|
|
lsblk_output = ('''KNAME="test" UUID="" TYPE="disk"
|
|
|
|
|
KNAME="test1" UUID="256a39e3-ca3c-4fb8-9cc2-b32eec441f47" TYPE="part"
|
|
|
|
|
KNAME="test2" UUID="%s" TYPE="part"''' % self.fake_root_uuid)
|
2015-03-19 02:40:39 -07:00
|
|
|
mock_execute.side_effect = (None, None, [lsblk_output])
|
2015-02-19 13:42:39 +00:00
|
|
|
|
2015-10-02 10:01:00 -07:00
|
|
|
root_part = image._get_partition(self.fake_dev, self.fake_root_uuid)
|
2015-02-19 13:42:39 +00:00
|
|
|
self.assertEqual('/dev/test2', root_part)
|
|
|
|
|
expected = [mock.call('partx', '-u', self.fake_dev, attempts=3,
|
|
|
|
|
delay_on_retry=True),
|
2015-03-19 02:40:39 -07:00
|
|
|
mock.call('udevadm', 'settle'),
|
2018-05-17 15:23:54 -05:00
|
|
|
mock.call('lsblk', '-PbioKNAME,UUID,PARTUUID,TYPE',
|
|
|
|
|
self.fake_dev)]
|
2015-02-19 13:42:39 +00:00
|
|
|
mock_execute.assert_has_calls(expected)
|
|
|
|
|
self.assertFalse(mock_dispatch.called)
|
2019-08-29 16:22:50 -04:00
|
|
|
self.assertFalse(mock_is_bootloader.called)
|
2015-02-19 13:42:39 +00:00
|
|
|
|
2019-02-26 17:56:20 +01:00
|
|
|
@mock.patch.object(hardware, 'is_md_device', autospec=True)
|
|
|
|
|
def test__get_partition_no_device_found(self, mock_is_md_device,
|
|
|
|
|
mock_execute, mock_dispatch):
|
2019-10-17 11:14:04 +00:00
|
|
|
mock_is_md_device.side_effect = [False, False]
|
2015-02-19 13:42:39 +00:00
|
|
|
lsblk_output = ('''KNAME="test" UUID="" TYPE="disk"
|
|
|
|
|
KNAME="test1" UUID="256a39e3-ca3c-4fb8-9cc2-b32eec441f47" TYPE="part"
|
|
|
|
|
KNAME="test2" UUID="" TYPE="part"''')
|
2019-10-14 14:30:11 -07:00
|
|
|
mock_execute.side_effect = (
|
|
|
|
|
None, None, [lsblk_output],
|
|
|
|
|
processutils.ProcessExecutionError('boom'),
|
|
|
|
|
processutils.ProcessExecutionError('kaboom'))
|
2015-02-19 13:42:39 +00:00
|
|
|
|
|
|
|
|
self.assertRaises(errors.DeviceNotFound,
|
2015-03-12 09:15:49 +00:00
|
|
|
image._get_partition, self.fake_dev,
|
2015-02-19 13:42:39 +00:00
|
|
|
self.fake_root_uuid)
|
|
|
|
|
expected = [mock.call('partx', '-u', self.fake_dev, attempts=3,
|
|
|
|
|
delay_on_retry=True),
|
2015-03-19 02:40:39 -07:00
|
|
|
mock.call('udevadm', 'settle'),
|
2018-05-17 15:23:54 -05:00
|
|
|
mock.call('lsblk', '-PbioKNAME,UUID,PARTUUID,TYPE',
|
|
|
|
|
self.fake_dev)]
|
2015-02-19 13:42:39 +00:00
|
|
|
mock_execute.assert_has_calls(expected)
|
2019-10-14 14:30:11 -07:00
|
|
|
self.assertFalse(mock_dispatch.called)
|
|
|
|
|
|
|
|
|
|
@mock.patch.object(hardware, 'is_md_device', autospec=True)
|
|
|
|
|
def test__get_partition_fallback_partuuid(self, mock_is_md_device,
|
|
|
|
|
mock_execute, mock_dispatch):
|
|
|
|
|
mock_is_md_device.side_effect = [False]
|
|
|
|
|
lsblk_output = ('''KNAME="test" UUID="" TYPE="disk"
|
|
|
|
|
KNAME="test1" UUID="256a39e3-ca3c-4fb8-9cc2-b32eec441f47" TYPE="part"
|
|
|
|
|
KNAME="test2" UUID="" TYPE="part"''')
|
|
|
|
|
findfs_output = ('/dev/loop0\n', None)
|
|
|
|
|
mock_execute.side_effect = (
|
|
|
|
|
None, None, [lsblk_output],
|
|
|
|
|
processutils.ProcessExecutionError('boom'),
|
|
|
|
|
findfs_output)
|
|
|
|
|
|
|
|
|
|
result = image._get_partition(self.fake_dev, self.fake_root_uuid)
|
|
|
|
|
self.assertEqual('/dev/loop0', result)
|
|
|
|
|
expected = [mock.call('partx', '-u', self.fake_dev, attempts=3,
|
|
|
|
|
delay_on_retry=True),
|
|
|
|
|
mock.call('udevadm', 'settle'),
|
|
|
|
|
mock.call('lsblk', '-PbioKNAME,UUID,PARTUUID,TYPE',
|
|
|
|
|
self.fake_dev),
|
|
|
|
|
mock.call('findfs', 'UUID=%s' % self.fake_root_uuid),
|
|
|
|
|
mock.call('findfs', 'PARTUUID=%s' % self.fake_root_uuid)]
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
2015-02-19 13:42:39 +00:00
|
|
|
self.assertFalse(mock_dispatch.called)
|
|
|
|
|
|
2019-02-26 17:56:20 +01:00
|
|
|
@mock.patch.object(hardware, 'is_md_device', autospec=True)
|
|
|
|
|
def test__get_partition_command_fail(self, mock_is_md_device,
|
|
|
|
|
mock_execute, mock_dispatch):
|
2019-10-17 11:14:04 +00:00
|
|
|
mock_is_md_device.side_effect = [False, False]
|
2015-03-19 02:40:39 -07:00
|
|
|
mock_execute.side_effect = (None, None,
|
2015-02-19 13:42:39 +00:00
|
|
|
processutils.ProcessExecutionError('boom'))
|
|
|
|
|
self.assertRaises(errors.CommandExecutionError,
|
2015-03-12 09:15:49 +00:00
|
|
|
image._get_partition, self.fake_dev,
|
2015-02-19 13:42:39 +00:00
|
|
|
self.fake_root_uuid)
|
|
|
|
|
|
|
|
|
|
expected = [mock.call('partx', '-u', self.fake_dev, attempts=3,
|
|
|
|
|
delay_on_retry=True),
|
2015-03-19 02:40:39 -07:00
|
|
|
mock.call('udevadm', 'settle'),
|
2018-05-17 15:23:54 -05:00
|
|
|
mock.call('lsblk', '-PbioKNAME,UUID,PARTUUID,TYPE',
|
|
|
|
|
self.fake_dev)]
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
|
|
|
|
self.assertFalse(mock_dispatch.called)
|
|
|
|
|
|
2019-02-26 17:56:20 +01:00
|
|
|
@mock.patch.object(hardware, 'is_md_device', autospec=True)
|
|
|
|
|
def test__get_partition_partuuid(self, mock_is_md_device, mock_execute,
|
|
|
|
|
mock_dispatch):
|
2019-10-17 11:14:04 +00:00
|
|
|
mock_is_md_device.side_effect = [False, False]
|
2018-05-17 15:23:54 -05:00
|
|
|
lsblk_output = ('''KNAME="test" UUID="" TYPE="disk"
|
|
|
|
|
KNAME="test1" UUID="256a39e3-ca3c-4fb8-9cc2-b32eec441f47" TYPE="part"
|
|
|
|
|
KNAME="test2" PARTUUID="%s" TYPE="part"''' % self.fake_root_uuid)
|
|
|
|
|
mock_execute.side_effect = (None, None, [lsblk_output])
|
|
|
|
|
|
|
|
|
|
root_part = image._get_partition(self.fake_dev, self.fake_root_uuid)
|
|
|
|
|
self.assertEqual('/dev/test2', root_part)
|
|
|
|
|
expected = [mock.call('partx', '-u', self.fake_dev, attempts=3,
|
|
|
|
|
delay_on_retry=True),
|
|
|
|
|
mock.call('udevadm', 'settle'),
|
|
|
|
|
mock.call('lsblk', '-PbioKNAME,UUID,PARTUUID,TYPE',
|
|
|
|
|
self.fake_dev)]
|
2015-02-19 13:42:39 +00:00
|
|
|
mock_execute.assert_has_calls(expected)
|
|
|
|
|
self.assertFalse(mock_dispatch.called)
|
2019-08-29 16:22:50 -04:00
|
|
|
|
|
|
|
|
def test__is_bootloader_loaded(self, mock_execute,
|
|
|
|
|
mock_dispatch):
|
2020-03-23 07:46:48 -07:00
|
|
|
mock_dispatch.return_value = hardware.BootInfo(
|
|
|
|
|
current_boot_mode='bios')
|
2019-08-29 16:22:50 -04:00
|
|
|
parted_output = ('BYT;\n'
|
|
|
|
|
'/dev/loop1:46.1MB:loopback:512:512:gpt:Loopback '
|
|
|
|
|
'device:;\n'
|
|
|
|
|
'15:1049kB:9437kB:8389kB:::boot;\n'
|
|
|
|
|
'1:9437kB:46.1MB:36.7MB:ext3::;\n')
|
|
|
|
|
disk_file_output = ('/dev/loop1: partition 1: ID=0xee, starthead 0, '
|
|
|
|
|
'startsector 1, 90111 sectors, extended '
|
|
|
|
|
'partition table (last)\011, code offset 0x48')
|
|
|
|
|
|
|
|
|
|
part_file_output = ('/dev/loop1p15: x86 boot sector, mkdosfs boot '
|
|
|
|
|
'message display, code offset 0x3c, OEM-ID '
|
|
|
|
|
'"mkfs.fat", sectors/cluster 8, root entries '
|
|
|
|
|
'512, sectors 16384 (volumes <=32 MB) , Media '
|
|
|
|
|
'descriptor 0xf8, sectors/FAT 8, heads 255, '
|
|
|
|
|
'serial number 0x23a08feb, unlabeled, '
|
|
|
|
|
'FAT (12 bit)')
|
|
|
|
|
|
|
|
|
|
# NOTE(TheJulia): File evaluates this data, so it is pointless to
|
|
|
|
|
# try and embed the raw bytes in the test.
|
|
|
|
|
dd_output = ('')
|
|
|
|
|
|
|
|
|
|
file_output = ('/dev/loop1: DOS executable (COM)\n')
|
|
|
|
|
|
|
|
|
|
mock_execute.side_effect = iter([(parted_output, ''),
|
|
|
|
|
(disk_file_output, ''),
|
|
|
|
|
(part_file_output, ''),
|
|
|
|
|
(dd_output, ''),
|
|
|
|
|
(file_output, '')])
|
|
|
|
|
|
|
|
|
|
result = image._is_bootloader_loaded(self.fake_dev)
|
|
|
|
|
self.assertTrue(result)
|
|
|
|
|
|
|
|
|
|
def test__is_bootloader_loaded_not_bootable(self,
|
|
|
|
|
mock_execute,
|
|
|
|
|
mock_dispatch):
|
|
|
|
|
parted_output = ('BYT;\n'
|
|
|
|
|
'/dev/loop1:46.1MB:loopback:512:512:gpt:Loopback '
|
|
|
|
|
'device:;\n'
|
|
|
|
|
'15:1049kB:9437kB:8389kB:::;\n'
|
|
|
|
|
'1:9437kB:46.1MB:36.7MB:ext3::;\n')
|
|
|
|
|
mock_execute.return_value = (parted_output, '')
|
|
|
|
|
result = image._is_bootloader_loaded(self.fake_dev)
|
|
|
|
|
self.assertFalse(result)
|
|
|
|
|
|
|
|
|
|
def test__is_bootloader_loaded_empty(self,
|
|
|
|
|
mock_execute,
|
|
|
|
|
mock_dispatch):
|
|
|
|
|
parted_output = ('BYT;\n'
|
|
|
|
|
'/dev/loop1:46.1MB:loopback:512:512:gpt:Loopback '
|
|
|
|
|
'device:;\n')
|
|
|
|
|
mock_execute.return_value = (parted_output, '')
|
|
|
|
|
result = image._is_bootloader_loaded(self.fake_dev)
|
|
|
|
|
self.assertFalse(result)
|
2019-12-02 16:56:42 +01:00
|
|
|
|
2020-03-23 07:46:48 -07:00
|
|
|
def test__is_bootloader_loaded_uefi_mode(self, mock_execute,
|
|
|
|
|
mock_dispatch):
|
|
|
|
|
|
|
|
|
|
mock_dispatch.return_value = hardware.BootInfo(
|
|
|
|
|
current_boot_mode='uefi')
|
|
|
|
|
result = image._is_bootloader_loaded(self.fake_dev)
|
|
|
|
|
self.assertFalse(result)
|
|
|
|
|
mock_dispatch.assert_any_call('get_boot_info')
|
|
|
|
|
self.assertEqual(0, mock_execute.call_count)
|
|
|
|
|
|
2019-12-02 16:56:42 +01:00
|
|
|
@mock.patch.object(image, '_get_partition', autospec=True)
|
|
|
|
|
@mock.patch.object(utils, 'get_efi_part_on_device', autospec=True)
|
|
|
|
|
def test__manage_uefi_no_partition(self, mock_utils_efi_part,
|
|
|
|
|
mock_get_part_uuid,
|
|
|
|
|
mock_execute, mock_dispatch):
|
|
|
|
|
mock_utils_efi_part.return_value = None
|
|
|
|
|
mock_get_part_uuid.return_value = self.fake_root_part
|
|
|
|
|
result = image._manage_uefi(self.fake_dev, self.fake_root_uuid)
|
|
|
|
|
self.assertFalse(result)
|
|
|
|
|
|
|
|
|
|
@mock.patch.object(os.path, 'exists', lambda *_: False)
|
|
|
|
|
@mock.patch.object(image, '_get_efi_bootloaders', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_partition', autospec=True)
|
|
|
|
|
@mock.patch.object(utils, 'get_efi_part_on_device', autospec=True)
|
|
|
|
|
@mock.patch.object(os, 'makedirs', autospec=True)
|
|
|
|
|
def test__manage_uefi(self, mkdir_mock, mock_utils_efi_part,
|
|
|
|
|
mock_get_part_uuid, mock_efi_bl, mock_execute,
|
|
|
|
|
mock_dispatch):
|
|
|
|
|
mock_utils_efi_part.return_value = '1'
|
|
|
|
|
mock_get_part_uuid.return_value = self.fake_dev
|
|
|
|
|
|
|
|
|
|
mock_efi_bl.return_value = ['\\EFI\\BOOT\\BOOTX64.EFI']
|
|
|
|
|
|
|
|
|
|
mock_execute.side_effect = iter([('', ''), ('', ''),
|
2020-02-08 07:18:12 -08:00
|
|
|
('', ''), ('', ''),
|
2019-12-02 16:56:42 +01:00
|
|
|
('', ''), ('', ''),
|
|
|
|
|
('', '')])
|
|
|
|
|
|
2020-02-08 07:18:12 -08:00
|
|
|
expected = [mock.call('partx', '-u', '/dev/fake', attempts=3,
|
|
|
|
|
delay_on_retry=True),
|
|
|
|
|
mock.call('udevadm', 'settle'),
|
|
|
|
|
mock.call('mount', self.fake_efi_system_part,
|
2019-12-02 16:56:42 +01:00
|
|
|
self.fake_dir + '/boot/efi'),
|
|
|
|
|
mock.call('efibootmgr'),
|
|
|
|
|
mock.call('efibootmgr', '-c', '-d', self.fake_dev,
|
|
|
|
|
'-p', '1', '-w',
|
|
|
|
|
'-L', 'ironic1', '-l',
|
|
|
|
|
'\\EFI\\BOOT\\BOOTX64.EFI'),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/boot/efi',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('sync')]
|
|
|
|
|
|
|
|
|
|
result = image._manage_uefi(self.fake_dev, self.fake_root_uuid)
|
|
|
|
|
self.assertTrue(result)
|
|
|
|
|
mkdir_mock.assert_called_once_with(self.fake_dir + '/boot/efi')
|
|
|
|
|
mock_efi_bl.assert_called_once_with(self.fake_dir + '/boot/efi')
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
2020-02-08 07:18:12 -08:00
|
|
|
self.assertEqual(7, mock_execute.call_count)
|
2019-12-02 16:56:42 +01:00
|
|
|
|
|
|
|
|
@mock.patch.object(os.path, 'exists', lambda *_: False)
|
|
|
|
|
@mock.patch.object(image, '_get_efi_bootloaders', autospec=True)
|
|
|
|
|
@mock.patch.object(image, '_get_partition', autospec=True)
|
|
|
|
|
@mock.patch.object(utils, 'get_efi_part_on_device', autospec=True)
|
|
|
|
|
@mock.patch.object(os, 'makedirs', autospec=True)
|
|
|
|
|
def test__manage_uefi_wholedisk(
|
|
|
|
|
self, mkdir_mock, mock_utils_efi_part,
|
|
|
|
|
mock_get_part_uuid, mock_efi_bl, mock_execute,
|
|
|
|
|
mock_dispatch):
|
|
|
|
|
mock_utils_efi_part.return_value = '1'
|
|
|
|
|
mock_get_part_uuid.side_effect = Exception
|
|
|
|
|
|
|
|
|
|
mock_efi_bl.return_value = ['\\EFI\\BOOT\\BOOTX64.EFI']
|
|
|
|
|
|
|
|
|
|
mock_execute.side_effect = iter([('', ''), ('', ''),
|
2020-02-08 07:18:12 -08:00
|
|
|
('', ''), ('', ''),
|
2019-12-02 16:56:42 +01:00
|
|
|
('', ''), ('', ''),
|
|
|
|
|
('', '')])
|
|
|
|
|
|
2020-02-08 07:18:12 -08:00
|
|
|
expected = [mock.call('partx', '-u', '/dev/fake', attempts=3,
|
|
|
|
|
delay_on_retry=True),
|
|
|
|
|
mock.call('udevadm', 'settle'),
|
|
|
|
|
mock.call('mount', self.fake_efi_system_part,
|
2019-12-02 16:56:42 +01:00
|
|
|
self.fake_dir + '/boot/efi'),
|
|
|
|
|
mock.call('efibootmgr'),
|
|
|
|
|
mock.call('efibootmgr', '-c', '-d', self.fake_dev,
|
|
|
|
|
'-p', '1', '-w',
|
|
|
|
|
'-L', 'ironic1', '-l',
|
|
|
|
|
'\\EFI\\BOOT\\BOOTX64.EFI'),
|
|
|
|
|
mock.call('umount', self.fake_dir + '/boot/efi',
|
|
|
|
|
attempts=3, delay_on_retry=True),
|
|
|
|
|
mock.call('sync')]
|
|
|
|
|
|
|
|
|
|
result = image._manage_uefi(self.fake_dev, None)
|
|
|
|
|
self.assertTrue(result)
|
|
|
|
|
mkdir_mock.assert_called_once_with(self.fake_dir + '/boot/efi')
|
|
|
|
|
mock_efi_bl.assert_called_once_with(self.fake_dir + '/boot/efi')
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
2020-02-08 07:18:12 -08:00
|
|
|
self.assertEqual(7, mock_execute.call_count)
|
2019-12-02 16:56:42 +01:00
|
|
|
|
|
|
|
|
@mock.patch.object(os, 'walk', autospec=True)
|
|
|
|
|
@mock.patch.object(os, 'access', autospec=False)
|
|
|
|
|
def test__no_efi_bootloaders(self, mock_access, mock_walk, mock_execute,
|
|
|
|
|
mock_dispatch):
|
|
|
|
|
# No valid efi file.
|
|
|
|
|
mock_walk.return_value = [
|
|
|
|
|
('/boot/efi', ['EFI'], []),
|
|
|
|
|
('/boot/efi/EFI', ['centos', 'BOOT'], []),
|
|
|
|
|
('/boot/efi/EFI/centos', ['fw', 'fonts'],
|
|
|
|
|
['shimx64-centos.efi', 'BOOT.CSV', 'BOOTX64.CSV',
|
|
|
|
|
'MokManager.efi', 'mmx64.efi', 'shim.efi', 'fwupia32.efi',
|
|
|
|
|
'fwupx64.efi', 'shimx64.efi', 'grubenv', 'grubx64.efi',
|
|
|
|
|
'grub.cfg']),
|
|
|
|
|
('/boot/efi/EFI/centos/fw', [], []),
|
|
|
|
|
('/boot/efi/EFI/centos/fonts', [], ['unicode.pf2']),
|
|
|
|
|
('/boot/efi/EFI/BOOT', [], [])
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
result = image._get_efi_bootloaders("/boot/efi")
|
|
|
|
|
self.assertEqual(result, [])
|
|
|
|
|
mock_access.assert_not_called()
|
|
|
|
|
|
|
|
|
|
@mock.patch.object(os, 'walk', autospec=True)
|
|
|
|
|
@mock.patch.object(os, 'access', autospec=True)
|
|
|
|
|
def test__get_efi_bootloaders(self, mock_access, mock_walk, mock_execute,
|
|
|
|
|
mock_dispatch):
|
|
|
|
|
mock_walk.return_value = [
|
|
|
|
|
('/boot/efi', ['EFI'], []),
|
|
|
|
|
('/boot/efi/EFI', ['centos', 'BOOT'], []),
|
|
|
|
|
('/boot/efi/EFI/centos', ['fw', 'fonts'],
|
|
|
|
|
['shimx64-centos.efi', 'BOOT.CSV', 'BOOTX64.CSV',
|
|
|
|
|
'MokManager.efi', 'mmx64.efi', 'shim.efi', 'fwupia32.efi',
|
|
|
|
|
'fwupx64.efi', 'shimx64.efi', 'grubenv', 'grubx64.efi',
|
|
|
|
|
'grub.cfg']),
|
|
|
|
|
('/boot/efi/EFI/centos/fw', [], []),
|
|
|
|
|
('/boot/efi/EFI/centos/fonts', [], ['unicode.pf2']),
|
|
|
|
|
('/boot/efi/EFI/BOOT', [],
|
|
|
|
|
['BOOTX64.EFI', 'fallback.efi', 'fbx64.efi'])
|
|
|
|
|
]
|
|
|
|
|
mock_access.return_value = True
|
|
|
|
|
result = image._get_efi_bootloaders("/boot/efi")
|
|
|
|
|
self.assertEqual(result[0], '\\EFI\\BOOT\\BOOTX64.EFI')
|
|
|
|
|
|
|
|
|
|
@mock.patch.object(os, 'walk', autospec=True)
|
|
|
|
|
@mock.patch.object(os, 'access', autospec=True)
|
|
|
|
|
def test__get_windows_efi_bootloaders(self, mock_access, mock_walk,
|
|
|
|
|
mock_execute, mock_dispatch):
|
|
|
|
|
mock_walk.return_value = [
|
|
|
|
|
('/boot/efi', ['WINDOWS'], []),
|
|
|
|
|
('/boot/efi/WINDOWS', ['system32'], []),
|
|
|
|
|
('/boot/efi/WINDOWS/system32', [],
|
|
|
|
|
['winload.efi'])
|
|
|
|
|
]
|
|
|
|
|
mock_access.return_value = True
|
|
|
|
|
result = image._get_efi_bootloaders("/boot/efi")
|
|
|
|
|
self.assertEqual(result[0], '\\WINDOWS\\system32\\winload.efi')
|
|
|
|
|
|
|
|
|
|
def test__run_efibootmgr_no_bootloaders(self, mock_execute, mock_dispatch):
|
|
|
|
|
result = image._run_efibootmgr([], self.fake_dev,
|
|
|
|
|
self.fake_efi_system_part)
|
|
|
|
|
expected = []
|
|
|
|
|
self.assertIsNone(result)
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
|
|
|
|
|
|
|
|
|
def test__run_efibootmgr(self, mock_execute, mock_dispatch):
|
|
|
|
|
result = image._run_efibootmgr(['\\EFI\\BOOT\\BOOTX64.EFI'],
|
|
|
|
|
self.fake_dev,
|
|
|
|
|
self.fake_efi_system_part)
|
|
|
|
|
expected = [mock.call('efibootmgr'),
|
|
|
|
|
mock.call('efibootmgr', '-c', '-d', self.fake_dev,
|
|
|
|
|
'-p', self.fake_efi_system_part, '-w',
|
|
|
|
|
'-L', 'ironic1', '-l',
|
|
|
|
|
'\\EFI\\BOOT\\BOOTX64.EFI')]
|
|
|
|
|
self.assertIsNone(result)
|
|
|
|
|
mock_execute.assert_has_calls(expected)
|
