def test_prepare_for_quota_reservation_task_revert(self):
mock_db = mock.MagicMock()
mock_driver = mock.MagicMock()
mock_result = mock.MagicMock()
mock_flow_failures = mock.MagicMock()
mock_error_out = self.mock_object(flow_common, 'error_out')
volume_ref = self._stub_volume_object_get(self)
task = manager.PrepareForQuotaReservationTask(mock_db, mock_driver)
task.revert(self.ctxt, mock_result, mock_flow_failures, volume_ref)
mock_error_out.assert_called_once_with(volume_ref, reason=mock.ANY)
def test_prepare_for_quota_reservation_with_wrong_volume(self):
"""Test the class PrepareForQuotaReservationTas with wrong vol."""
mock_db = mock.MagicMock()
mock_driver = mock.MagicMock()
wrong_volume = mock.MagicMock()
mock_manage_existing_ref = mock.MagicMock()
mock_except = exception.CinderException
mock_driver.manage_existing_get_size.side_effect = mock_except
task = manager.PrepareForQuotaReservationTask(mock_db, mock_driver)
self.assertRaises(exception.CinderException, task.execute, self.ctxt,
wrong_volume, mock_manage_existing_ref)
def test_prepare_for_quota_reserveration_task_execute(self):
mock_db = mock.MagicMock()
mock_driver = mock.MagicMock()
mock_manage_existing_ref = mock.MagicMock()
mock_get_size = self.mock_object(mock_driver,
'manage_existing_get_size')
mock_get_size.return_value = '5'
volume_ref = self._stub_volume_object_get(self)
task = manager.PrepareForQuotaReservationTask(mock_db, mock_driver)
result = task.execute(self.ctxt, volume_ref, mock_manage_existing_ref)
self.assertEqual(volume_ref, result['volume_properties'])
self.assertEqual('5', result['size'])
self.assertEqual(volume_ref.id, result['volume_spec']['volume_id'])
mock_get_size.assert_called_once_with(volume_ref,
mock_manage_existing_ref)
