Update domain for event handler

[pti/o2.git] / o2ims / domain / ocloud_repo.py

diff --git a/o2ims/domain/ocloud_repo.py b/o2ims/domain/ocloud_repo.py
index 48ad840..f3b43d2 100644 (file)
--- a/o2ims/domain/ocloud_repo.py
+++ b/o2ims/domain/ocloud_repo.py
@@ -36,6 +36,7 @@ class OcloudRepository(abc.ABC):
 \r
     def update(self, ocloud: ocloud.Ocloud):\r
         self._update(ocloud)\r
+        self.seen.add(ocloud)\r
 \r
     # def update_fields(self, ocloudid: str, updatefields: dict):\r
     #     self._update(ocloudid, updatefields)\r
@@ -72,6 +73,7 @@ class ResourceTypeRepository(abc.ABC):
 \r
     def update(self, resource_type: ocloud.ResourceType):\r
         self._update(resource_type)\r
+        self.seen.add(resource_type)\r
 \r
     @abc.abstractmethod\r
     def _add(self, resource_type: ocloud.ResourceType):\r
@@ -105,6 +107,7 @@ class ResourcePoolRepository(abc.ABC):
 \r
     def update(self, resource_pool: ocloud.ResourcePool):\r
         self._update(resource_pool)\r
+        self.seen.add(resource_pool)\r
 \r
     @abc.abstractmethod\r
     def _add(self, resource_pool: ocloud.ResourcePool):\r
@@ -138,6 +141,7 @@ class ResourceRepository(abc.ABC):
 \r
     def update(self, resource: ocloud.Resource):\r
         self._update(resource)\r
+        self.seen.add(resource)\r
 \r
     @abc.abstractmethod\r
     def _add(self, resource: ocloud.Resource):\r