chore: import upstream snapshot with attribution

python/ray/tests/test_protobuf_compatibility.py

@@ -0,0 +1,33 @@
+import sys
+
+import pytest
+
+import ray
+
+
+@pytest.mark.skipif(
+    sys.platform == "win32",
+    reason="Failing on Windows, affected by protocolbuffers/protobuf#10075.",
+)
+def test_protobuf_compatibility(shutdown_only):
+    protobuf_4_21_0 = {"pip": ["protobuf==4.21.0"]}
+    protobuf_3_20_3 = {"pip": ["protobuf==3.20.3"]}
+
+    ray.init()
+
+    @ray.remote
+    def load_ray():
+        import google.protobuf
+
+        # verfiy this no longer crashes. see
+        # https://github.com/ray-project/ray/issues/25282
+        import ray  # noqa
+
+        return google.protobuf.__version__
+
+    assert "4.21.0" == ray.get(load_ray.options(runtime_env=protobuf_4_21_0).remote())
+    assert "3.20.3" == ray.get(load_ray.options(runtime_env=protobuf_3_20_3).remote())
+
+
+if __name__ == "__main__":
+    sys.exit(pytest.main(["-sv", __file__]))