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)