test(slack): cover the langfuse trace url built from the logger host

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>

tests/test_litellm/integrations/SlackAlerting/test_slack_alerting_utils.py

@@ -1,15 +1,9 @@
-import json
-from typing import Optional
 from unittest.mock import MagicMock
 
 import pytest
 
 # Adds the grandparent directory to sys.path to allow importing project modules
-
 import litellm
-from litellm.integrations.langfuse.langfuse_prompt_management import (
-    LangfusePromptManagement,
-)
 from litellm.integrations.SlackAlerting.utils import _add_langfuse_trace_id_to_alert
 from litellm.litellm_core_utils.logging_callback_manager import LoggingCallbackManager
 
@@ -34,3 +28,32 @@ async def test_langfuse_not_initialized_returns_none_early():
 
     # Verify the litellm_logging_obj was never accessed (early return)
     request_data["litellm_logging_obj"].assert_not_called()
+
+
+@pytest.mark.asyncio
+async def test_langfuse_trace_url_uses_logger_host(monkeypatch):
+    from litellm.integrations.langfuse.langfuse import LangFuseLogger
+
+    monkeypatch.setattr(litellm, "success_callback", ["langfuse"])
+    logger = LangFuseLogger(
+        langfuse_public_key="pk-slack-test",
+        langfuse_secret="sk-slack-test",
+        langfuse_host="http://127.0.0.1:1",
+    )
+    logging_obj = MagicMock()
+    logging_obj._get_trace_id.return_value = "abc123"
+    logging_obj._get_callback_object.return_value = logger
+
+    result = await _add_langfuse_trace_id_to_alert({"litellm_logging_obj": logging_obj})
+
+    assert result == "http://127.0.0.1:1/trace/abc123"
+
+
+@pytest.mark.asyncio
+async def test_langfuse_trace_url_skips_non_langfuse_callback(monkeypatch):
+    monkeypatch.setattr(litellm, "success_callback", ["langfuse"])
+    logging_obj = MagicMock()
+    logging_obj._get_trace_id.return_value = "abc123"
+    logging_obj._get_callback_object.return_value = object()
+
+    assert await _add_langfuse_trace_id_to_alert({"litellm_logging_obj": logging_obj}) is None