@Test
public void testRemoveLoadBalancer() throws Exception {

    Map<JobLoadBalancer, JobLoadBalancer.State> testData = generateTestData(10, 20, 1).getAssociations();
    CassandraLoadBalancerStore store = getInitdStore();
    testData.forEach((jobLoadBalancer, state) ->
            assertThat(store.addOrUpdateLoadBalancer(jobLoadBalancer, state).await(TIMEOUT_MS, TimeUnit.MILLISECONDS)).isTrue()
    );
    testData.forEach((jobLoadBalancer, state) ->
            assertThat(store.removeLoadBalancer(jobLoadBalancer).await(TIMEOUT_MS, TimeUnit.MILLISECONDS)).isTrue()
    );
    Map<String, List<JobLoadBalancerState>> byJobId = store.getAssociations().stream()
            .collect(Collectors.groupingBy(JobLoadBalancerState::getJobId));
    testData.forEach((jobLoadBalancer, state) -> {
        assertThat(byJobId.get(jobLoadBalancer.getJobId())).isNullOrEmpty();
    });
}