修复 TaskWrapper.throw 方法以支持取消任务时传递异常

[tingQian]

No description provided.

[tingQian]

修复 TaskWrapper.throw 方法以支持取消任务时传递异常

[mosquito]

Could you please add some tests reproduces this problem?

[tingQian]

@tingQian ➜ /workspaces/aiormq (feature/abc-docs-update) $ poetry run pytest tests/test_taskwrapper_throw.py -s
============================================================================================== test session starts ===============================================================================================
platform linux -- Python 3.12.1, pytest-9.0.1, pluggy-1.6.0
rootdir: /workspaces/aiormq
configfile: pyproject.toml
plugins: aiomisc-pytest-1.3.4, asyncio-1.3.0
asyncio: mode=Mode.STRICT, debug=False, asyncio_default_fixture_loop_scope=None, asyncio_default_test_loop_scope=function
collected 3 items

tests/test_taskwrapper_throw.py
test_throw_with_instance 1 <ExceptionInfo CancelledError(RuntimeError('boom')) tblen=2>

test_throw_with_instance 2 <ExceptionInfo RuntimeError('boom') tblen=2>
.
test_throw_with_type <ExceptionInfo CancelledError(<class 'RuntimeError'>) tblen=2>
.
test_throw_with_cancellederror 1 <ExceptionInfo CancelledError(CancelledError()) tblen=2>

test_throw_with_cancellederror 2 <ExceptionInfo CancelledError() tblen=2>
.

[tingQian]

这是问题分析
TaskWrapper.throw 方法（当前实现）

aiormq/abc.py:TaskWrapper.throw()

def throw(self, exception: ExceptionType) -> None:
self._exception = exception # 保存原始异常
self.task.cancel() # ❌ 问题：没有传递异常信息！
TaskWrapper 的异常替换机制

aiormq/abc.py:TaskWrapper.__inner()

async def __inner(self) -> Any:
try:
return await self.task
except asyncio.CancelledError as e:
raise self._exception from e # 只有在 await task_wrapper 时才会替换