def test_clear_index_delete_error(self, mock_range, mock_delete,
mock_logging):
mock_range.return_value = [
search.ScoredDocument(doc_id='unused_doc_id')
]
mock_delete.side_effect = search.DeleteError(message='Delete Fail!',
results=[])
base_model.BaseModel.clear_index()
self.assertEqual(mock_logging.exception.call_count, 1)
def _get_delete_error(self, num_res, transient=None):
"""returns a DeleteError. with num_res results.
If transient is given, it should be an index in the
results array. The result at that index will have a transient
error code."""
non_trans_code = search.OperationResult.INVALID_REQUEST
trans_code = search.OperationResult.TRANSIENT_ERROR
results = [
search.DeleteResult(code=non_trans_code) for _ in range(num_res)]
if transient is not None:
results[transient] = search.PutResult(code=trans_code)
return search.DeleteError('lol', results=results)
def test_remove_doc_by_id_delete_error(self, mock_delete, mock_logging):
mock_delete.side_effect = search.DeleteError(message='Delete Fail!',
results=[])
base_model.BaseModel.remove_doc_by_id('test_id')
mock_logging.error.assert_called_once_with(
base_model._REMOVE_DOC_ERR_MSG, 'test_id')
