/** @see <a href="https://jira.spring.io/browse/DATACASS-317">DATACASS-317</a> */
@Test
public void shouldSetClusterNameWithClusterNameProperty() throws Exception {
final Cluster.Builder mockClusterBuilder = mock(Cluster.Builder.class);
when(mockClusterBuilder.addContactPoints(Matchers.<String[]>anyVararg()))
.thenReturn(mockClusterBuilder);
CassandraCqlClusterFactoryBean bean =
new CassandraCqlClusterFactoryBean() {
@Override
Cluster.Builder newClusterBuilder() {
return mockClusterBuilder;
}
};
bean.setBeanName("ABC");
bean.setClusterName("XYZ");
bean.afterPropertiesSet();
verify(mockClusterBuilder, times(1)).withClusterName(eq("XYZ"));
}
@Override
protected Cluster.Builder configure(Cluster.Builder builder) {
return builder.withRetryPolicy(
new RetryPolicy() {
@Override
public RetryDecision onReadTimeout(
Statement statement,
ConsistencyLevel cl,
int requiredResponses,
int receivedResponses,
boolean dataRetrieved,
int nbRetry) {
return retryDecision;
}
@Override
public RetryDecision onWriteTimeout(
Statement statement,
ConsistencyLevel cl,
WriteType writeType,
int requiredAcks,
int receivedAcks,
int nbRetry) {
return retryDecision;
}
@Override
public RetryDecision onUnavailable(
Statement statement,
ConsistencyLevel cl,
int requiredReplica,
int aliveReplica,
int nbRetry) {
return retryDecision;
}
});
}
