public Set<ContainerModel> group(Set<TaskModel> taskModels, GrouperMetadata grouperMetadata) { if (this.standbyTasksEnabled) { return generateStandbyTasks(this.taskNameGrouper.group(taskModels, grouperMetadata), replicationFactor); } else { return this.taskNameGrouper.group(taskModels, grouperMetadata); } }
public Set<ContainerModel> group(Set<TaskModel> taskModels, List<String> containersIds) { if (this.standbyTasksEnabled) { return generateStandbyTasks(this.taskNameGrouper.group(taskModels, containersIds), replicationFactor); } else { return this.taskNameGrouper.group(taskModels, containersIds); } }
@Test public void testBuddyContainerBasedGenerationIdentity() { this.standbyTaskGenerator = new TaskNameGrouperProxy(Mockito.mock(TaskNameGrouper.class), true, 2); Assert.assertEquals("Shouldnt add standby tasks to empty container map", Collections.emptySet(), this.standbyTaskGenerator.generateStandbyTasks(Collections.emptySet(), 1)); Assert.assertEquals("Shouldnt add standby tasks when repl factor = 1", getContainerMap(), this.standbyTaskGenerator.generateStandbyTasks(getContainerMap(), 1)); }
standbyTaskGenerator.generateStandbyTasks(initialContainerModels, replicationFactor);