@Test
public void testOnlyFenzoPartitionIsIncluded() {

    AgentInstanceGroup instanceGroup = createRemovableInstanceGroup("instanceGroup1", 0);
    when(agentManagementService.getInstanceGroups()).thenReturn(singletonList(instanceGroup));
    when(agentManagementService.isOwnedByFenzo(any(AgentInstanceGroup.class))).thenReturn(false);
    AgentInstance agentInstance1 = AgentInstance.newBuilder()
            .withId("agentInstance1")
            .withInstanceGroupId("instanceGroup1")
            .build();
    when(agentManagementService.getAgentInstances("instanceGroup1")).thenReturn(asList(agentInstance1));
    ClusterRemovableInstanceGroupAgentRemover clusterRemovableInstanceGroupAgentRemover = new ClusterRemovableInstanceGroupAgentRemover(titusRuntime, configuration,
            agentManagementService, v3JobOperations, testScheduler);
    clusterRemovableInstanceGroupAgentRemover.doRemoveAgents().await();
    verify(agentManagementService, times(0)).terminateAgents("instanceGroup1", singletonList("agentInstance1"), true);
}