@Test
public void testOnlyFenzoPartitionIsIncluded() {

    List<AgentInstanceGroup> instanceGroups = asList(
            getInstanceGroup(Tier.Critical, R4_8XLARGE_ID, 0, 1).toBuilder().withId("fenzo").build(),
            getInstanceGroup(Tier.Critical, R4_8XLARGE_ID, 0, 1).toBuilder().withId("kubeScheduler").build()
    );
    when(agentManagementService.getInstanceGroups()).thenReturn(instanceGroups);
    when(agentManagementService.isOwnedByFenzo(any(AgentInstanceGroup.class))).thenAnswer(invocation -> {
        AgentInstanceGroup instanceGroup = invocation.getArgument(0);
        return instanceGroup.getId().equals("fenzo");
    });
    ApplicationSLA bigSLA = ApplicationSlaSample.CriticalLarge.builder().withInstanceCount(100).build();
    CapacityRequirements requirements = new CapacityRequirements(singletonMap(Tier.Critical, singletonList(bigSLA)));
    CapacityAllocations allocations = strategy.compute(requirements);
    assertThat(allocations.getInstanceGroups()).hasSize(1);
    assertThat(CollectionsExt.first(allocations.getInstanceGroups()).getId()).isEqualTo("fenzo");
}