def testNoKernelConfig(self):
"""Test the normal path"""
with self.assertRaises(imagefile.SignImageError) as e:
imagefile.UpdateLegacyBootloader(self.image, '/dev/loop9999p4')
# Empty kernel cmdline raises this error.
self.assertEqual('Could not find root digest', str(e.exception))
self.rc.assertCommandCalled(
['sudo', '--', 'dump_kernel_config', '/dev/loop9999p4'],
capture_output=True,
print_cmd=False,
check=True,
encoding='utf-8')
def testNoGrubCfg(self):
"""Test with no efi/boot/grub.cfg."""
uefi = self._CreateCfgFiles(True, False)
imagefile.UpdateLegacyBootloader(self.image, '/dev/loop9999p2')
self.assertEqual(2, self.rc.call_count)
self.rc.assertCommandCalled(
['sudo', '--', 'dump_kernel_config', '/dev/loop9999p2'],
capture_output=True,
print_cmd=False,
check=True,
encoding='utf-8')
sed_command = self.rc.call_args_list[1]
sed_args = (sed_command[0][0][:5] + sorted(sed_command[0][0][5:]), )
sed_args += sed_command[0][1:]
self.assertEqual(([
'sudo', '--', 'sed', '-iE', 's/\\broot_hexdigest=[a-z0-9]+/root_h'
'exdigest=9999999999999999999999999999999999999999/g'
] + sorted(uefi['sys_cfgs']), ), sed_args)
self.assertEqual({'check': False}, sed_command[1])
def testNoGrubCfgSedFails(self):
"""Test the normal path"""
uefi = self._CreateCfgFiles(True, False)
self.rc.AddCmdResult(partial_mock.In('sed'), returncode=1)
with self.assertRaises(imagefile.SignImageError):
imagefile.UpdateLegacyBootloader(self.image, '/dev/loop9999p2')
self.assertEqual(2, self.rc.call_count)
self.rc.assertCommandCalled(
['sudo', '--', 'dump_kernel_config', '/dev/loop9999p2'],
capture_output=True,
print_cmd=False,
check=True,
encoding='utf-8')
sed_command = self.rc.call_args_list[1]
sed_args = (sed_command[0][0][:5] + sorted(sed_command[0][0][5:]), )
sed_args += sed_command[0][1:]
self.assertEqual(([
'sudo', '--', 'sed', '-iE', 's/\\broot_hexdigest=[a-z0-9]+/root_h'
'exdigest=9999999999999999999999999999999999999999/g'
] + sorted(uefi['sys_cfgs']), ), sed_args)
self.assertEqual({'check': False}, sed_command[1])