def testExistingLinkNotCleanedUp(self):
parts = {
42: cros_build_lib.PartitionInfo(42, 0, 0, 0, '', 'label', ''),
}
symlink = os.path.join(self.tempdir, 'dir-label')
os.symlink('/tmp', symlink)
self.assertEqual('/tmp', os.readlink(symlink))
self._testWithParts(parts, [42], check_links=False)
self.assertEqual('/tmp', os.readlink(symlink))
def setUp(self):
# create dummy image file
self.image_file = os.path.join(self.tempdir,
constants.BASE_IMAGE_NAME + '.bin')
osutils.WriteFile(self.image_file, '')
fake_partitions = {
1: cros_build_lib.PartitionInfo(1, 0, 0, 0, 'fs', 'STATE', 'flag'),
2: cros_build_lib.PartitionInfo(2, 0, 0, 0, 'fs', 'KERN-A',
'flag'),
3: cros_build_lib.PartitionInfo(3, 0, 0, 0, 'fs', 'ROOT-A',
'flag'),
}
self.PatchObject(cros_build_lib,
'GetImageDiskPartitionInfo',
autospec=True,
return_value=fake_partitions)
self.PatchObject(osutils.MountImageContext, '_Mount', autospec=True)
self.PatchObject(osutils.MountImageContext, '_Unmount', autospec=True)
def testFailOnExistingMount(self):
parts = {
42: cros_build_lib.PartitionInfo(42, 0, 0, 0, '', 'label', ''),
}
os.makedirs(os.path.join(self.tempdir, 'dir-42'))
self.assertRaises(ValueError, self._testWithParts, parts, [42])
def testInvalidPartSelector(self):
parts = {
42: cros_build_lib.PartitionInfo(42, 0, 0, 0, '', 'label', ''),
}
self.assertRaises(ValueError, self._testWithParts, parts, ['label404'])
self.assertRaises(ValueError, self._testWithParts, parts, [404])
def testWithPartitionLabel(self):
parts = {
42: cros_build_lib.PartitionInfo(42, 0, 0, 0, '', 'label', ''),
}
self._testWithParts(parts, ['label'])
def testWithPartitionNumber(self):
parts = {
1: cros_build_lib.PartitionInfo(1, 0, 0, 0, '', 'my-stateful', ''),
3: cros_build_lib.PartitionInfo(3, 0, 0, 0, '', 'my-root-a', ''),
}
self._testWithParts(parts, [1, 3])
def setUp(self):
self._gpt_table = {
3: cros_build_lib.PartitionInfo(3, 1, 3, 2, 'fs', 'Label', 'flag')
}
