cache key after message publish

Author: tomkralidis

msc-wis2node-management/msc_wis2node/publisher.py

@@ -189,9 +189,6 @@ def publish_to_wis2(self, dataset: dict, url: str) -> None:
                 update_link = deepcopy(message['links'][0])
                 update_link['rel'] = 'update'
                 message['links'].append(update_link)
-            else:
-                self.cache.set(message['properties']['data_id'], 'published',
-                               ex=CACHE_EXPIRY_SECONDS)
 
         cache = dataset.get('cache', True)
         if not cache:
@@ -222,6 +219,11 @@ def publish_to_wis2(self, dataset: dict, url: str) -> None:
             }
         )
 
+        if self.cache is not None:
+            LOGGER.debug('Setting cache key')
+            self.cache.set(message['properties']['data_id'], 'published',
+                           ex=CACHE_EXPIRY_SECONDS)
+
     def _subtopic2dirpath(self, subtopic: str) -> str:
         """
         Transforms AMQP subtopic to directory path