chore: import upstream snapshot with attribution

python/ray/serve/tests/test_runtime_env_2.py

@@ -0,0 +1,93 @@
+import sys
+
+import pytest
+
+from ray._common.test_utils import run_string_as_driver
+
+
+@pytest.mark.skipif(sys.platform == "win32", reason="Fail to create temp dir.")
+def test_working_dir_deploy_new_version(ray_start, tmp_dir):
+    with open("hello", "w") as f:
+        f.write("world")
+
+    driver1 = """
+import ray
+from ray import serve
+
+job_config = ray.job_config.JobConfig(runtime_env={"working_dir": "."})
+ray.init(address="auto", namespace="serve", job_config=job_config)
+
+
+@serve.deployment
+class Test:
+    def __call__(self, *args):
+        return open("hello").read()
+
+Test = Test.options(_internal=True, version="1")
+handle = serve.run(Test.bind())
+assert handle.remote().result() == "world"
+"""
+
+    run_string_as_driver(driver1)
+
+    with open("hello", "w") as f:
+        f.write("world2")
+
+    driver2 = """
+import ray
+from ray import serve
+from ray.serve._private.constants import SERVE_DEFAULT_APP_NAME
+
+job_config = ray.job_config.JobConfig(runtime_env={"working_dir": "."})
+ray.init(address="auto", namespace="serve", job_config=job_config)
+
+
+@serve.deployment
+class Test:
+    def __call__(self, *args):
+        return open("hello").read()
+
+Test = Test.options(_internal=True, version="2")
+handle = serve.run(Test.bind())
+assert handle.remote().result() == "world2"
+serve.delete(SERVE_DEFAULT_APP_NAME)
+"""
+
+    run_string_as_driver(driver2)
+
+
+@pytest.mark.skipif(
+    sys.platform == "win32", reason="Runtime env unsupported on Windows"
+)
+def test_pip_no_working_dir(ray_start):
+    driver = """
+import ray
+from ray import serve
+import httpx
+
+ray.init(address="auto")
+
+
+@serve.deployment
+def httpx_version(request):
+    return httpx.__version__
+
+
+serve.run(httpx_version.options(
+    ray_actor_options={
+        "runtime_env": {
+            "pip": ["httpx==0.25.1"]
+        }
+    }).bind())
+
+assert httpx.get("http://127.0.0.1:8000/httpx_version").text == "0.25.1"
+"""
+
+    output = run_string_as_driver(driver)
+    print(output)
+
+
+if __name__ == "__main__":
+    import sys
+
+    sys.exit(pytest.main(["-sv", __file__]))