Added gather function.

.flake8

@@ -95,6 +95,8 @@ per-file-ignores =
   test_*.py,tests.py,tests_*.py,*/tests/*:
   ; Use of assert detected
   S101,
+  ; Found magic number
+  WPS432,
 
 exclude =
   ./.git,

.github/workflows/test.yml

@@ -46,6 +46,10 @@ jobs:
       - name: Run mypy check
         run: poetry run mypy .
   pytest:
+    permissions:
+      checks: write
+      pull-requests: write
+      contents: write
     strategy:
       matrix:
         py_version: ["3.7", "3.8", "3.9", "3.10"]
@@ -66,4 +70,9 @@ jobs:
         env:
           POETRY_VIRTUALENVS_CREATE: false
       - name: Run pytest check
-        run: poetry run pytest -vv --cov="taskiq" .
+        run: poetry run pytest -vv -n auto --cov="taskiq" .
+      - name: Generate report
+        run: coverage xml
+      - name: Upload coverage reports to Codecov with GitHub Action
+        uses: codecov/codecov-action@v3
+        if: matrix.os == 'ubuntu-latest' && matrix.py_version == '3.9'

.gitignore

@@ -5,6 +5,7 @@ __pycache__/
 
 # C extensions
 *.so
+coverage.*
 
 # Distribution / packaging
 .Python

pyproject.toml

@@ -45,6 +45,10 @@ autoflake = "^1.4"
 wemake-python-styleguide = "^0.16.1"
 coverage = "^6.4.2"
 pytest-cov = "^3.0.0"
+mock = "^4.0.3"
+anyio = "^3.6.1"
+pytest-xdist = {version = "^2.5.0", extras = ["psutil"]}
+types-mock = "^4.0.15"
 
 [tool.poetry.extras]
 zmq = ["pyzmq"]

taskiq/__init__.py

@@ -7,11 +7,13 @@
 from taskiq.brokers.shared_broker import async_shared_broker
 from taskiq.brokers.zmq_broker import ZeroMQBroker
 from taskiq.exceptions import TaskiqError
+from taskiq.funcs import gather
 from taskiq.message import BrokerMessage, TaskiqMessage
 from taskiq.result import TaskiqResult
 from taskiq.task import AsyncTaskiqTask
 
 __all__ = [
+    "gather",
     "AsyncBroker",
     "TaskiqError",
     "TaskiqResult",