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') }