Fix after pre-commit

Author: Вадим Козыревский

.pre-commit-config.yaml

@@ -9,7 +9,7 @@ repos:
   - id: check-added-large-files
   - args:
     - --pytest-test-first
-    exclude: (^tests/mock/|^tests/integration/|^tests/fixtures)
+    exclude: (^tests/mock/|^tests/integration/|^tests/fixtures|conftest\.py$)
     id: name-tests-test
   - id: check-merge-conflict
   - id: check-json

tests/benchmarks/__init__.py

@@ -1 +1 @@
-# Benchmark package; shared helpers in storage_legacy.py
+# Benchmark package; shared helpers in conftest.py

tests/benchmarks/conftest.py

@@ -1,12 +1,37 @@
-"""Shared fixtures for benchmarks. Engine and loop are session-scoped for saga SQLAlchemy benchmarks."""
+"""Shared fixtures and legacy storage classes for benchmarks."""
+
+from __future__ import annotations
 
 import asyncio
+import contextlib
 import os
 
 import pytest
 from sqlalchemy.ext.asyncio import create_async_engine
 
-from cqrs.saga.storage.sqlalchemy import Base
+from cqrs.saga.storage.memory import MemorySagaStorage
+from cqrs.saga.storage.protocol import SagaStorageRun
+from cqrs.saga.storage.sqlalchemy import Base, SqlAlchemySagaStorage
+
+
+class MemorySagaStorageLegacy(MemorySagaStorage):
+    """Memory storage without create_run: forces legacy path (commit per call)."""
+
+    def create_run(
+        self,
+    ) -> contextlib.AbstractAsyncContextManager[SagaStorageRun]:
+        """Raise NotImplementedError so benchmarks use the legacy commit-per-call path."""
+        raise NotImplementedError("Legacy storage: create_run disabled for benchmark")
+
+
+class SqlAlchemySagaStorageLegacy(SqlAlchemySagaStorage):
+    """SQLAlchemy storage without create_run: forces legacy path (commit per call)."""
+
+    def create_run(
+        self,
+    ) -> contextlib.AbstractAsyncContextManager[SagaStorageRun]:
+        """Raise NotImplementedError so benchmarks use the legacy commit-per-call path."""
+        raise NotImplementedError("Legacy storage: create_run disabled for benchmark")
 
 
 @pytest.fixture(scope="session")

tests/benchmarks/dataclasses/test_benchmark_saga_memory.py

@@ -16,7 +16,7 @@
 from cqrs.saga.step import SagaStepHandler, SagaStepResult
 from cqrs.saga.storage.memory import MemorySagaStorage
 
-from ..storage_legacy import MemorySagaStorageLegacy
+from ..conftest import MemorySagaStorageLegacy
 
 
 @dataclasses.dataclass

tests/benchmarks/dataclasses/test_benchmark_saga_sqlalchemy.py

@@ -10,7 +10,7 @@
 from cqrs.saga.saga import Saga
 from cqrs.saga.storage.sqlalchemy import SqlAlchemySagaStorage
 
-from ..storage_legacy import SqlAlchemySagaStorageLegacy
+from ..conftest import SqlAlchemySagaStorageLegacy
 from .test_benchmark_saga_memory import (
     OrderContext,
     ProcessPaymentStep,

tests/benchmarks/default/test_benchmark_saga_memory.py

@@ -16,7 +16,7 @@
 from cqrs.saga.step import SagaStepHandler, SagaStepResult
 from cqrs.saga.storage.memory import MemorySagaStorage
 
-from ..storage_legacy import MemorySagaStorageLegacy
+from ..conftest import MemorySagaStorageLegacy
 
 
 @dataclasses.dataclass

tests/benchmarks/default/test_benchmark_saga_sqlalchemy.py

@@ -10,7 +10,7 @@
 from cqrs.saga.saga import Saga
 from cqrs.saga.storage.sqlalchemy import SqlAlchemySagaStorage
 
-from ..storage_legacy import SqlAlchemySagaStorageLegacy
+from ..conftest import SqlAlchemySagaStorageLegacy
 from .test_benchmark_saga_memory import (
     OrderContext,
     ProcessPaymentStep,