Fix tests

Author: robertnishihara

doc/source/conf.py

@@ -42,6 +42,7 @@
     "ray.core.generated.ClientTableData",
     "ray.core.generated.GcsTableEntry",
     "ray.core.generated.HeartbeatTableData",
+    "ray.core.generated.HeartbeatBatchTableData",
     "ray.core.generated.DriverTableData",
     "ray.core.generated.ErrorTableData",
     "ray.core.generated.ProfileTableData",

test/failure_test.py

@@ -500,10 +500,10 @@ def test_warning_monitor_died(shutdown_only):
     # addition to the monitor.
     fake_id = 20 * b"\x00"
     malformed_message = "asdf"
-    redis_client = ray.worker.global_state.redis_clients[0]
+    redis_client = ray.worker.global_worker.redis_client
     redis_client.execute_command(
-        "RAY.TABLE_ADD", ray.gcs_utils.TablePrefix.HEARTBEAT,
-        ray.gcs_utils.TablePubsub.HEARTBEAT, fake_id, malformed_message)
+        "RAY.TABLE_ADD", ray.gcs_utils.TablePrefix.HEARTBEAT_BATCH,
+        ray.gcs_utils.TablePubsub.HEARTBEAT_BATCH, fake_id, malformed_message)
 
     wait_for_errors(ray_constants.MONITOR_DIED_ERROR, 1)