Update o2dms lcm api

[pti/o2.git] / o2app / adapter / unit_of_work.py

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