fix: check semaphore values

Author: Anton

taskiq/receiver/receiver.py

@@ -307,6 +307,8 @@ def task_cb(task: "asyncio.Task[Any]") -> None:
             self.sem_prefetch.release()
             message = await queue.get()
             if message is QUEUE_DONE:
+                if self.sem is not None:
+                    self.sem.release()
                 break
             if message is QUEUE_SKIP:
                 # Decrease max_prefetch

tests/cli/worker/test_receiver.py

@@ -349,6 +349,9 @@ async def task_map(vals: List[int], ctx: Context = Depends()) -> List[int]:
     await broker.shutdown()
     await listen_task
 
+    assert receiver.sem_idle._value == 1  # type: ignore
+    assert receiver.sem._value == 1  # type: ignore
+
 
 @pytest.mark.anyio
 async def test_tasks_chain_deep() -> None:
@@ -390,6 +393,9 @@ async def wait_for_task(
     await broker.shutdown()
     await listen_task
 
+    assert receiver.sem_idle._value == 10  # type: ignore
+    assert receiver.sem._value == 1  # type: ignore
+
 
 @pytest.mark.anyio
 async def test_tasks_sleep() -> None:
@@ -416,3 +422,6 @@ async def task_run(ind: int, ctx: Context = Depends()) -> int:
 
     await broker.shutdown()
     await listen_task
+
+    assert receiver.sem_idle._value == 20  # type: ignore
+    assert receiver.sem._value == 1  # type: ignore