TestSSRFProtectedUrlRelaxedType Class — langchain Architecture

Source Code

libs/core/tests/unit_tests/test_ssrf_protection.py lines 252–271

class TestSSRFProtectedUrlRelaxedType:
    """Tests for SSRFProtectedUrlRelaxed Pydantic type."""

    def test_localhost_accepted(self) -> None:
        """Test that localhost URLs are accepted by relaxed schema."""

        class WebhookSchema(BaseModel):
            url: SSRFProtectedUrlRelaxed

        schema = WebhookSchema(url="http://localhost:8080")
        assert str(schema.url).startswith("http://localhost")

    def test_cloud_metadata_still_rejected(self) -> None:
        """Test that cloud metadata is still rejected by relaxed schema."""

        class WebhookSchema(BaseModel):
            url: SSRFProtectedUrlRelaxed

        with pytest.raises(ValidationError):
            WebhookSchema(url="http://169.254.169.254/latest/meta-data/")