AI-1169 add links to more tests

Author: Radoslav Klic

tests/conftest.py

@@ -21,7 +21,7 @@ def keboola_client(mocker) -> KeboolaClient:
     client.storage_client = mocker.MagicMock(AsyncStorageClient)
     client.storage_client.base_api_url = 'test://api.keboola.com'
     client.storage_client.branch_id = 'default'
-    client.storage_client.project_id.return_value = '123'
+    client.storage_client.project_id.return_value = '69420'
     client.jobs_queue_client = mocker.MagicMock(JobsQueueClient)
     client.ai_service_client = mocker.MagicMock(AIServiceClient)
 

tests/tools/components/test_tools.py

@@ -5,6 +5,7 @@
 from pytest_mock import MockerFixture
 
 from keboola_mcp_server.client import KeboolaClient
+from keboola_mcp_server.links import Link
 from keboola_mcp_server.tools.components import (
     ComponentWithConfigurations,
     create_sql_transformation,
@@ -242,6 +243,21 @@ async def test_get_config(
     assert result.component is not None
     assert result.component.component_id == mock_component['id']
     assert result.component.component_name == mock_component['name']
+    assert set(result.links) == {
+        Link(
+            type='ui-detail',
+            title=f'Configuration: {mock_configuration["name"]}',
+            url=(
+                f'test://api.keboola.com/admin/projects/69420/components/'
+                f'{mock_component["id"]}/{mock_configuration["id"]}'
+            ),
+        ),
+        Link(
+            type='ui-dashboard',
+            title=f'{mock_component["id"]} Configurations Dashboard',
+            url=f'test://api.keboola.com/admin/projects/69420/components/{mock_component["id"]}',
+        ),
+    }
 
     # Verify the calls were made with the correct arguments
     keboola_client.storage_client.configuration_detail.assert_called_once_with(

tests/tools/test_jobs.py

@@ -244,8 +244,8 @@ async def test_run_job(
     assert job_detail.config_id == configuration_id
     assert job_detail.result == {}
     assert set(job_detail.links) == {
-        Link(type='ui-detail', title='Job: 123', url='test://api.keboola.com/admin/projects/123/queue/123'),
-        Link(type='ui-dashboard', title='Jobs in the project', url='test://api.keboola.com/admin/projects/123/queue'),
+        Link(type='ui-detail', title='Job: 123', url='test://api.keboola.com/admin/projects/69420/queue/123'),
+        Link(type='ui-dashboard', title='Jobs in the project', url='test://api.keboola.com/admin/projects/69420/queue'),
     }
 
     keboola_client.jobs_queue_client.create_job.assert_called_once_with(

tests/tools/test_storage.py

@@ -7,6 +7,7 @@
 
 from keboola_mcp_server.client import KeboolaClient
 from keboola_mcp_server.config import Config, MetadataField
+from keboola_mcp_server.links import Link
 from keboola_mcp_server.tools.storage import (
     BucketDetail,
     ListBucketsOutput,
@@ -186,6 +187,18 @@ async def test_get_bucket(
     assert result.id == expected_bucket['id']
     assert result.name == expected_bucket['name']
     assert result.display_name == expected_bucket['display_name']
+    assert set(result.links) == {
+        Link(
+            type='ui-detail',
+            title=f'Bucket: {expected_bucket["name"]}',
+            url=f'test://api.keboola.com/admin/projects/69420/storage/{bucket_id}',
+        ),
+        Link(
+            type='ui-dashboard',
+            title='Buckets in the project',
+            url='test://api.keboola.com/admin/projects/69420/storage',
+        ),
+    }
 
     # Check optional fields only if they are present in the expected bucket
     if 'description' in expected_bucket:
@@ -257,6 +270,25 @@ async def test_get_table(
     assert result.data_size_bytes == mock_table_data['raw_table_data']['data_size_bytes']
     assert result.fully_qualified_name == str(mock_table_data['additional_data']['table_fqn'])
     assert result.columns == mock_table_data['additional_data']['columns']
+    assert set(result.links) == {
+        Link(
+            type='ui-detail',
+            title=f'Table: {mock_table_data["raw_table_data"]["name"]}',
+            url=(
+                f'test://api.keboola.com/admin/projects/69420/storage/'
+                f'{mock_table_data["raw_table_data"]["bucket"]["id"]}/table/'
+                f'{mock_table_data["raw_table_data"]["name"]}'
+            ),
+        ),
+        Link(
+            type='ui-detail',
+            title=f'Bucket: {mock_table_data["raw_table_data"]["bucket"]["id"]}',
+            url=(
+                f'test://api.keboola.com/admin/projects/69420/storage/'
+                f'{mock_table_data["raw_table_data"]["bucket"]["id"]}'
+            ),
+        ),
+    }
 
 
 @pytest.mark.asyncio