X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2ims%2Fadapter%2Funit_of_work.py;h=23c4cffae25fccb202ec681b2881246fac6543bd;hb=b10370cb16f07807860f8171432e554cec51e207;hp=c958ce24f5222e53b2f548ce0d333ecff536d0f7;hpb=8be81dfad35b08c4de77168e885bb18253069771;p=pti%2Fo2.git

diff --git a/o2ims/adapter/unit_of_work.py b/o2ims/adapter/unit_of_work.py
index c958ce2..23c4cff 100644
--- a/o2ims/adapter/unit_of_work.py
+++ b/o2ims/adapter/unit_of_work.py
@@ -19,7 +19,7 @@ from sqlalchemy.orm import sessionmaker
 from sqlalchemy.orm.session import Session
 
 from o2ims import config
-from o2ims.adapter.ocloud_repository import OcloudSqlAlchemyRepository
+from o2ims.adapter import ocloud_repository
 from o2ims.adapter.stx_repository import StxObjectSqlAlchemyRepository
 from o2ims.service.unit_of_work import AbstractUnitOfWork
 
@@ -38,7 +38,14 @@ class SqlAlchemyUnitOfWork(AbstractUnitOfWork):
 
     def __enter__(self):
         self.session = self.session_factory()  # type: Session
-        self.oclouds = OcloudSqlAlchemyRepository(self.session)
+        self.oclouds = ocloud_repository\
+            .OcloudSqlAlchemyRepository(self.session)
+        self.resource_types = ocloud_repository\
+            .ResouceTypeSqlAlchemyRepository(self.session)
+        self.resource_pools = ocloud_repository\
+            .ResourcePoolSqlAlchemyRepository(self.session)
+        self.resources = ocloud_repository\
+            .ResourceSqlAlchemyRepository(self.session)
         self.stxobjects = StxObjectSqlAlchemyRepository(self.session)
         return super().__enter__()