fix after review

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

tests/integration/test_saga_mediator_sqlalchemy_mysql.py

@@ -1,19 +1,21 @@
 """Integration tests for SagaMediator with SqlAlchemySagaStorage (MySQL)."""
 
 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 tests.integration.test_saga_mediator_memory import (
-    FailingOrderSaga,
+    _TestContainer, FailingOrderSaga,
     InventoryReservedEventHandler,
     OrderContext,
-    OrderShippedEventHandler,
+    OrderSaga, OrderShippedEventHandler,
     PaymentProcessedEventHandler,
     ProcessPaymentResponse,
     ProcessPaymentStep,

tests/integration/test_saga_mediator_sqlalchemy_postgres.py

@@ -16,7 +16,8 @@
     FailingOrderSaga,
     InventoryReservedEventHandler,
     OrderContext,
-    OrderSaga, OrderShippedEventHandler,
+    OrderSaga,
+    OrderShippedEventHandler,
     PaymentProcessedEventHandler,
     ProcessPaymentResponse,
     ProcessPaymentStep,