def test_usage_lib_cluster_metadata_generation( monkeypatch, ray_start_cluster, reset_lib_usage ): with monkeypatch.context() as m: m.setenv("RAY_USAGE_STATS_ENABLED", "1") m.setenv("RAY_USAGE_STATS_REPORT_URL", "http://127.0.0.1:8000") cluster = ray_start_cluster cluster.add_node(num_cpus=0) ray.init(address=cluster.address) """ Test metadata stored is equivalent to `_generate_cluster_metadata`. """ meta = ray_usage_lib._generate_cluster_metadata() cluster_metadata = ray_usage_lib.get_cluster_metadata( ray.experimental.internal_kv.internal_kv_get_gcs_client(), num_retries=20 ) # Remove fields that are dynamically changed. assert meta.pop("session_id") assert meta.pop("session_start_timestamp_ms") assert cluster_metadata.pop("session_id") assert cluster_metadata.pop("session_start_timestamp_ms") assert meta == cluster_metadata """ Make sure put & get works properly. """ cluster_metadata = ray_usage_lib.put_cluster_metadata( ray.experimental.internal_kv.internal_kv_get_gcs_client(), num_retries=20 ) assert cluster_metadata == ray_usage_lib.get_cluster_metadata( ray.experimental.internal_kv.internal_kv_get_gcs_client(), num_retries=20 )
def test_usage_lib_cluster_metadata_generation(monkeypatch, shutdown_only): with monkeypatch.context() as m: m.setenv("RAY_USAGE_STATS_ENABLED", "1") ray.init(num_cpus=0) """ Test metadata stored is equivalent to `_generate_cluster_metadata`. """ meta = ray_usage_lib._generate_cluster_metadata() cluster_metadata = ray_usage_lib.get_cluster_metadata( ray.experimental.internal_kv.internal_kv_get_gcs_client(), num_retries=20) # Remove fields that are dynamically changed. assert meta.pop("session_id") assert meta.pop("session_start_timestamp_ms") assert cluster_metadata.pop("session_id") assert cluster_metadata.pop("session_start_timestamp_ms") assert meta == cluster_metadata """ Make sure put & get works properly. """ cluster_metadata = ray_usage_lib.put_cluster_metadata( ray.experimental.internal_kv.internal_kv_get_gcs_client(), num_retries=20) assert cluster_metadata == ray_usage_lib.get_cluster_metadata( ray.experimental.internal_kv.internal_kv_get_gcs_client(), num_retries=20)
def test_usage_lib_cluster_metadata_generation_usage_disabled(shutdown_only): """ Make sure only version information is generated when usage stats are not enabled. """ meta = ray_usage_lib._generate_cluster_metadata() assert "ray_version" in meta assert "python_version" in meta assert len(meta) == 2
def test_usage_lib_cluster_metadata_generation_usage_disabled( monkeypatch, shutdown_only): """ Make sure only version information is generated when usage stats are not enabled. """ with monkeypatch.context() as m: m.setenv("RAY_USAGE_STATS_ENABLED", "0") meta = ray_usage_lib._generate_cluster_metadata() assert "ray_version" in meta assert "python_version" in meta assert len(meta) == 2
def test_usage_lib_cluster_metadata_generation(shutdown_only): ray.init(num_cpus=0) """ Test metadata stored is equivalent to `_generate_cluster_metadata`. """ meta = ray_usage_lib._generate_cluster_metadata() cluster_metadata = ray_usage_lib.get_cluster_metadata( ray.experimental.internal_kv.internal_kv_get_gcs_client(), num_retries=20) # Session id is random. meta.pop("session_id") cluster_metadata.pop("session_id") assert meta == cluster_metadata """ Make sure put & get works properly. """ cluster_metadata = ray_usage_lib.put_cluster_metadata( ray.experimental.internal_kv.internal_kv_get_gcs_client(), num_retries=20) assert cluster_metadata == ray_usage_lib.get_cluster_metadata( ray.experimental.internal_kv.internal_kv_get_gcs_client(), num_retries=20)