Function query_yields

datafusion/core/tests/execution/coop.rs:822–835

(
    plan: Arc<dyn ExecutionPlan>,
    task_ctx: Arc<TaskContext>,
)

}

async fn query_yields(
    plan: Arc<dyn ExecutionPlan>,
    task_ctx: Arc<TaskContext>,
) -> Result<(), Box<dyn Error>> {
    // Run plan through EnsureCooperative
    let optimized =
        EnsureCooperative::new().optimize(plan, task_ctx.session_config().options())?;

    // Get the stream
    let stream = physical_plan::execute_stream(optimized, task_ctx)?;

    // Spawn a task that tries to poll the stream and check whether given stream yields
    stream_yields(stream).await
}