small fixes

Author: catalinsymphony

symphony/bdk/core/service/datafeed/datafeed_loop_v1.py

@@ -81,7 +81,7 @@ async def _read_datafeed(self):
         events = await self._datafeed_api.v4_datafeed_id_read_get(
             id=self._datafeed_id, session_token=session_token, key_manager_token=key_manager_token
         )
-        return events.value if events is not None and events.value else []
+        return events if events is not None else []
 
     async def _log_listener_exception(self, task):
         exception = task.exception()

symphony/bdk/core/service/message/message_parser.py

@@ -7,7 +7,7 @@
 import json
 from enum import Enum
 from json import JSONDecodeError
-from typing import Dict
+from typing import Dict, List
 
 from defusedxml.ElementTree import ParseError, fromstring, tostring
 
@@ -30,7 +30,7 @@ def get_text_content_from_message(message: V4Message) -> str:
         ) from exc
 
 
-def get_mentions(message: V4Message) -> [int]:
+def get_mentions(message: V4Message) -> List[int]:
     """Parse data inside an incoming message and returns a list containing the user ids corresponding
     to the users mentioned
 
@@ -41,7 +41,7 @@ def get_mentions(message: V4Message) -> [int]:
     return [int(user_id) for user_id in mentions_list]
 
 
-def get_hashtags(message: V4Message) -> [str]:
+def get_hashtags(message: V4Message) -> List[str]:
     """Parse data inside an incoming message and returns a list containing the text of the hashtags found
 
     :param message: message incoming V4 message to be parsed
@@ -50,7 +50,7 @@ def get_hashtags(message: V4Message) -> [str]:
     return _get_tags(message, _EntityTypeEnum.HASHTAG)
 
 
-def get_cashtags(message: V4Message) -> [str]:
+def get_cashtags(message: V4Message) -> List[str]:
     """Parse data inside an incoming message and returns a list containing the text of the cashtags found
 
     :param message: message incoming V4 message to be parsed

tests/bdk/integration/helpers.py

@@ -65,7 +65,7 @@ async def send_messages(messages, stream_id, since, uuid):
 
 async def get_test_messages(bdk, since, uuid):
     messages = await bdk.messages().list_messages(STREAM_ID, since=since)
-    cleaned_messages_text = [re.sub(r"<[^>]+>", " ", msg["message"]).strip() for msg in messages]
+    cleaned_messages_text = [re.sub(r"<[^>]+>", " ", msg.message.rstrip()) for msg in messages]
     return list(
         filter(
             lambda msg: msg.startswith(uuid),