fix: requeue message (#154)

Author: vvanglro

taskiq/context.py

@@ -1,4 +1,3 @@
-from copy import copy
 from typing import TYPE_CHECKING
 
 from taskiq.abc.broker import AsyncBroker
@@ -28,10 +27,9 @@ async def requeue(self) -> None:
 
         :raises NoResultError: to not store result for current task.
         """
-        message = copy(self.message)
-        requeue_count = int(message.labels.get("X-Taskiq-requeue", 0))
+        requeue_count = int(self.message.labels.get("X-Taskiq-requeue", 0))
         requeue_count += 1
-        message.labels["X-Taskiq-requeue"] = str(requeue_count)
+        self.message.labels["X-Taskiq-requeue"] = str(requeue_count)
         await self.broker.kick(self.broker.formatter.dumps(self.message))
         raise NoResultError()
 

tests/test_requeue.py

@@ -18,6 +18,10 @@ async def task(context: Context = TaskiqDepends()) -> None:
 
     kicked = await task.kiq()
     await kicked.wait_result()
+    assert (
+        broker.custom_dependency_context[Context].message.labels["X-Taskiq-requeue"]
+        == "1"
+    )
 
     assert runs_count == 2
 
@@ -40,5 +44,9 @@ async def task(_: None = TaskiqDepends(dep_func)) -> None:
 
     kicked = await task.kiq()
     await kicked.wait_result()
+    assert (
+        broker.custom_dependency_context[Context].message.labels["X-Taskiq-requeue"]
+        == "1"
+    )
 
     assert runs_count == 2