Spring/spring-test-data-geode
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add mocking support for o.a.g.cache.server.CacheServer.getLoadProbe() and o.a.g.cache.server.CacheServer.setLoadProbe(:ServerLoadProbe).

Browse Source
This commit is contained in:
John Blum
2018-09-19 12:03:20 -07:00
parent 4ea6c3921d
commit a233949add
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/mock/GemFireMockObjectsSupport.java
View File

@@ -116,6 +116,7 @@ import org.apache.geode.cache.query.QueryService;
import org.apache.geode.cache.query.QueryStatistics;
import org.apache.geode.cache.server.CacheServer;
import org.apache.geode.cache.server.ClientSubscriptionConfig;
import org.apache.geode.cache.server.ServerLoadProbe;
import org.apache.geode.cache.wan.GatewayEventFilter;
import org.apache.geode.cache.wan.GatewayEventSubstitutionFilter;
import org.apache.geode.cache.wan.GatewayReceiver;
@@ -836,6 +837,7 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport {
AtomicReference<String> bindAddress = new AtomicReference<>(CacheServer.DEFAULT_BIND_ADDRESS);
AtomicReference<String> hostnameForClients = new AtomicReference<>(CacheServer.DEFAULT_HOSTNAME_FOR_CLIENTS);
AtomicReference<ServerLoadProbe> serverLoadProbe = new AtomicReference<>(null);
doAnswer(newSetter(bindAddress, null))
.when(mockCacheServer).setBindAddress(anyString());
@@ -864,6 +866,9 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport {
doAnswer(newSetter(port, null))
.when(mockCacheServer).setPort(anyInt());
doAnswer(newSetter(serverLoadProbe, null))
.when(mockCacheServer).setLoadProbe(any(ServerLoadProbe.class));
doAnswer(newSetter(socketBufferSize, null))
.when(mockCacheServer).setSocketBufferSize(anyInt());
@@ -879,7 +884,7 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport {
when(mockCacheServer.getHostnameForClients()).thenAnswer(newGetter(hostnameForClients));
when(mockCacheServer.getInterestRegistrationListeners()).thenReturn(Collections.emptySet());
when(mockCacheServer.getLoadPollInterval()).thenAnswer(newGetter(loadPollInterval));
when(mockCacheServer.getLoadProbe()).thenThrow(newUnsupportedOperationException(NOT_SUPPORTED));
when(mockCacheServer.getLoadProbe()).thenAnswer(newGetter(serverLoadProbe));
when(mockCacheServer.getMaxConnections()).thenAnswer(newGetter(maxConnections));
when(mockCacheServer.getMaximumMessageCount()).thenAnswer(newGetter(maxMessageCount));
when(mockCacheServer.getMaximumTimeBetweenPings()).thenAnswer(newGetter(maxTimeBetweenPings));