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__]))