|
5 | 5 | import pytest |
6 | 6 | from mcp import ListToolsResult |
7 | 7 | from mcp.types import CallToolResult as MCPCallToolResult |
8 | | -from mcp.types import GetPromptResult, ListPromptsResult, Prompt, PromptMessage |
| 8 | +from mcp.types import ( |
| 9 | + GetPromptResult, |
| 10 | + ListPromptsResult, |
| 11 | + ListResourcesResult, |
| 12 | + ListResourceTemplatesResult, |
| 13 | + Prompt, |
| 14 | + PromptMessage, |
| 15 | + ReadResourceResult, |
| 16 | + Resource, |
| 17 | + ResourceTemplate, |
| 18 | + TextResourceContents, |
| 19 | +) |
9 | 20 | from mcp.types import TextContent as MCPTextContent |
10 | 21 | from mcp.types import Tool as MCPTool |
| 22 | +from pydantic import AnyUrl |
11 | 23 |
|
12 | 24 | from strands.tools.mcp import MCPClient |
13 | 25 | from strands.tools.mcp.mcp_types import MCPToolResult |
@@ -772,3 +784,143 @@ def test_call_tool_sync_with_meta_and_structured_content(mock_transport, mock_se |
772 | 784 | assert result["metadata"] == metadata |
773 | 785 | assert "structuredContent" in result |
774 | 786 | assert result["structuredContent"] == structured_content |
| 787 | + |
| 788 | + |
| 789 | +# Resource Tests - Sync Methods |
| 790 | + |
| 791 | + |
| 792 | +def test_list_resources_sync(mock_transport, mock_session): |
| 793 | + """Test that list_resources_sync correctly retrieves resources.""" |
| 794 | + mock_resource = Resource( |
| 795 | + uri=AnyUrl("file://documents/test.txt"), name="test.txt", description="A test document", mimeType="text/plain" |
| 796 | + ) |
| 797 | + mock_session.list_resources.return_value = ListResourcesResult(resources=[mock_resource]) |
| 798 | + |
| 799 | + with MCPClient(mock_transport["transport_callable"]) as client: |
| 800 | + result = client.list_resources_sync() |
| 801 | + |
| 802 | + mock_session.list_resources.assert_called_once_with(cursor=None) |
| 803 | + assert len(result.resources) == 1 |
| 804 | + assert result.resources[0].name == "test.txt" |
| 805 | + assert str(result.resources[0].uri) == "file://documents/test.txt" |
| 806 | + assert result.nextCursor is None |
| 807 | + |
| 808 | + |
| 809 | +def test_list_resources_sync_with_pagination_token(mock_transport, mock_session): |
| 810 | + """Test that list_resources_sync correctly passes pagination token and returns next cursor.""" |
| 811 | + mock_resource = Resource( |
| 812 | + uri=AnyUrl("file://documents/test.txt"), name="test.txt", description="A test document", mimeType="text/plain" |
| 813 | + ) |
| 814 | + mock_session.list_resources.return_value = ListResourcesResult(resources=[mock_resource], nextCursor="next_page") |
| 815 | + |
| 816 | + with MCPClient(mock_transport["transport_callable"]) as client: |
| 817 | + result = client.list_resources_sync(pagination_token="current_page") |
| 818 | + |
| 819 | + mock_session.list_resources.assert_called_once_with(cursor="current_page") |
| 820 | + assert len(result.resources) == 1 |
| 821 | + assert result.resources[0].name == "test.txt" |
| 822 | + assert result.nextCursor == "next_page" |
| 823 | + |
| 824 | + |
| 825 | +def test_list_resources_sync_session_not_active(): |
| 826 | + """Test that list_resources_sync raises an error when session is not active.""" |
| 827 | + client = MCPClient(MagicMock()) |
| 828 | + |
| 829 | + with pytest.raises(MCPClientInitializationError, match="client session is not running"): |
| 830 | + client.list_resources_sync() |
| 831 | + |
| 832 | + |
| 833 | +def test_read_resource_sync(mock_transport, mock_session): |
| 834 | + """Test that read_resource_sync correctly reads a resource.""" |
| 835 | + mock_content = TextResourceContents( |
| 836 | + uri=AnyUrl("file://documents/test.txt"), text="Resource content", mimeType="text/plain" |
| 837 | + ) |
| 838 | + mock_session.read_resource.return_value = ReadResourceResult(contents=[mock_content]) |
| 839 | + |
| 840 | + with MCPClient(mock_transport["transport_callable"]) as client: |
| 841 | + result = client.read_resource_sync("file://documents/test.txt") |
| 842 | + |
| 843 | + # Verify the session method was called |
| 844 | + mock_session.read_resource.assert_called_once() |
| 845 | + # Check the URI argument (it will be wrapped as AnyUrl) |
| 846 | + call_args = mock_session.read_resource.call_args[0] |
| 847 | + assert str(call_args[0]) == "file://documents/test.txt" |
| 848 | + |
| 849 | + assert len(result.contents) == 1 |
| 850 | + assert result.contents[0].text == "Resource content" |
| 851 | + |
| 852 | + |
| 853 | +def test_read_resource_sync_with_anyurl(mock_transport, mock_session): |
| 854 | + """Test that read_resource_sync correctly handles AnyUrl input.""" |
| 855 | + mock_content = TextResourceContents( |
| 856 | + uri=AnyUrl("file://documents/test.txt"), text="Resource content", mimeType="text/plain" |
| 857 | + ) |
| 858 | + mock_session.read_resource.return_value = ReadResourceResult(contents=[mock_content]) |
| 859 | + |
| 860 | + with MCPClient(mock_transport["transport_callable"]) as client: |
| 861 | + uri = AnyUrl("file://documents/test.txt") |
| 862 | + result = client.read_resource_sync(uri) |
| 863 | + |
| 864 | + mock_session.read_resource.assert_called_once() |
| 865 | + call_args = mock_session.read_resource.call_args[0] |
| 866 | + assert str(call_args[0]) == "file://documents/test.txt" |
| 867 | + |
| 868 | + assert len(result.contents) == 1 |
| 869 | + assert result.contents[0].text == "Resource content" |
| 870 | + |
| 871 | + |
| 872 | +def test_read_resource_sync_session_not_active(): |
| 873 | + """Test that read_resource_sync raises an error when session is not active.""" |
| 874 | + client = MCPClient(MagicMock()) |
| 875 | + |
| 876 | + with pytest.raises(MCPClientInitializationError, match="client session is not running"): |
| 877 | + client.read_resource_sync("file://documents/test.txt") |
| 878 | + |
| 879 | + |
| 880 | +def test_list_resource_templates_sync(mock_transport, mock_session): |
| 881 | + """Test that list_resource_templates_sync correctly retrieves resource templates.""" |
| 882 | + mock_template = ResourceTemplate( |
| 883 | + uriTemplate="file://documents/{name}", |
| 884 | + name="document_template", |
| 885 | + description="Template for documents", |
| 886 | + mimeType="text/plain", |
| 887 | + ) |
| 888 | + mock_session.list_resource_templates.return_value = ListResourceTemplatesResult(resourceTemplates=[mock_template]) |
| 889 | + |
| 890 | + with MCPClient(mock_transport["transport_callable"]) as client: |
| 891 | + result = client.list_resource_templates_sync() |
| 892 | + |
| 893 | + mock_session.list_resource_templates.assert_called_once_with(cursor=None) |
| 894 | + assert len(result.resourceTemplates) == 1 |
| 895 | + assert result.resourceTemplates[0].name == "document_template" |
| 896 | + assert result.resourceTemplates[0].uriTemplate == "file://documents/{name}" |
| 897 | + assert result.nextCursor is None |
| 898 | + |
| 899 | + |
| 900 | +def test_list_resource_templates_sync_with_pagination_token(mock_transport, mock_session): |
| 901 | + """Test that list_resource_templates_sync correctly passes pagination token and returns next cursor.""" |
| 902 | + mock_template = ResourceTemplate( |
| 903 | + uriTemplate="file://documents/{name}", |
| 904 | + name="document_template", |
| 905 | + description="Template for documents", |
| 906 | + mimeType="text/plain", |
| 907 | + ) |
| 908 | + mock_session.list_resource_templates.return_value = ListResourceTemplatesResult( |
| 909 | + resourceTemplates=[mock_template], nextCursor="next_page" |
| 910 | + ) |
| 911 | + |
| 912 | + with MCPClient(mock_transport["transport_callable"]) as client: |
| 913 | + result = client.list_resource_templates_sync(pagination_token="current_page") |
| 914 | + |
| 915 | + mock_session.list_resource_templates.assert_called_once_with(cursor="current_page") |
| 916 | + assert len(result.resourceTemplates) == 1 |
| 917 | + assert result.resourceTemplates[0].name == "document_template" |
| 918 | + assert result.nextCursor == "next_page" |
| 919 | + |
| 920 | + |
| 921 | +def test_list_resource_templates_sync_session_not_active(): |
| 922 | + """Test that list_resource_templates_sync raises an error when session is not active.""" |
| 923 | + client = MCPClient(MagicMock()) |
| 924 | + |
| 925 | + with pytest.raises(MCPClientInitializationError, match="client session is not running"): |
| 926 | + client.list_resource_templates_sync() |
0 commit comments