Add new banchmarks (#52)

* Added new speed banchmarks

* Fixes after pre-commit

* Separate dataclasses and pydantic banchmarks

* Fixes after pre-commit

* 📝 Add docstrings to `add-new-banchmarks` (#53)

Docstrings generation was requested by @vadikko2.

* #52 (comment)

The following files were modified:

* `benchmarks/dataclasses/test_benchmark_cor_request_handler.py`
* `benchmarks/dataclasses/test_benchmark_event_handling.py`
* `benchmarks/dataclasses/test_benchmark_request_handling.py`
* `benchmarks/dataclasses/test_benchmark_saga_fallback.py`
* `benchmarks/dataclasses/test_benchmark_saga_memory.py`
* `benchmarks/dataclasses/test_benchmark_saga_sqlalchemy.py`
* `benchmarks/dataclasses/test_benchmark_stream_request_handler.py`
* `benchmarks/pydantic/test_benchmark_cor_request_handler.py`
* `benchmarks/pydantic/test_benchmark_event_handling.py`
* `benchmarks/pydantic/test_benchmark_request_handling.py`
* `benchmarks/pydantic/test_benchmark_saga_fallback.py`
* `benchmarks/pydantic/test_benchmark_saga_memory.py`
* `benchmarks/pydantic/test_benchmark_saga_sqlalchemy.py`
* `benchmarks/pydantic/test_benchmark_serialization.py`
* `benchmarks/pydantic/test_benchmark_stream_request_handler.py`

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>

* Update ci files

* move banchmark tests

* Try to fix banchmarks

* Try to fix banchmarks

* Try to fix banchmarks

---------

Co-authored-by: Вадим Козыревский <v.kozyrevskiy@timeweb.ru>
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>

Author: 3 people

.github/workflows/codspeed.yml

@@ -29,8 +29,42 @@ jobs:
           python -m pip install --upgrade pip
           pip install -e ".[dev]"
 
+      - name: Start infrastructure
+        run: |
+          docker compose -f docker-compose-test.yml up -d
+
+      - name: Wait for MySQL
+        run: |
+          for i in $(seq 1 30); do
+            if docker compose -f docker-compose-test.yml exec -T mysql_tests mysqladmin ping -h localhost -ucqrs -pcqrs --silent 2>/dev/null; then
+              echo "MySQL is ready"
+              exit 0
+            fi
+            echo "Waiting for MySQL... ($i/30)"
+            sleep 2
+          done
+          echo "MySQL did not become ready in time"
+          exit 1
+
+      - name: Wait for Redis
+        run: |
+          for i in $(seq 1 15); do
+            if docker compose -f docker-compose-test.yml exec -T redis_tests redis-cli ping 2>/dev/null | grep -q PONG; then
+              echo "Redis is ready"
+              exit 0
+            fi
+            echo "Waiting for Redis... ($i/15)"
+            sleep 1
+          done
+          echo "Redis did not become ready in time"
+          exit 1
+
       - name: Run benchmarks
         uses: CodSpeedHQ/action@v4
         with:
           mode: simulation
-          run: pytest benchmarks/ --codspeed
+          run: pytest -c ./tests/pytest-config.ini tests/benchmarks/ --codspeed
+
+      - name: Stop infrastructure
+        if: always()
+        run: docker compose -f docker-compose-test.yml down -v

.github/workflows/tests.yml

@@ -120,7 +120,7 @@ jobs:
 
     - name: Run all tests with coverage
       run: |
-        pytest -c ./tests/pytest-config.ini --cov=src --cov-report=xml --cov-report=term -o cache_dir=/tmp/pytest_cache ./tests
+        pytest -c ./tests/pytest-config.ini --cov=src --cov-report=xml --cov-report=term -o cache_dir=/tmp/pytest_cache ./tests/unit ./tests/integration
 
     - name: Upload coverage to Codecov
       uses: codecov/codecov-action@v4

.pre-commit-config.yaml

@@ -2,7 +2,7 @@ repos:
 - hooks:
   - id: check-docstring-first
   - id: check-ast
-  - exclude: (^tests/mock/|^tests/integration/|^tests/fixtures)
+  - exclude: (^tests/mock/|^tests/integration/|^tests/fixtures|benchmarks)
     id: trailing-whitespace
   - id: end-of-file-fixer
   - id: check-yaml

pyrightconfig.json

@@ -1,7 +1,7 @@
 {
   "venvPath": ".",
   "venv": "venv",
-  "include": ["src", "tests"],
+  "include": ["src", "tests", "benchmarks"],
   "exclude": ["**/__pycache__", "venv", "**/*_pb2.py"],
   "reportMissingImports": "error",
   "reportMissingTypeStubs": false,