From 6889b3c696efd465146297e5efd43e081852423e Mon Sep 17 00:00:00 2001 From: John Blum Date: Wed, 8 Sep 2021 14:13:00 -0700 Subject: [PATCH] Add mock configuration for the ClientCacheFactory's (DEFAULT Pool) and Pool's socketFactory property. --- .../gemfire/tests/mock/GemFireMockObjectsSupport.java | 11 +++++++++++ 1 file changed, 11 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 10e0f41..15ffbac 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 @@ -115,6 +115,7 @@ import org.apache.geode.cache.client.ClientRegionShortcut; import org.apache.geode.cache.client.Pool; import org.apache.geode.cache.client.PoolFactory; import org.apache.geode.cache.client.PoolManager; +import org.apache.geode.cache.client.SocketFactory; import org.apache.geode.cache.control.ResourceManager; import org.apache.geode.cache.execute.RegionFunctionContext; import org.apache.geode.cache.lucene.LuceneIndex; @@ -1782,6 +1783,7 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { AtomicLong idleTimeout = new AtomicLong(PoolFactory.DEFAULT_IDLE_TIMEOUT); AtomicLong pingInterval = new AtomicLong(PoolFactory.DEFAULT_PING_INTERVAL); + AtomicReference socketFactory = new AtomicReference<>(PoolFactory.DEFAULT_SOCKET_FACTORY); AtomicReference serverGroup = new AtomicReference<>(PoolFactory.DEFAULT_SERVER_GROUP); List locators = new ArrayList<>(); @@ -1839,6 +1841,9 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { when(mockPoolFactory.setSocketConnectTimeout(anyInt())) .thenAnswer(newSetter(socketConnectTimeout, mockPoolFactory)); + when(mockPoolFactory.setSocketFactory(any(SocketFactory.class))) + .thenAnswer(newSetter(socketFactory, () -> mockPoolFactory)); + when(mockPoolFactory.setStatisticInterval(anyInt())) .thenAnswer(newSetter(statisticInterval, mockPoolFactory)); @@ -1893,6 +1898,7 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { when(mockPool.getServers()).thenReturn(servers); when(mockPool.getSocketBufferSize()).thenReturn(socketBufferSize.get()); when(mockPool.getSocketConnectTimeout()).thenReturn(socketConnectTimeout.get()); + when(mockPool.getSocketFactory()).thenReturn(socketFactory.get()); when(mockPool.getStatisticInterval()).thenReturn(statisticInterval.get()); when(mockPool.getSubscriptionAckInterval()).thenReturn(subscriptionAckInterval.get()); when(mockPool.getSubscriptionEnabled()).thenReturn(subscriptionEnabled.get()); @@ -3457,6 +3463,11 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { return clientCacheFactorySpy; }).when(clientCacheFactorySpy).setPoolSocketConnectTimeout(anyInt()); + doAnswer(invocation -> { + mockPoolFactory.setSocketFactory(invocation.getArgument(0)); + return clientCacheFactorySpy; + }).when(clientCacheFactorySpy).setPoolSocketFactory(any(SocketFactory.class)); + doAnswer(invocation -> { mockPoolFactory.setStatisticInterval(invocation.getArgument(0)); return clientCacheFactorySpy;