public SqlTask createInitialTask()
{
TaskId taskId = new TaskId("query", 0, nextTaskId.incrementAndGet());
URI location = URI.create("fake://task/" + taskId);
DefaultQueryContext queryContext = new DefaultQueryContext(new QueryId("query"),
new DataSize(1, MEGABYTE),
new DataSize(2, MEGABYTE),
new MemoryPool(new MemoryPoolId("test"), new DataSize(1, GIGABYTE)),
new TestingGcMonitor(),
taskNotificationExecutor,
driverYieldExecutor,
new DataSize(1, MEGABYTE),
new SpillSpaceTracker(new DataSize(1, GIGABYTE)));
queryContext.addTaskContext(new TaskStateMachine(taskId, taskNotificationExecutor), testSessionBuilder().build(), false, false, OptionalInt.empty());
return createSqlTask(
taskId,
location,
"fake",
queryContext,
sqlTaskExecutionFactory,
taskNotificationExecutor,
Functions.identity(),
new DataSize(32, MEGABYTE),
new CounterStat());
}
}