"""Test that runtime_env raises good errors if ray[default] is not installed.


In CI, this file is run with a minimal ray installation (`pip install ray`.)

To run this test file locally, remove some dependency that appears in
ray[default] but not in ray (e.g., `pip uninstall aiohttp`) and set
`export RAY_MINIMAL=1`.

"""

import os
import sys

import pytest

import ray
from ray.exceptions import RuntimeEnvSetupError


def _test_task_and_actor():
    @ray.remote
    def f():
        return 1

    with pytest.raises(RuntimeEnvSetupError, match="install virtualenv"):
        ray.get(f.options(runtime_env={"pip": ["requests"]}).remote())

    @ray.remote
    class A:
        def task(self):
            return 1

    a = A.options(runtime_env={"pip": ["requests"]}).remote()

    with pytest.raises(RuntimeEnvSetupError, match="install virtualenv"):
        ray.get(a.task.remote())


@pytest.mark.skipif(
    sys.platform == "win32", reason="runtime_env unsupported on Windows."
)
@pytest.mark.skipif(
    os.environ.get("RAY_MINIMAL") != "1",
    reason="This test is only run in CI with a minimal Ray installation.",
)
def test_task_actor(shutdown_only):
    ray.init()
    _test_task_and_actor()


@pytest.mark.skipif(
    sys.platform == "win32", reason="runtime_env unsupported on Windows."
)
@pytest.mark.skipif(
    os.environ.get("RAY_MINIMAL") != "1",
    reason="This test is only run in CI with a minimal Ray installation.",
)
def test_ray_init(shutdown_only):
    ray.init(runtime_env={"pip": ["requests"]})

    @ray.remote
    def f():
        return 1

    with pytest.raises(RuntimeEnvSetupError, match="install virtualenv"):
        ray.get(f.remote())


if __name__ == "__main__":
    sys.exit(pytest.main(["-sv", __file__]))