fix tests

Author: renardeinside

tests/test_functions.py

@@ -54,9 +54,9 @@ class DummySettings:
 
 
 @mock.patch(
-    "unitycatalog_mcp.tools.functions.DatabricksFunctionClient", new=DummyClient
+    "databricks.labs.mcp.servers.unity_catalog.tools.functions.DatabricksFunctionClient", new=DummyClient
 )
-@mock.patch("unitycatalog_mcp.tools.functions.UCFunctionToolkit", new=DummyToolkit)
+@mock.patch("databricks.labs.mcp.servers.unity_catalog.tools.functions.UCFunctionToolkit", new=DummyToolkit)
 def test_list_uc_function_tools():
     settings = DummySettings()
     tools = list_uc_function_tools(settings)

tests/test_genie.py

@@ -63,7 +63,7 @@ class DummyWorkspaceClient:
     pass
 
 
-@mock.patch("unitycatalog_mcp.tools.genie.WorkspaceClient", new=DummyWorkspaceClient)
+@mock.patch("databricks.labs.mcp.servers.unity_catalog.tools.genie.WorkspaceClient", new=DummyWorkspaceClient)
 def test_genie_tool_execute():
     mock_func = mock.Mock()
     mock_func.return_value = [mock.Mock(text="hello world")]

tests/test_vector_search.py

@@ -32,9 +32,9 @@ class DummySettings:
 
 
 @mock.patch(
-    "unitycatalog_mcp.tools.vector_search.WorkspaceClient", new=DummyWorkspaceClient
+    "databricks.labs.mcp.servers.unity_catalog.tools.vector_search.WorkspaceClient", new=DummyWorkspaceClient
 )
-@mock.patch("unitycatalog_mcp.tools.vector_search.VectorSearchRetrieverTool")
+@mock.patch("databricks.labs.mcp.servers.unity_catalog.tools.vector_search.VectorSearchRetrieverTool")
 def test_list_vector_search_tools_filters_and_returns_expected(
     MockVectorSearchRetrieverTool,
 ):
@@ -52,7 +52,7 @@ def test_list_vector_search_tools_filters_and_returns_expected(
 
 def test_internal_list_vector_search_tools_direct():
     with mock.patch(
-        "unitycatalog_mcp.tools.vector_search.VectorSearchRetrieverTool"
+        "databricks.labs.mcp.servers.unity_catalog.tools.vector_search.VectorSearchRetrieverTool"
     ) as MockVectorSearchRetrieverTool:
         MockVectorSearchRetrieverTool.side_effect = lambda index_name: mock.Mock(
             tool={