From 933721a9691ab32aa9a2b79fac837206d3ef7c27 Mon Sep 17 00:00:00 2001 From: John Blum Date: Mon, 13 Sep 2021 12:09:01 -0700 Subject: [PATCH] Mock the Pool.getQueryService() method. --- .../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 25b9a9b..29d56e8 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 @@ -1995,6 +1995,8 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { Pool mockPool = mock(Pool.class, name); + AtomicReference queryService = new AtomicReference<>(null); + AtomicBoolean destroyed = new AtomicBoolean(false); doAnswer(invocationOnMock -> { @@ -2033,6 +2035,12 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { when(mockPool.getSubscriptionRedundancy()).thenReturn(subscriptionRedundancy.get()); when(mockPool.getThreadLocalConnections()).thenReturn(threadLocalConnections.get()); + doAnswer(getQueryServiceInvocation -> + resolveAnyGemFireCache() + .map(GemFireCache::getQueryService) + .orElseGet(() -> queryService.updateAndGet(it -> it != null ? it : mockQueryService())) + ).when(mockPool).getQueryService(); + register(mockPool); return mockPool; @@ -3698,6 +3706,8 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { Pool mockDefaultPool = mockPoolFactory.create("DEFAULT"); + doAnswer(invocation -> mockClientCache.getQueryService()).when(mockDefaultPool).getQueryService(); + when(mockClientCache.getCurrentServers()).thenAnswer(invocation -> Collections.unmodifiableSet(new HashSet<>(mockClientCache.getDefaultPool().getServers())));