test_mermaid_base_url_custom() — langchain Function Reference

Architecture documentation for the test_mermaid_base_url_custom() function in test_graph.py from the langchain codebase.

Function python LangChainCore Runnables

Entity Profile

LangChainCore→ Runnables→ test_mermaid_base_url_custom() — langchain Function Reference

Dependency Diagram

graph TD
  cbb8f6d7_81f7_6a3f_39f9_a5cfca2af404["test_mermaid_base_url_custom()"]
  83e2cebc_592b_5a1b_c581_48dfbd3d53cf["test_graph.py"]
  cbb8f6d7_81f7_6a3f_39f9_a5cfca2af404 -->|defined in| 83e2cebc_592b_5a1b_c581_48dfbd3d53cf
  style cbb8f6d7_81f7_6a3f_39f9_a5cfca2af404 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

libs/core/tests/unit_tests/runnables/test_graph.py lines 592–610

def test_mermaid_base_url_custom() -> None:
    """Test that _render_mermaid_using_api uses custom base_url when provided."""
    custom_url = "https://custom.mermaid.com"
    mock_response = MagicMock()
    mock_response.status_code = 200
    mock_response.content = b"fake image data"

    with patch("requests.get", return_value=mock_response) as mock_get:
        # Call the function with custom base_url.
        _render_mermaid_using_api(
            "graph TD;\n    A --> B;",
            base_url=custom_url,
        )

        # Verify that the URL was constructed with our custom base URL
        assert mock_get.called
        args = mock_get.call_args[0]
        url = args[0]  # First argument to request.get is the URL
        assert url.startswith(custom_url)