for (Map.Entry<Node, RemoteTask> entry : tasks1.entrySet()) {
assertEquals(entry.getValue().getPartitionedSplitCount(), 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 with 5 splits. Since the new node does not have any splits, all 5 splits are assigned to the new node
SqlStageExecution sqlStageExecution2 = createSqlStageExecution(nodeScheduler, 5, 5);
Future future2 = sqlStageExecution2.start();