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')