Add the command that registers to the SMO; Make the create registration and create...

[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 facfc45..de48001 100644 (file)
--- a/o2app/adapter/unit_of_work.py
+++ b/o2app/adapter/unit_of_work.py
@@ -51,6 +51,8 @@ class SqlAlchemyUnitOfWork(AbstractUnitOfWork):
             .ResourceSqlAlchemyRepository(self.session)\r
         self.subscriptions = ocloud_repository\\r
             .SubscriptionSqlAlchemyRepository(self.session)\r
+        self.registrations = ocloud_repository\\r
+            .RegistrationSqlAlchemyRepository(self.session)\r
         self.deployment_managers = ocloud_repository\\r
             .DeploymentManagerSqlAlchemyRepository(self.session)\r
         self.nfdeployment_descs = dms_repository\\r
@@ -88,6 +90,12 @@ class SqlAlchemyUnitOfWork(AbstractUnitOfWork):
         for entry in self.deployment_managers.seen:\r
             while hasattr(entry, 'events') and len(entry.events) > 0:\r
                 yield entry.events.pop(0)\r
+        for entry in self.subscriptions.seen:\r
+            while hasattr(entry, 'events') and len(entry.events) > 0:\r
+                yield entry.events.pop(0)\r
+        for entry in self.registrations.seen:\r
+            while hasattr(entry, 'events') and len(entry.events) > 0:\r
+                yield entry.events.pop(0)\r
         for entry in self.nfdeployment_descs.seen:\r
             while hasattr(entry, 'events') and len(entry.events) > 0:\r
                 yield entry.events.pop(0)\r