def test_handle_message_fail(self): res = AsyncResult('testid') body = '{"task_name": "test", "exception": {"value": "I123\\n."}}' msg = MagicMock(body=body, correlation_id='testid') AsyncResult._handle_message(msg, res) self.assertEqual(123, res._exc) self.assertFalse(hasattr(res, '_return'))
def test_handle_message(self): res = AsyncResult('testid') msg = MagicMock(body='{"task_name": "test", "return": 123}', correlation_id='testid') AsyncResult._handle_message(msg, res) self.assertEqual(123, res._return) self.assertFalse(hasattr(res, '_exc'))