private void setUp(Supplier<List<Driver>> driversSupplier)
{
checkState(localQueryRunner == null, "Already set up");
Session session = testSessionBuilder()
.setCatalog("tpch")
.setSchema("tiny")
.setSystemProperty("task_default_concurrency", "1")
.build();
localQueryRunner = queryRunnerWithInitialTransaction(session);
localQueryRunner.createCatalog("tpch", new TpchConnectorFactory(1), ImmutableMap.of());
userPool = new MemoryPool(new MemoryPoolId("test"), TEN_MEGABYTES);
fakeQueryId = new QueryId("fake");
SpillSpaceTracker spillSpaceTracker = new SpillSpaceTracker(new DataSize(1, GIGABYTE));
DefaultQueryContext queryContext = new DefaultQueryContext(new QueryId("query"),
TEN_MEGABYTES,
new DataSize(20, MEGABYTE),
userPool,
new TestingGcMonitor(),
localQueryRunner.getExecutor(),
localQueryRunner.getScheduler(),
TEN_MEGABYTES,
spillSpaceTracker);
taskContext = createTaskContext(queryContext, localQueryRunner.getExecutor(), session);
drivers = driversSupplier.get();
}