From 8650a2e1ec6b28811282c91c96a4a05b4608ba8d Mon Sep 17 00:00:00 2001
From: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
Date: Thu, 11 Jun 2026 19:16:59 +0000
Subject: [PATCH] test(harness): move TestDatabricksCompletion to
 chat_live_longtail

Audit 5 marks the class MIGRATE (live BaseLLMChatTest + BaseAnthropicChatTest
subclass) though the 8c suite mapping omits databricks; chat_live_longtail is
the env-gated long-tail suite, so it lands there. Only the offline
api-base/api-key error-path test remains in test_databricks.py.
---
 .../test_databricks_live.py                   | 23 +++++++++++++++++++
 tests/llm_translation/test_databricks.py      | 15 ------------
 2 files changed, 23 insertions(+), 15 deletions(-)
 create mode 100644 tests/harness_suites/chat_live_longtail/test_databricks_live.py

diff --git a/tests/harness_suites/chat_live_longtail/test_databricks_live.py b/tests/harness_suites/chat_live_longtail/test_databricks_live.py
new file mode 100644
index 00000000000..951d83e2519
--- /dev/null
+++ b/tests/harness_suites/chat_live_longtail/test_databricks_live.py
@@ -0,0 +1,23 @@
+import os
+import sys
+
+import pytest
+
+sys.path.insert(0, os.path.abspath("../.."))
+
+from base_llm_unit_tests import BaseLLMChatTest, BaseAnthropicChatTest
+
+
+@pytest.mark.skip(reason="Databricks rate limit errors")
+class TestDatabricksCompletion(BaseLLMChatTest, BaseAnthropicChatTest):
+    def get_base_completion_call_args(self) -> dict:
+        return {"model": "databricks/databricks-claude-3-7-sonnet"}
+
+    def get_base_completion_call_args_with_thinking(self) -> dict:
+        return {
+            "model": "databricks/databricks-claude-3-7-sonnet",
+            "thinking": {"type": "enabled", "budget_tokens": 1024},
+        }
+
+    def test_pdf_handling(self, pdf_messages):
+        pytest.skip("Databricks does not support PDF handling")
diff --git a/tests/llm_translation/test_databricks.py b/tests/llm_translation/test_databricks.py
index fec82b45043..d33c81d5b2c 100644
--- a/tests/llm_translation/test_databricks.py
+++ b/tests/llm_translation/test_databricks.py
@@ -8,7 +8,6 @@ sys.path.insert(
 )  # Adds the parent directory to the system path
 import litellm
 from litellm.exceptions import BadRequestError
-from base_llm_unit_tests import BaseLLMChatTest, BaseAnthropicChatTest
 
 
 @pytest.mark.parametrize("set_base", [True, False])
@@ -49,17 +48,3 @@ def test_throws_if_api_base_or_api_key_not_set_without_databricks_sdk(
     assert any(msg in str(exc) for msg in err_msg)
 
 
-@pytest.mark.skip(reason="Databricks rate limit errors")
-class TestDatabricksCompletion(BaseLLMChatTest, BaseAnthropicChatTest):
-    def get_base_completion_call_args(self) -> dict:
-        return {"model": "databricks/databricks-claude-3-7-sonnet"}
-
-    def get_base_completion_call_args_with_thinking(self) -> dict:
-        return {
-            "model": "databricks/databricks-claude-3-7-sonnet",
-            "thinking": {"type": "enabled", "budget_tokens": 1024},
-        }
-
-    def test_pdf_handling(self, pdf_messages):
-        pytest.skip("Databricks does not support PDF handling")
-