def _get_put_error(self, num_res, transient=None):
"""returns a PutError. 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.PutResult(code=non_trans_code) for _ in range(num_res)]
if transient is not None:
results[transient] = search.PutResult(code=trans_code)
return search.PutError('lol', results)
def test_add_docs_to_index_put_error(self, mock_put):
mock_put.side_effect = [
search.PutError(
message='Fail!',
results=[
search.PutResult(
code=search.OperationResult.TRANSIENT_ERROR)
]), None
]
base_model.BaseModel.add_docs_to_index([
search.Document(
doc_id='test_id',
fields=[search.TextField(name='field_one', value='value_one')])
])
self.assertEqual(mock_put.call_count, 2)
