X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=tests%2Fconftest.py;h=b653ae8578572e7d5848fba189da09590e20b435;hb=refs%2Fheads%2Fi-release;hp=68da7c48bdd26a4148b0d7f101168ad999c109b1;hpb=9ee90ffa2414326c26fd10edc59bf315204254e2;p=pti%2Fo2.git diff --git a/tests/conftest.py b/tests/conftest.py index 68da7c4..b653ae8 100644 --- 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