Method runQueryWithWriterConfigs

bolt/exec/tests/TableWriteTest.cpp:367–398

  }

  RowVectorPtr runQueryWithWriterConfigs(
      const core::PlanNodePtr& plan,
      bool spillEnabled = false) {
    if (!spillEnabled) {
      return AssertQueryBuilder(plan, duckDbQueryRunner_)
          .maxDrivers(
              2 *
              std::max(kNumTableWriterCount, kNumPartitionedTableWriterCount))
          .config(
              QueryConfig::kTaskWriterCount,
              std::to_string(numTableWriterCount_))
          .config(
              QueryConfig::kTaskPartitionedWriterCount,
              std::to_string(numPartitionedTableWriterCount_))
          .copyResults(pool());
    }

    const auto spillDirectory = exec::test::TempDirectoryPath::create();
    return AssertQueryBuilder(plan, duckDbQueryRunner_)
        .spillDirectory(spillDirectory->path)
        .maxDrivers(
            2 * std::max(kNumTableWriterCount, kNumPartitionedTableWriterCount))
        .config(
            QueryConfig::kTaskWriterCount, std::to_string(numTableWriterCount_))
        .config(
            QueryConfig::kTaskPartitionedWriterCount,
            std::to_string(numPartitionedTableWriterCount_))
        .config(core::QueryConfig::kSpillEnabled, "true")
        .config(QueryConfig::kWriterSpillEnabled, "true")
        .config(QueryConfig::kTestingSpillPct, "100")
        .copyResults(pool());
  }

  void setCommitStrategy(CommitStrategy commitStrategy) {
    commitStrategy_ = commitStrategy;