diff --git a/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/mock/CacheMockObjects.java b/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/mock/CacheMockObjects.java index e7a12be..0e918ca 100644 --- a/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/mock/CacheMockObjects.java +++ b/spring-data-geode-test/src/main/java/org/springframework/data/gemfire/tests/mock/CacheMockObjects.java @@ -44,6 +44,7 @@ import org.springframework.data.gemfire.util.RegionUtils; * @see org.apache.geode.cache.GemFireCache * @see org.apache.geode.cache.Region * @see org.apache.geode.cache.client.ClientCache + * @see org.apache.geode.cache.control.ResourceManager * @see org.apache.geode.distributed.DistributedMember * @see org.apache.geode.distributed.DistributedSystem * @since 1.0.0 @@ -51,9 +52,12 @@ import org.springframework.data.gemfire.util.RegionUtils; @SuppressWarnings("unused") public abstract class CacheMockObjects { - private static T mockGemFireCache(T mockGemFireCache, + @SuppressWarnings("unchecked") + public static T mockGemFireCache(T mockGemFireCache, DistributedSystem distributedSystem, ResourceManager resourceManager, Region... regions) { + mockGemFireCache = mockGemFireCache != null ? mockGemFireCache : (T) mock(GemFireCache.class); + when(mockGemFireCache.getDistributedSystem()).thenReturn(distributedSystem); when(mockGemFireCache.getResourceManager()).thenReturn(resourceManager); when(mockGemFireCache.rootRegions()).thenReturn(asSet(regions)); @@ -94,6 +98,19 @@ public abstract class CacheMockObjects { } + public static ResourceManager mockResourceManager(float criticalHeapPercentage, float criticalOffHeapPercentage, + float evictionHeapPercentage, float evictionOffHeapPercentage) { + + ResourceManager mockResourceManager = mock(ResourceManager.class); + + when(mockResourceManager.getCriticalHeapPercentage()).thenReturn(criticalHeapPercentage); + when(mockResourceManager.getCriticalOffHeapPercentage()).thenReturn(criticalOffHeapPercentage); + when(mockResourceManager.getEvictionHeapPercentage()).thenReturn(evictionHeapPercentage); + when(mockResourceManager.getEvictionOffHeapPercentage()).thenReturn(evictionOffHeapPercentage); + + return mockResourceManager; + } + @SuppressWarnings("unchecked") public static Region mockRegion(String name, DataPolicy dataPolicy) {