def test_sfdisk_info(self):
"""verify sfdisk_info returns correct info dictionary for device."""
self.assertEqual(self.expected, block.sfdisk_info(self.device))
self.assertEqual([mock.call(self.device)],
self.m_get_blockdev_for_partition.call_args_list)
self.assertEqual(
[mock.call(['sfdisk', '--json', self.disk], capture=True)],
self.m_subp.call_args_list)
self.assertEqual([mock.call(self.m_subp.return_value[0])],
self.m_load_json.call_args_list)
def test_sfdisk_info_returns_empty_on_subp_error(self):
"""verify sfdisk_info returns empty dict on subp errors."""
self.m_subp.side_effect = (util.ProcessExecutionError(
stdout="",
stderr="sfdisk: cannot open /dev/vdb: Permission denied",
exit_code=1))
self.assertEqual({}, block.sfdisk_info(self.device))
self.assertEqual([mock.call(self.device)],
self.m_get_blockdev_for_partition.call_args_list)
self.assertEqual(
[mock.call(['sfdisk', '--json', self.disk], capture=True)],
self.m_subp.call_args_list)
self.assertEqual([], self.m_load_json.call_args_list)