From 907df5023f3b8fe7fee8ad27ac20a0f5a4877561 Mon Sep 17 00:00:00 2001
From: gifrerenom <lluis.gifre@cttc.es>
Date: Tue, 12 Sep 2023 09:15:55 +0000
Subject: [PATCH] Context component:

- Corrected signature of method consume_events
- Corrected topic name parsing of method consume_events
---
 src/context/service/database/Events.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/context/service/database/Events.py b/src/context/service/database/Events.py
index 8794032b1..36774a517 100644
--- a/src/context/service/database/Events.py
+++ b/src/context/service/database/Events.py
@@ -77,8 +77,9 @@ def notify_event_policy_rule(messagebroker : MessageBroker, event_type : EventTy
     notify_event(messagebroker, EventTopicEnum.POLICY_RULE, event_type, {'policyrule_id': policyrule_id})
 
 def consume_events(
-    messagebroker : MessageBroker, topic_names : Set[str], consume_timeout : float = CONSUME_TIMEOUT
+    messagebroker : MessageBroker, topic_enums : Set[EventTopicEnum], consume_timeout : float = CONSUME_TIMEOUT
 ) -> Iterator:
+    topic_names = [topic_enum.value for topic_enum in topic_enums]
     for message in messagebroker.consume(topic_names, consume_timeout=consume_timeout):
         event_class = TOPIC_TO_EVENTCLASS.get(message.topic)
         if event_class is None:
-- 
GitLab