def test_predicted_drive_type(self):
"""Default prediction is 'harddisk'."""
self.assertEqual(
'harddisk',
DiskRepresentation(path=self.blank_vmdk).predicted_drive_type)
def test_capacity_qemu_error(self, mock_check_output):
"""Test error handline if qemu-img reports an error."""
mock_check_output.return_value = "qemu-img info: unsupported command"
with self.assertRaises(RuntimeError):
assert DiskRepresentation(path=self.blank_vmdk).capacity
def test_files(self):
"""No default files getter logic."""
with self.assertRaises(NotImplementedError):
assert DiskRepresentation(path=self.blank_vmdk).files
