def test_show_with_invalid_id(self):
self.mock_object(
db, 'access_info_get',
mock.Mock(side_effect=exception.AccessInfoNotFound('fake_id')))
req = fakes.HTTPRequest.blank('/storages/fake_id/access-info')
self.assertRaises(exception.AccessInfoNotFound, self.controller.show,
req, 'fake_id')
def _access_info_get(context, storage_id, session=None):
result = (_access_info_get_query(
context, session=session).filter_by(storage_id=storage_id).first())
if not result:
raise exception.AccessInfoNotFound(storage_id)
return result
def test_clear_alerts(self, mock_clear_alert):
task = alerts.AlertSyncTask()
storage_id = fake_storage['id']
task.clear_alerts(context, storage_id, [])
self.assertEqual(mock_clear_alert.call_count, 0)
sequence_number_list = ['sequence_number_1', 'sequence_number_2']
task.clear_alerts(context, storage_id, sequence_number_list)
self.assertEqual(mock_clear_alert.call_count,
len(sequence_number_list))
mock_clear_alert.side_effect = \
exception.AccessInfoNotFound(storage_id)
ret = task.clear_alerts(context, storage_id, sequence_number_list)
self.assertEqual(ret, [])
mock_clear_alert.side_effect = \
exception.Invalid('Fake exception')
ret = task.clear_alerts(context, storage_id, sequence_number_list)
self.assertEqual(ret, sequence_number_list)