fix after review

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

tests/integration/conftest.py

@@ -23,11 +23,8 @@
     OrderSaga,
     PaymentProcessedEvent,
     PaymentProcessedEventHandler,
-    ProcessPaymentResponse,
     ProcessPaymentStep,
-    ReserveInventoryResponse,
     ReserveInventoryStep,
-    ShipOrderResponse,
     ShipOrderStep,
 )
 

tests/integration/test_saga_mediator_sqlalchemy_postgres.py

@@ -1,19 +1,22 @@
 """Integration tests for SagaMediator with SqlAlchemySagaStorage (PostgreSQL)."""
 
 import uuid
+from unittest import mock
 
 import pytest
 from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker
 
 import cqrs
+from cqrs import events, SagaMap
 from cqrs.saga.storage.enums import SagaStatus
 from cqrs.saga.storage.sqlalchemy import SqlAlchemySagaStorage
+from integration.conftest import _TestContainer
 
 from tests.integration.test_saga_mediator_memory import (
     FailingOrderSaga,
     InventoryReservedEventHandler,
     OrderContext,
-    OrderShippedEventHandler,
+    OrderSaga, OrderShippedEventHandler,
     PaymentProcessedEventHandler,
     ProcessPaymentResponse,
     ProcessPaymentStep,

tests/integration/test_saga_storage_sqlalchemy_postgres.py

@@ -2,7 +2,6 @@
 Uses DATABASE_DSN_POSTGRESQL from fixtures (pytest-config.ini / env).
 """
 
-import asyncio
 import uuid
 from collections.abc import AsyncGenerator
 from datetime import datetime, timedelta, timezone
@@ -253,9 +252,7 @@ async def test_get_sagas_for_recovery_ordered_by_updated_at(
         later = datetime.now(timezone.utc) + timedelta(seconds=10)
         async with storage.session_factory() as session:
             await session.execute(
-                update(SagaExecutionModel)
-                .where(SagaExecutionModel.id == id2)
-                .values(updated_at=later),
+                update(SagaExecutionModel).where(SagaExecutionModel.id == id2).values(updated_at=later),
             )
             await session.commit()
         ids = await storage.get_sagas_for_recovery(limit=10)