def test_callbacks__EncodeError(self, maybe_signature):
sig = Mock(name='sig')
request = {'callbacks': [sig], 'root_id': 'root'}
maybe_signature.return_value = sig
sig.apply_async.side_effect = EncodeError()
retval, einfo = self.trace(self.add, (2, 2), {}, request=request)
self.assertEqual(einfo.state, states.FAILURE)
def _store_result(self,
task_id,
result,
state,
traceback=None,
request=None,
**kwargs):
"""Store return value and state of an executed task."""
meta = {
#'_id': task_id,
#'status': state,
'result': result, #'result': self.encode(result),
#'date_done': datetime.utcnow(),
#'traceback': self.encode(traceback),
#'children': self.encode(
# self.current_task_children(request),
#),
}
if result != {}:
try:
self.collection.save(meta)
except InvalidDocument as exc:
raise EncodeError(exc)
return result
def _store_result(self,
task_id,
result,
state,
traceback=None,
request=None,
**kwargs):
"""Store return value and state of an executed task."""
meta = {
'_id': task_id,
'status': state,
'result': self.encode(result),
'date_done': datetime.utcnow(),
'traceback': self.encode(traceback),
'children': self.encode(self.current_task_children(request), ),
}
if request and getattr(request, 'parent_id', None):
meta['parent_id'] = request.parent_id
try:
self.collection.save(meta)
except InvalidDocument as exc:
raise EncodeError(exc)
return result
def _store_result(self,
task_id,
result,
state,
traceback=None,
request=None,
**kwargs):
"""Store return value and state of an executed task."""
meta = self._get_result_meta(result=result,
state=state,
traceback=traceback,
request=request)
# Add the _id for mongodb
meta['_id'] = task_id
try:
self.collection.replace_one({'_id': task_id}, meta, upsert=True)
except InvalidDocument as exc:
raise EncodeError(exc)
return result
