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 18ee747..61795b8 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 @@ -2358,6 +2358,7 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { }).when(mockRegion).containsValueForKey(any()); + // Map.forEach(:BiConsumer) doAnswer(invocation -> { BiConsumer consumer = invocation.getArgument(0); @@ -2368,8 +2369,8 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { }).when(mockRegion).forEach(any(BiConsumer.class)); - // Region.get(key) - when(mockRegion.get(ArgumentMatchers.any())).thenAnswer(invocation -> { + // Map.get(key) / Region.get(key) + doAnswer(invocation -> { K key = invocation.getArgument(0); V value = invalidatedKeys.contains(key) ? null : data.get(key); @@ -2400,7 +2401,8 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport { } return value; - }); + + }).when(mockRegion).get(ArgumentMatchers.any()); // Region.getEntry(key) when(mockRegion.getEntry(ArgumentMatchers.any())).thenAnswer(regionGetEntryInvocation ->