From ee16dd37ade518ec94dd7622e8240359a5f7d9bf Mon Sep 17 00:00:00 2001 From: John Blum Date: Fri, 10 Sep 2021 17:41:01 -0700 Subject: [PATCH] Add mock support for CacheServer groups configuration. --- .../data/gemfire/tests/mock/GemFireMockObjectsSupport.java | 6 ++++++ 1 file changed, 6 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 1aa99c4..30bce3f 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 @@ -1112,9 +1112,14 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { AtomicReference hostnameForClients = new AtomicReference<>(CacheServer.DEFAULT_HOSTNAME_FOR_CLIENTS); AtomicReference serverLoadProbe = new AtomicReference<>(null); + AtomicReference groups = new AtomicReference<>(CacheServer.DEFAULT_GROUPS); + doAnswer(newSetter(bindAddress, () -> null)) .when(mockCacheServer).setBindAddress(anyString()); + doAnswer(newSetter(groups, () -> null)) + .when(mockCacheServer).setGroups(any(String[].class)); + doAnswer(newSetter(hostnameForClients, () -> null)) .when(mockCacheServer).setHostnameForClients(anyString()); @@ -1154,6 +1159,7 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { when(mockCacheServer.getClientSession(any(DistributedMember.class))) .thenThrow(newUnsupportedOperationException(NOT_SUPPORTED)); when(mockCacheServer.getClientSession(anyString())).thenThrow(newUnsupportedOperationException(NOT_SUPPORTED)); + when(mockCacheServer.getGroups()).thenAnswer(newGetter(groups)); when(mockCacheServer.getHostnameForClients()).thenAnswer(newGetter(hostnameForClients)); when(mockCacheServer.getInterestRegistrationListeners()).thenReturn(Collections.emptySet()); when(mockCacheServer.getLoadPollInterval()).thenAnswer(newGetter(loadPollInterval));