for (int i = 0; i < splitCount; i++) {
splits.add(new DualSplit(HostAddress.fromString("127.0.0.1")));
}
SplitSource splitSource = new ConnectorAwareSplitSource(DualConnector.CONNECTOR_ID, new FixedSplitSource(null, splits.build()));
return new StageExecutionPlan(testFragment,
Optional.of(splitSource),
ImmutableList.<StageExecutionPlan>of()
);
}