def test_lookup_invalid_part_returns_404(self, mock_get_part_details):
mock_get_part_details.side_effect = iter([NotFoundError])
part = diskparts.PartitionModel(objstore=self._objstore)
with self.assertRaises(NotFoundError):
part.lookup('/a/invalid/partition')
def test_change_part_type(self, mock_change_type):
part = diskparts.PartitionModel(objstore=self._objstore)
part_name = 'sdb1'
type = '82'
mock_change_type.return_value = 'sdb1'
part.change_type(part_name, type)
mock_change_type.assert_called_with(part_name, type)
def test_format_part(self, mock_is_mntd, mock_makefs):
mock_is_mntd.return_value = False
part = diskparts.PartitionModel(objstore=self._objstore)
name = 'a_partition_name'
fstype = 'ext4'
task_obj = part.format(name, fstype)
self.task_model.wait(task_obj.get('id'))
mock_makefs.assert_called_with(fstype, name)
def test_delete_part(self, mock_delete_part):
part = diskparts.PartitionModel(objstore=self._objstore)
part_name = 'sdb1'
part.delete(part_name)
mock_delete_part.assert_called_with(part_name)