Function waitForTaskStateChange

bolt/exec/tests/utils/QueryAssertions.cpp:1471–1485

}

bool waitForTaskStateChange(
    exec::Task* task,
    TaskState state,
    uint64_t maxWaitMicros) {
  // Wait for task to transition to finished state.
  if (task->state() != state) {
    auto& executor = folly::QueuedImmediateExecutor::instance();
    auto future = task->taskCompletionFuture()
                      .within(std::chrono::microseconds(maxWaitMicros))
                      .via(&executor);
    future.wait();
  }

  return task->state() == state;
}

void waitForAllTasksToBeDeleted(uint64_t maxWaitUs) {
  const uint64_t numCreatedTasks = Task::numCreatedTasks();