changess

Author: Avi-Robusta

src/robusta/core/sinks/slack/preview/slack_sink_preview.py

@@ -7,8 +7,5 @@ class SlackSinkPreview(SlackSink):
     params: SlackSinkPreviewParams
 
     def __init__(self, sink_config: SlackSinkPreviewConfigWrapper, registry):
-        self.slack_sender = slack_module.SlackSender(
-            self.api_key, self.account_id, self.cluster_name, self.signing_key, self.slack_channel, is_preview=True
-        )
-        super().__init__(sink_config.slack_sink, registry, self.slack_sender)
+        super().__init__(sink_config.slack_sink_preview, registry, is_preview=True)
 

src/robusta/core/sinks/slack/preview/slack_sink_preview_params.py

@@ -37,7 +37,7 @@ def get_custom_template(self) -> Optional[str]:
 
 
 class SlackSinkPreviewConfigWrapper(SinkConfigBase):
-    slack_sink: SlackSinkPreviewParams
+    slack_sink_preview: SlackSinkPreviewParams
 
     def get_params(self) -> SinkBaseParams:
-        return self.slack_sink
+        return self.slack_sink_preview

src/robusta/core/sinks/slack/slack_sink.py

@@ -10,16 +10,13 @@
 class SlackSink(SinkBase):
     params: SlackSinkParams
 
-    def __init__(self, sink_config: SlackSinkConfigWrapper, registry, slack_sender=None):
+    def __init__(self, sink_config: SlackSinkConfigWrapper, registry, is_preview=False):
         super().__init__(sink_config.slack_sink, registry)
         self.slack_channel = sink_config.slack_sink.slack_channel
         self.api_key = sink_config.slack_sink.api_key
-        if slack_sender:
-            self.slack_sender = slack_sender
-        else:
-            self.slack_sender = slack_module.SlackSender(
-                self.api_key, self.account_id, self.cluster_name, self.signing_key, self.slack_channel
-            )
+        self.slack_sender = slack_module.SlackSender(
+            self.api_key, self.account_id, self.cluster_name, self.signing_key, self.slack_channel, is_preview
+        )
         self.registry.subscribe("replace_callback_with_string", self)
 
     def handle_event(self, event_name: str, **kwargs):