From 4c86e7c93e7ef8ff589f8ce2f71508dd65803c9b Mon Sep 17 00:00:00 2001 From: John Blum Date: Wed, 8 Sep 2021 13:38:05 -0700 Subject: [PATCH] Add mock configuration for the ClientCacheFactory's (DEFAULT Pool) and Pool's serverConnectionTimeout property. --- .../gemfire/tests/mock/GemFireMockObjectsSupport.java | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/mock/GemFireMockObjectsSupport.java b/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/mock/GemFireMockObjectsSupport.java index 007b7ea..10e0f41 100644 --- a/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/mock/GemFireMockObjectsSupport.java +++ b/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/mock/GemFireMockObjectsSupport.java @@ -1771,6 +1771,7 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { AtomicInteger minConnections = new AtomicInteger(PoolFactory.DEFAULT_MIN_CONNECTIONS); AtomicInteger readTimeout = new AtomicInteger(PoolFactory.DEFAULT_READ_TIMEOUT); AtomicInteger retryAttempts = new AtomicInteger(PoolFactory.DEFAULT_RETRY_ATTEMPTS); + AtomicInteger serverConnectionTimeout = new AtomicInteger(PoolFactory.DEFAULT_SERVER_CONNECTION_TIMEOUT); AtomicInteger socketBufferSize = new AtomicInteger(PoolFactory.DEFAULT_SOCKET_BUFFER_SIZE); AtomicInteger socketConnectTimeout = new AtomicInteger(PoolFactory.DEFAULT_SOCKET_CONNECT_TIMEOUT); AtomicInteger statisticInterval = new AtomicInteger(PoolFactory.DEFAULT_STATISTIC_INTERVAL); @@ -1826,6 +1827,9 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { when(mockPoolFactory.setRetryAttempts(anyInt())) .thenAnswer(newSetter(retryAttempts, mockPoolFactory)); + when(mockPoolFactory.setServerConnectionTimeout(anyInt())) + .thenAnswer(newSetter(serverConnectionTimeout, mockPoolFactory)); + when(mockPoolFactory.setServerGroup(anyString())) .thenAnswer(newSetter(serverGroup, () -> mockPoolFactory)); @@ -1884,6 +1888,7 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { when(mockPool.getPRSingleHopEnabled()).thenReturn(prSingleHopEnabled.get()); when(mockPool.getReadTimeout()).thenReturn(readTimeout.get()); when(mockPool.getRetryAttempts()).thenReturn(retryAttempts.get()); + when(mockPool.getServerConnectionTimeout()).thenReturn(serverConnectionTimeout.get()); when(mockPool.getServerGroup()).thenReturn(serverGroup.get()); when(mockPool.getServers()).thenReturn(servers); when(mockPool.getSocketBufferSize()).thenReturn(socketBufferSize.get()); @@ -3432,6 +3437,11 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { return clientCacheFactorySpy; }).when(clientCacheFactorySpy).setPoolRetryAttempts(anyInt()); + doAnswer(invocation -> { + mockPoolFactory.setServerConnectionTimeout(invocation.getArgument(0)); + return clientCacheFactorySpy; + }).when(clientCacheFactorySpy).setPoolServerConnectionTimeout(anyInt()); + doAnswer(invocation -> { mockPoolFactory.setServerGroup(invocation.getArgument(0)); return clientCacheFactorySpy;