@Test(groups = "unit") public void should_filter_init_hosts_with_predicate() { Predicate<Host> predicate = Predicates.in(Lists.newArrayList(host1, host2)); HostFilterPolicy policy = new HostFilterPolicy(wrappedPolicy, predicate); policy.init(cluster, Lists.newArrayList(host1, host2, host3)); verify(wrappedPolicy).init(eq(cluster), hostsCaptor.capture()); assertThat(hostsCaptor.getValue()).containsOnly(host1, host2); }
@Test(groups = "unit", expectedExceptions = IllegalArgumentException.class) public void should_throw_if_predicate_filters_out_all_init_hosts() { Predicate<Host> predicate = Predicates.alwaysFalse(); HostFilterPolicy policy = new HostFilterPolicy(wrappedPolicy, predicate); policy.init(cluster, Lists.newArrayList(host1, host2, host3)); }
@Test(groups = "unit") public void should_filter_init_hosts_with_predicate() { Predicate<Host> predicate = Predicates.in(Lists.newArrayList(host1, host2)); HostFilterPolicy policy = new HostFilterPolicy(wrappedPolicy, predicate); policy.init(cluster, Lists.newArrayList(host1, host2, host3)); verify(wrappedPolicy).init(eq(cluster), hostsCaptor.capture()); assertThat(hostsCaptor.getValue()).containsOnly(host1, host2); }
@Test(groups = "unit", expectedExceptions = IllegalArgumentException.class) public void should_throw_if_predicate_filters_out_all_init_hosts() { Predicate<Host> predicate = Predicates.alwaysFalse(); HostFilterPolicy policy = new HostFilterPolicy(wrappedPolicy, predicate); policy.init(cluster, Lists.newArrayList(host1, host2, host3)); }