X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=o2app%2Fadapter%2Funit_of_work.py;h=e4885423e9b2ff20868d02453676c7fe24045bb2;hb=7fb3d02a3744ecb368b0531f16175034b987d42f;hp=2a8663d703600f828c3eeec11399b3fa8a2bb7c5;hpb=84867b76a65efc4e9add52c86eae7e451c4a038d;p=pti%2Fo2.git

diff --git a/o2app/adapter/unit_of_work.py b/o2app/adapter/unit_of_work.py
index 2a8663d..e488542 100644
--- a/o2app/adapter/unit_of_work.py
+++ b/o2app/adapter/unit_of_work.py
@@ -52,6 +52,10 @@ class SqlAlchemyUnitOfWork(AbstractUnitOfWork):
             .DeploymentManagerSqlAlchemyRepository(self.session)
         self.nfdeployment_descs = dms_repository\
             .NfDeploymentDescSqlAlchemyRepository(self.session)
+        self.nfdeployments = dms_repository\
+            .NfDeploymentSqlAlchemyRepository(self.session)
+        self.ocloudvresources = dms_repository\
+            .NfOCloudVResourceSqlAlchemyRepository(self.session)
         return super().__enter__()
 
     def __exit__(self, *args):
@@ -83,3 +87,9 @@ class SqlAlchemyUnitOfWork(AbstractUnitOfWork):
         for entry in self.nfdeployment_descs.seen:
             while entry.events:
                 yield entry.events.pop(0)
+        for entry in self.nfdeployments.seen:
+            while entry.events:
+                yield entry.events.pop(0)
+        for entry in self.ocloudvresources.seen:
+            while entry.events:
+                yield entry.events.pop(0)