Blockscout MCP Server

from unittest.mock import AsyncMock, MagicMock, patch import httpx import pytest from blockscout_mcp_server.models import AddressInfoData, ToolResponse from blockscout_mcp_server.tools.address.get_address_info import get_address_info @pytest.mark.asyncio async def test_get_address_info_success_with_metadata(mock_ctx): """ Verify get_address_info correctly combines data from Blockscout and Metadata APIs. """ chain_id = "1" address = "0x123abc" mock_base_url = "https://eth.blockscout.com" mock_blockscout_response = { "hash": address, "is_contract": True, } mock_metadata_response = {"addresses": {address: {"tags": [{"name": "Test Tag"}]}}} with ( patch( "blockscout_mcp_server.tools.address.get_address_info.get_blockscout_base_url", new_callable=AsyncMock ) as mock_get_url, patch( "blockscout_mcp_server.tools.address.get_address_info.make_blockscout_request", new_callable=AsyncMock ) as mock_bs_request, patch( "blockscout_mcp_server.tools.address.get_address_info.make_metadata_request", new_callable=AsyncMock ) as mock_meta_request, ): mock_get_url.return_value = mock_base_url mock_bs_request.return_value = mock_blockscout_response mock_meta_request.return_value = mock_metadata_response result = await get_address_info(chain_id=chain_id, address=address, ctx=mock_ctx) mock_get_url.assert_called_once_with(chain_id) mock_bs_request.assert_called_once_with(base_url=mock_base_url, api_path=f"/api/v2/addresses/{address}") mock_meta_request.assert_called_once_with( api_path="/api/v1/metadata", params={"addresses": address, "chainId": chain_id} ) assert isinstance(result, ToolResponse) assert isinstance(result.data, AddressInfoData) assert result.data.basic_info == mock_blockscout_response assert result.data.metadata == mock_metadata_response["addresses"][address] assert result.notes is None expected_instructions = [ (f"Use `direct_api_call` with endpoint `/api/v2/addresses/{address}/logs` to get Logs Emitted by Address."), ( f"Use `direct_api_call` with endpoint `/api/v2/addresses/{address}/coin-balance-history-by-day` " "to get daily native coin balance history." ), ( f"Use `direct_api_call` with endpoint `/api/v2/addresses/{address}/coin-balance-history` " "to get native coin balance history." ), ( f"Use `direct_api_call` with endpoint `/api/v2/addresses/{address}/blocks-validated` " "to get Blocks Validated by this Address." ), ( f"Use `direct_api_call` with endpoint `/api/v2/proxy/account-abstraction/accounts/{address}` " "to get Account Abstraction info." ), ( f"Use `direct_api_call` with endpoint `/api/v2/proxy/account-abstraction/operations` " f"and query_params={{'sender': '{address}'}} to get User Operations sent by this Address." ), ] assert result.instructions is not None for instr in expected_instructions: assert instr in result.instructions assert mock_ctx.report_progress.await_count == 4 assert mock_ctx.info.await_count == 4 @pytest.mark.asyncio async def test_get_address_info_success_without_metadata(mock_ctx): """ Verify get_address_info correctly omits metadata section when it's not found. """ chain_id = "1" address = "0x123abc" mock_base_url = "https://eth.blockscout.com" mock_blockscout_response = {"hash": address, "is_contract": False} mock_metadata_response = {"addresses": {}} with ( patch( "blockscout_mcp_server.tools.address.get_address_info.get_blockscout_base_url", new_callable=AsyncMock ) as mock_get_url, patch( "blockscout_mcp_server.tools.address.get_address_info.make_blockscout_request", new_callable=AsyncMock ) as mock_bs_request, patch( "blockscout_mcp_server.tools.address.get_address_info.make_metadata_request", new_callable=AsyncMock ) as mock_meta_request, ): mock_get_url.return_value = mock_base_url mock_bs_request.return_value = mock_blockscout_response mock_meta_request.return_value = mock_metadata_response result = await get_address_info(chain_id=chain_id, address=address, ctx=mock_ctx) assert isinstance(result, ToolResponse) assert isinstance(result.data, AddressInfoData) assert result.data.basic_info == mock_blockscout_response assert result.data.metadata is None assert result.notes is None assert result.instructions is not None and len(result.instructions) > 0 assert mock_ctx.report_progress.await_count == 4 assert mock_ctx.info.await_count == 4 @pytest.mark.asyncio async def test_get_address_info_metadata_failure(mock_ctx): """Return ToolResponse with notes when metadata API fails.""" chain_id = "1" address = "0x123abc" mock_base_url = "https://eth.blockscout.com" mock_blockscout_response = {"hash": address, "is_contract": False} metadata_error = httpx.RequestError("Network error") with ( patch( "blockscout_mcp_server.tools.address.get_address_info.get_blockscout_base_url", new_callable=AsyncMock, ) as mock_get_url, patch( "blockscout_mcp_server.tools.address.get_address_info.make_blockscout_request", new_callable=AsyncMock, ) as mock_bs_request, patch( "blockscout_mcp_server.tools.address.get_address_info.make_metadata_request", new_callable=AsyncMock, ) as mock_meta_request, ): mock_get_url.return_value = mock_base_url mock_bs_request.return_value = mock_blockscout_response mock_meta_request.side_effect = metadata_error result = await get_address_info(chain_id=chain_id, address=address, ctx=mock_ctx) assert isinstance(result, ToolResponse) assert isinstance(result.data, AddressInfoData) assert result.data.basic_info == mock_blockscout_response assert result.data.metadata is None assert result.notes is not None and len(result.notes) == 1 assert "Could not retrieve address metadata" in result.notes[0] assert mock_ctx.report_progress.await_count == 4 assert mock_ctx.info.await_count == 4 @pytest.mark.asyncio async def test_get_address_info_blockscout_failure(mock_ctx): """Ensure exception is raised when primary Blockscout call fails.""" chain_id = "1" address = "0x123abc" mock_base_url = "https://eth.blockscout.com" api_error = httpx.HTTPStatusError("Not Found", request=MagicMock(), response=MagicMock(status_code=404)) with ( patch( "blockscout_mcp_server.tools.address.get_address_info.get_blockscout_base_url", new_callable=AsyncMock, ) as mock_get_url, patch( "blockscout_mcp_server.tools.address.get_address_info.make_blockscout_request", new_callable=AsyncMock, ) as mock_bs_request, patch( "blockscout_mcp_server.tools.address.get_address_info.make_metadata_request", new_callable=AsyncMock, ) as mock_meta_request, ): mock_get_url.return_value = mock_base_url mock_bs_request.side_effect = api_error mock_meta_request.return_value = {} with pytest.raises(httpx.HTTPStatusError): await get_address_info(chain_id=chain_id, address=address, ctx=mock_ctx) mock_get_url.assert_called_once_with(chain_id) mock_bs_request.assert_called_once_with(base_url=mock_base_url, api_path=f"/api/v2/addresses/{address}") mock_meta_request.assert_called_once_with( api_path="/api/v1/metadata", params={"addresses": address, "chainId": chain_id} ) assert mock_ctx.report_progress.await_count == 2 assert mock_ctx.info.await_count == 2