chore: import upstream snapshot with attribution

python/ray/tests/test_client_metadata.py

@@ -0,0 +1,53 @@
+import sys
+
+import pytest
+
+from ray._raylet import NodeID
+from ray.runtime_context import RuntimeContext
+from ray.util.client.ray_client_helpers import ray_start_client_server
+
+
+def test_get_ray_metadata(ray_start_regular_shared):
+    """Test the ClusterInfo client data pathway and API surface"""
+    with ray_start_client_server() as ray:
+        ip_address = ray_start_regular_shared["node_ip_address"]
+
+        initialized = ray.is_initialized()
+        assert initialized
+
+        nodes = ray.nodes()
+        assert len(nodes) == 1, nodes
+        assert nodes[0]["NodeManagerAddress"] == ip_address
+
+        current_node_id = "node:" + ip_address
+
+        cluster_resources = ray.cluster_resources()
+        available_resources = ray.available_resources()
+
+        assert cluster_resources["CPU"] == 1.0
+        assert current_node_id in cluster_resources
+        assert current_node_id in available_resources
+
+
+def test_get_runtime_context(ray_start_regular_shared):
+    """Test the get_runtime_context data through the metadata API"""
+    with ray_start_client_server() as ray:
+        rtc = ray.get_runtime_context()
+        assert isinstance(rtc, RuntimeContext)
+        assert isinstance(rtc.node_id, NodeID)
+        assert len(rtc.node_id.hex()) == 56
+        assert isinstance(rtc.namespace, str)
+
+        wid = ray.get_runtime_context().get_worker_id()
+        assert isinstance(wid, str)
+        assert len(wid) == 56
+
+        # Ensure this doesn't throw
+        ray.get_runtime_context().get()
+
+        with pytest.raises(Exception):
+            _ = rtc.task_id
+
+
+if __name__ == "__main__":
+    sys.exit(pytest.main(["-sv", __file__]))