{
TaskInfo taskInfo = sqlTaskManager.updateTask(session,
taskId,
testFragment,
ImmutableList.<TaskSource>of(new TaskSource(tableScanNodeId, ImmutableSet.of(new ScheduledSplit(0, split)), true)),
INITIAL_EMPTY_OUTPUT_BUFFERS.withBuffer("out", new UnpartitionedPagePartitionFunction()).withNoMoreBufferIds());
assertEquals(taskInfo.getState(), TaskState.RUNNING);
taskInfo = sqlTaskManager.getTaskInfo(taskInfo.getTaskId(), false);
assertEquals(taskInfo.getState(), TaskState.RUNNING);