def test_returns_none_tuple_if_no_partitions(self): self.m_blkpart.side_effect = iter([ ('/dev/disk-a', None), ('/dev/disk-b', None), ('/dev/disk-c', None), ]) devices = ['/dev/disk-a', '/dev/disk-b', '/dev/disc-c'] self.assertEqual((None, None), install_grub.get_efi_disk_part(devices))
def test_returns_first_result_with_partition(self): self.m_blkpart.side_effect = iter([ ('/dev/disk-a', None), ('/dev/disk-b', '1'), ('/dev/disk-c', None), ]) devices = ['/dev/disk-a', '/dev/disk-b', '/dev/disc-c'] self.assertEqual(('/dev/disk-b', '1'), install_grub.get_efi_disk_part(devices))