Pagination in request and response; Fix alarm client issue

[pti/o2.git] / tests / conftest.py

diff --git a/tests/conftest.py b/tests/conftest.py
index 68da7c4..b653ae8 100644 (file)
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -27,6 +27,7 @@ from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker, clear_mappers
 from tenacity import retry, stop_after_delay
 from unittest.mock import MagicMock
+from mock_alchemy.mocking import UnifiedAlchemyMagicMock
 
 from o2app.bootstrap import bootstrap
 from o2ims.views import configure_namespace
@@ -46,6 +47,13 @@ def mock_uow():
     return session, uow
 
 
+@pytest.fixture
+def mock_alchemy_uow():
+    session = UnifiedAlchemyMagicMock()
+    uow = unit_of_work.SqlAlchemyUnitOfWork(session_factory=session)
+    return session, uow
+
+
 @pytest.fixture
 def mock_flask_uow(mock_uow):
     session, uow = mock_uow