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()