def test_publish_and_subscribe_function_keys(ray_start_regular):
address_info = ray_start_regular
gcs_server_addr = address_info["gcs_address"]
subscriber = GcsFunctionKeySubscriber(address=gcs_server_addr)
subscriber.subscribe()
publisher = GcsPublisher(address=gcs_server_addr)
publisher.publish_function_key(b"111")
publisher.publish_function_key(b"222")
assert subscriber.poll() == b"111"
assert subscriber.poll() == b"222"
subscriber.close()
def test_publish_and_subscribe_function_keys(ray_start_regular):
address_info = ray_start_regular
redis = ray._private.services.create_redis_client(
address_info["redis_address"],
password=ray.ray_constants.REDIS_DEFAULT_PASSWORD)
gcs_server_addr = gcs_utils.get_gcs_address_from_redis(redis)
subscriber = GcsFunctionKeySubscriber(address=gcs_server_addr)
subscriber.subscribe()
publisher = GcsPublisher(address=gcs_server_addr)
publisher.publish_function_key(b"111")
publisher.publish_function_key(b"222")
assert subscriber.poll() == b"111"
assert subscriber.poll() == b"222"
subscriber.close()
