Merge branch 'KAB-920-fix-get-table-metadata' into KAB-876-MCP-tool-for-updating-metadata

Author: ncapek

pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "hatchling.build"
 [project]
 name = "keboola-mcp-server"
 
-version = "0.2.1"
+version = "0.2.2"
 description = "MCP server for interacting with Keboola Connection"
 readme = "README.md"
 requires-python = ">=3.10"

src/keboola_mcp_server/storage_tools.py

@@ -165,10 +165,12 @@ async def get_table_metadata(
     workspace_manager = WorkspaceManager.from_state(ctx.session.state)
     assert isinstance(workspace_manager, WorkspaceManager)
 
+    table_fqn = await workspace_manager.get_table_fqn(raw_table)
+
     return TableDetail(
         **raw_table,
         column_identifiers=column_info,
-        db_identifier=workspace_manager.get_table_fqn(raw_table).snowflake_fqn,
+        db_identifier=table_fqn.snowflake_fqn,
     )
 
 

tests/test_storage_tools.py

@@ -1,5 +1,5 @@
 from typing import Any, Dict, List
-from unittest.mock import MagicMock
+from unittest.mock import AsyncMock, MagicMock
 
 import pytest
 
@@ -161,7 +161,7 @@ async def test_get_table_metadata(mcp_context_client, mock_table_data) -> None:
     workspace_manager = mcp_context_client.session.state["workspace_manager"]
     mock_fqn = MagicMock()
     mock_fqn.snowflake_fqn = mock_table_data["additional_data"]["snowflake_fqn"]
-    workspace_manager.get_table_fqn = MagicMock(return_value=mock_fqn)
+    workspace_manager.get_table_fqn = AsyncMock(return_value=mock_fqn)
     result = await get_table_metadata(mock_table_data["raw_table_data"]["id"], mcp_context_client)
 
     assert isinstance(result, TableDetail)