update slack_utils

Author: Avi-Robusta

tests/test_slack_preview.py

@@ -55,7 +55,7 @@ def test_slack_preview_default_template(slack_channel: SlackChannel):
     )
 
     slack_sender.send_finding_to_slack(finding, preview_params, platform_enabled=True)
-    latest_message = slack_channel.get_latest_message()
+    latest_message = slack_channel.get_complete_latest_message()
 
     # Check for key elements in the preview message
     assert "Test Preview Template" in latest_message
@@ -114,7 +114,7 @@ def test_slack_preview_custom_template(slack_channel: SlackChannel):
     )
 
     slack_sender.send_finding_to_slack(finding, preview_params, platform_enabled=True)
-    latest_message = slack_channel.get_latest_message()
+    latest_message = slack_channel.get_complete_latest_message()
 
     # Check for custom template elements
     assert "CUSTOM PREVIEW: Test Custom Preview Template" in latest_message

tests/utils/slack_utils.py

@@ -22,6 +22,11 @@ def get_latest_message(self):
         messages = results["messages"]
         return messages[0]["text"]
 
+    def get_complete_latest_message(self):
+        results = self.client.conversations_history(channel=self.channel_id)
+        messages = results["messages"]
+        return messages
+
     @staticmethod
     def _create_or_join_channel(client: WebClient, channel_name: str) -> str:
         """Creates or joins the specified slack channel and returns its channel_id"""