private CassandraClientPoolImpl createClientPool() { return CassandraClientPoolImpl.createImplForTest( MetricsManagers.createForTests(), config, CassandraClientPoolImpl.StartupChecks.DO_NOT_RUN, deterministicExecutor, blacklist, cassandra); }
public static CassandraKeyValueService createForTesting( CassandraKeyValueServiceConfig config, Optional<LeaderConfig> leaderConfig) { MetricsManager metricsManager = MetricsManagers.createForTests(); CassandraClientPool clientPool = CassandraClientPoolImpl.createImplForTest(metricsManager, config, CassandraClientPoolImpl.StartupChecks.RUN, new Blacklist(config)); return createOrShutdownClientPool(metricsManager, config, clientPool, leaderConfig, CassandraMutationTimestampProviders.legacyModeForTestsOnly(), LoggerFactory.getLogger(CassandraKeyValueService.class), AtlasDbConstants.DEFAULT_INITIALIZE_ASYNC); }
@SuppressWarnings("OptionalUsedAsFieldOrParameterType") // Unpacking it seems less readable private CassandraClientPoolImpl clientPoolWith( ImmutableSet<InetSocketAddress> servers, ImmutableSet<InetSocketAddress> serversInPool, Optional<Exception> failureMode) { when(config.servers()).thenReturn(servers); CassandraClientPoolImpl cassandraClientPool = CassandraClientPoolImpl.createImplForTest( MetricsManagers.of(metricRegistry, taggedMetricRegistry), config, CassandraClientPoolImpl.StartupChecks.DO_NOT_RUN, blacklist); serversInPool.forEach(address -> cassandraClientPool.getCurrentPools() .put(address, getMockPoolingContainerForHost(address, failureMode))); return cassandraClientPool; }
@Test public void attemptsShouldBeCountedPerHost() { CassandraClientPoolImpl cassandraClientPool = CassandraClientPoolImpl.createImplForTest( MetricsManagers.of(metricRegistry, taggedMetricRegistry), config, CassandraClientPoolImpl.StartupChecks.DO_NOT_RUN, blacklist); host(HOST_1).throwsException(new SocketTimeoutException()) .throwsException(new InvalidRequestException()) .inPool(cassandraClientPool); host(HOST_2).throwsException(new SocketTimeoutException()) .inPool(cassandraClientPool); runNoopWithRetryOnHost(HOST_1, cassandraClientPool); assertThat(blacklist.contains(HOST_2), is(false)); }
public static CassandraKeyValueService createForTesting( CassandraKeyValueServiceConfig config, Optional<LeaderConfig> leaderConfig) { MetricsManager metricsManager = MetricsManagers.createForTests(); CassandraClientPool clientPool = CassandraClientPoolImpl.createImplForTest(metricsManager, config, CassandraClientPoolImpl.StartupChecks.RUN, new Blacklist(config)); return createOrShutdownClientPool(metricsManager, config, clientPool, leaderConfig, CassandraMutationTimestampProviders.legacyModeForTestsOnly(), LoggerFactory.getLogger(CassandraKeyValueService.class), AtlasDbConstants.DEFAULT_INITIALIZE_ASYNC); }