test preview

Author: Avi-Robusta

tests/test_slack_preview.py

@@ -3,7 +3,7 @@
 
 import pytest
 from robusta.core.reporting.base import Finding, FindingSeverity, FindingSource
-from robusta.core.sinks.slack.slack_sink_params import SlackSinkParams, SlackTemplateStyle
+from robusta.core.sinks.slack.preview.slack_sink_preview_params import SlackSinkPreviewParams
 from robusta.core.reporting.blocks import MarkdownBlock
 from robusta.integrations.slack.sender import SlackSender
 from tests.config import CONFIG
@@ -13,66 +13,96 @@
 TEST_CLUSTER = "test cluster"
 TEST_KEY = "test key"
 
-@pytest.mark.parametrize("template_style,custom_template,expected_phrases", [
-    (
-        SlackTemplateStyle.DEFAULT,
-        None,
-        [
-            "[Firing]",  # status block
-            ":bell: Type: Alert",  # context block
-            "Severity: Info",
-            ":globe_with_meridians: Cluster: test cluster"
-        ]
-    ),
-    (
-        SlackTemplateStyle.LEGACY,
-        None,
-        [
-            "Prometheus Alert Firing",  # header block
-            "*Source:* `test cluster`"
-        ]
-    ),
-    (
-        SlackTemplateStyle.DEFAULT,
-        "custom_template.j2",
-        [
-            "CUSTOM TEMPLATE:",
-        ]
+def test_slack_preview_default_template(slack_channel: SlackChannel):
+    slack_sender = SlackSender(
+        CONFIG.PYTEST_IN_CLUSTER_SLACK_TOKEN, TEST_ACCOUNT, TEST_CLUSTER, TEST_KEY, slack_channel.channel_name
+    )
+    finding = Finding(
+        title="Test Preview Template",
+        aggregation_key="test-preview-template",
+        severity=FindingSeverity.INFO,
+        source=FindingSource.PROMETHEUS,
+        description="Testing preview template rendering"
     )
-])
-def test_slack_template_styles(slack_channel: SlackChannel, template_style, template_name, expected_phrases):
+    finding.add_enrichment([MarkdownBlock("This is a test preview block.")])
+
+    preview_params = SlackSinkPreviewParams(
+        name="test_preview",
+        slack_channel=slack_channel.channel_name,
+        api_key="",
+        investigate_link=True,
+        prefer_redirect_to_platform=False,
+        max_log_file_limit_kb=1000
+    )
+
+    slack_sender.send_finding_to_slack(finding, preview_params, platform_enabled=True)
+    latest_message = slack_channel.get_latest_message()
+    assert "Test Preview Template" in latest_message
+    assert "This is a test preview block." in latest_message
+
+def test_slack_preview_custom_template(slack_channel: SlackChannel):
     slack_sender = SlackSender(
         CONFIG.PYTEST_IN_CLUSTER_SLACK_TOKEN, TEST_ACCOUNT, TEST_CLUSTER, TEST_KEY, slack_channel.channel_name
     )
     finding = Finding(
-        title="Test Template Style",
-        aggregation_key="test-template-style",
+        title="Test Custom Preview Template",
+        aggregation_key="test-custom-preview-template",
         severity=FindingSeverity.INFO,
         source=FindingSource.PROMETHEUS,
-        description="Testing template style rendering"
+        description="Testing custom preview template rendering"
     )
-    finding.add_enrichment([MarkdownBlock("This is a test block.")])
+    finding.add_enrichment([MarkdownBlock("This is a test custom preview block.")])
+
+    custom_template = """{
+      "type": "section",
+      "text": {
+        "type": "mrkdwn",
+        "text": "CUSTOM PREVIEW: {{ title }}"
+      }
+    }"""
 
-    slack_params = SlackSinkParams(
-        name="test_slack",
+    preview_params = SlackSinkPreviewParams(
+        name="test_custom_preview",
         slack_channel=slack_channel.channel_name,
         api_key="",
-        template_name=template_name,
-        slack_custom_templates={
-        "custom_template.j2": """
-        {
-          "type": "section",
-          "text": {
-            "type": "mrkdwn",
-            "text": "CUSTOM TEMPLATE: {{ title }}"
-          }
-        }
-        """,
-        })
+        investigate_link=True,
+        prefer_redirect_to_platform=False,
+        max_log_file_limit_kb=1000,
+        slack_custom_templates={"custom.j2": custom_template}
+    )
+
+    slack_sender.send_finding_to_slack(finding, preview_params, platform_enabled=True)
+    latest_message = slack_channel.get_latest_message()
+    assert "CUSTOM PREVIEW: Test Custom Preview Template" in latest_message
+    assert "This is a test custom preview block." in latest_message
+
+def test_slack_preview_channel_override(slack_channel: SlackChannel):
+    slack_sender = SlackSender(
+        CONFIG.PYTEST_IN_CLUSTER_SLACK_TOKEN, TEST_ACCOUNT, TEST_CLUSTER, TEST_KEY, slack_channel.channel_name
+    )
+    finding = Finding(
+        title="Test Channel Override",
+        aggregation_key="test-channel-override",
+        severity=FindingSeverity.INFO,
+        source=FindingSource.PROMETHEUS,
+        description="Testing channel override functionality"
+    )
+    finding.add_enrichment([MarkdownBlock("This is a test channel override block.")])
+
+    # Add a label that will be used for channel override
+    finding.subject.labels["slack"] = slack_channel.channel_name
+
+    preview_params = SlackSinkPreviewParams(
+        name="test_channel_override",
+        slack_channel="default-channel",  # This will be overridden
+        api_key="",
+        investigate_link=True,
+        prefer_redirect_to_platform=False,
+        max_log_file_limit_kb=1000,
+        channel_override="labels.slack"  # This will read from the slack label
+    )
 
-    slack_sender.send_finding_to_slack(finding, slack_params, False)
+    slack_sender.send_finding_to_slack(finding, preview_params, platform_enabled=True)
     latest_message = slack_channel.get_latest_message()
-    for phrase in expected_phrases:
-        assert phrase in latest_message
-    assert "Test Template Style" in latest_message
-    assert "This is a test block." in latest_message
+    assert "Test Channel Override" in latest_message
+    assert "This is a test channel override block." in latest_message