/** @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; } }); }