for (Map.Entry<Node, RemoteTask> entry : tasks1.entrySet()) {
assertEquals(entry.getValue().getQueuedSplits(), 5);
}
// Add new node
Node additionalNode = new PrestoNode("other4", URI.create("http://127.0.0.1:14"), NodeVersion.UNKNOWN);
nodeManager.addNode("foo", additionalNode);
// Schedule next query
SqlStageExecution sqlStageExecution2 = createSqlStageExecution(nodeScheduler, 2, 20);
Future future2 = sqlStageExecution2.start();