mirror of
https://github.com/dnlbauer/cordra-mcp.git
synced 2026-09-10 21:55:30 +00:00
feat: remove cordra object resource and made design object retrieval a tool call
This commit is contained in:
@@ -14,8 +14,7 @@ from cordra_mcp.client import (
|
||||
)
|
||||
from cordra_mcp.server import (
|
||||
count_objects,
|
||||
get_cordra_design,
|
||||
get_cordra_object,
|
||||
get_cordra_design_object,
|
||||
get_object,
|
||||
search_objects,
|
||||
)
|
||||
@@ -45,124 +44,6 @@ def sample_digital_object() -> DigitalObject:
|
||||
)
|
||||
|
||||
|
||||
class TestGetCordraObject:
|
||||
"""Test the get_cordra_object resource handler."""
|
||||
|
||||
@patch("cordra_mcp.server.cordra_client")
|
||||
async def test_get_object_success(
|
||||
self, mock_client: Any, sample_digital_object: DigitalObject
|
||||
) -> None:
|
||||
"""Test successful object retrieval."""
|
||||
mock_client.get_object = AsyncMock(return_value=sample_digital_object)
|
||||
|
||||
result = await get_cordra_object("people", "john-doe-123")
|
||||
|
||||
# Verify the result is valid JSON
|
||||
parsed_result = json.loads(result)
|
||||
assert parsed_result["id"] == "people/john-doe-123"
|
||||
assert parsed_result["type"] == "Person"
|
||||
assert parsed_result["content"]["name"] == "John Doe"
|
||||
assert parsed_result["content"]["birthday"] == "1990-05-15"
|
||||
assert parsed_result["metadata"]["created"] == "2023-01-01"
|
||||
assert len(parsed_result["payloads"]) == 1
|
||||
assert parsed_result["payloads"][0]["name"] == "profile_photo"
|
||||
|
||||
# Verify the client was called with the correct object ID
|
||||
mock_client.get_object.assert_called_once_with("people/john-doe-123")
|
||||
|
||||
@patch("cordra_mcp.server.cordra_client")
|
||||
async def test_get_object_not_found(self, mock_client: Any) -> None:
|
||||
"""Test object not found exception."""
|
||||
mock_client.get_object = AsyncMock(
|
||||
side_effect=CordraNotFoundError("Object not found: people/nonexistent")
|
||||
)
|
||||
|
||||
with pytest.raises(RuntimeError) as exc_info:
|
||||
await get_cordra_object("people", "nonexistent")
|
||||
|
||||
assert "Object not found: people/nonexistent" in str(exc_info.value)
|
||||
mock_client.get_object.assert_called_once_with("people/nonexistent")
|
||||
|
||||
@patch("cordra_mcp.server.cordra_client")
|
||||
async def test_get_object_client_error(self, mock_client: Any) -> None:
|
||||
"""Test general client error handling."""
|
||||
mock_client.get_object = AsyncMock(
|
||||
side_effect=CordraClientError("Connection failed")
|
||||
)
|
||||
|
||||
with pytest.raises(RuntimeError) as exc_info:
|
||||
await get_cordra_object("people", "john-doe-123")
|
||||
|
||||
assert "Failed to retrieve object people/john-doe-123" in str(exc_info.value)
|
||||
assert "Connection failed" in str(exc_info.value)
|
||||
mock_client.get_object.assert_called_once_with("people/john-doe-123")
|
||||
|
||||
@patch("cordra_mcp.server.cordra_client")
|
||||
async def test_object_id_construction(
|
||||
self, mock_client: Any, sample_digital_object: DigitalObject
|
||||
) -> None:
|
||||
"""Test that object ID is correctly constructed from prefix and suffix."""
|
||||
mock_client.get_object = AsyncMock(return_value=sample_digital_object)
|
||||
|
||||
# Test various prefix/suffix combinations
|
||||
test_cases = [
|
||||
("people", "john-doe-123", "people/john-doe-123"),
|
||||
("documents", "report-2023", "documents/report-2023"),
|
||||
("items", "item_with_underscores", "items/item_with_underscores"),
|
||||
]
|
||||
|
||||
for prefix, suffix, expected_id in test_cases:
|
||||
await get_cordra_object(prefix, suffix)
|
||||
mock_client.get_object.assert_called_with(expected_id)
|
||||
|
||||
@patch("cordra_mcp.server.cordra_client")
|
||||
async def test_json_formatting(
|
||||
self, mock_client: Any, sample_digital_object: DigitalObject
|
||||
) -> None:
|
||||
"""Test that the returned JSON is properly formatted."""
|
||||
mock_client.get_object = AsyncMock(return_value=sample_digital_object)
|
||||
|
||||
result = await get_cordra_object("people", "john-doe-123")
|
||||
|
||||
# Verify it's valid JSON with proper indentation
|
||||
parsed_result = json.loads(result)
|
||||
assert isinstance(parsed_result, dict)
|
||||
|
||||
# Check that the result contains indentation (pretty-printed)
|
||||
assert " " in result # Should have 2-space indentation
|
||||
|
||||
# Verify all expected fields are present
|
||||
assert "id" in parsed_result
|
||||
assert "type" in parsed_result
|
||||
assert "content" in parsed_result
|
||||
assert "metadata" in parsed_result
|
||||
assert "acl" in parsed_result
|
||||
assert "payloads" in parsed_result
|
||||
|
||||
@patch("cordra_mcp.server.cordra_client")
|
||||
async def test_minimal_object(self, mock_client: Any) -> None:
|
||||
"""Test handling of object with minimal data."""
|
||||
minimal_object = DigitalObject(
|
||||
id="test/minimal",
|
||||
type="",
|
||||
content={"id": "test/minimal"},
|
||||
metadata=None,
|
||||
acl=None,
|
||||
payloads=None,
|
||||
)
|
||||
mock_client.get_object = AsyncMock(return_value=minimal_object)
|
||||
|
||||
result = await get_cordra_object("test", "minimal")
|
||||
parsed_result = json.loads(result)
|
||||
|
||||
assert parsed_result["id"] == "test/minimal"
|
||||
assert parsed_result["type"] == ""
|
||||
assert parsed_result["content"]["id"] == "test/minimal"
|
||||
assert parsed_result["metadata"] is None
|
||||
assert parsed_result["acl"] is None
|
||||
assert parsed_result["payloads"] is None
|
||||
|
||||
|
||||
class TestGetObject:
|
||||
"""Test the get_object tool."""
|
||||
|
||||
@@ -663,7 +544,7 @@ class TestSearchObjects:
|
||||
|
||||
|
||||
class TestGetCordraDesign:
|
||||
"""Test the get_cordra_design resource handler."""
|
||||
"""Test the get_cordra_design_object tool."""
|
||||
|
||||
@patch("cordra_mcp.server.cordra_client")
|
||||
async def test_get_design_success(self, mock_client: Any) -> None:
|
||||
@@ -680,7 +561,7 @@ class TestGetCordraDesign:
|
||||
)
|
||||
mock_client.get_design = AsyncMock(return_value=mock_design)
|
||||
|
||||
result = await get_cordra_design()
|
||||
result = await get_cordra_design_object()
|
||||
|
||||
# Verify the result is valid JSON
|
||||
parsed_result = json.loads(result)
|
||||
@@ -701,7 +582,7 @@ class TestGetCordraDesign:
|
||||
)
|
||||
|
||||
with pytest.raises(RuntimeError) as exc_info:
|
||||
await get_cordra_design()
|
||||
await get_cordra_design_object()
|
||||
|
||||
assert "Design object not found" in str(exc_info.value)
|
||||
mock_client.get_design.assert_called_once()
|
||||
@@ -714,7 +595,7 @@ class TestGetCordraDesign:
|
||||
)
|
||||
|
||||
with pytest.raises(RuntimeError) as exc_info:
|
||||
await get_cordra_design()
|
||||
await get_cordra_design_object()
|
||||
|
||||
assert "Authentication failed" in str(exc_info.value)
|
||||
mock_client.get_design.assert_called_once()
|
||||
@@ -727,7 +608,7 @@ class TestGetCordraDesign:
|
||||
)
|
||||
|
||||
with pytest.raises(RuntimeError) as exc_info:
|
||||
await get_cordra_design()
|
||||
await get_cordra_design_object()
|
||||
|
||||
assert "Failed to retrieve design object" in str(exc_info.value)
|
||||
assert "Connection failed" in str(exc_info.value)
|
||||
@@ -744,7 +625,7 @@ class TestGetCordraDesign:
|
||||
)
|
||||
mock_client.get_design = AsyncMock(return_value=mock_design)
|
||||
|
||||
result = await get_cordra_design()
|
||||
result = await get_cordra_design_object()
|
||||
|
||||
# Verify it's valid JSON with proper indentation
|
||||
parsed_result = json.loads(result)
|
||||
|
||||
Reference in New Issue
Block a user