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

Refactor mockGemFireCache(..) and make public; add null check and mock the GemFireCache when mockGemFireCache argument is null.

Browse Source 
Add mockResourceManager(..) method.
This commit is contained in:
John Blum
2018-09-19 09:20:23 -07:00
parent 9523d37272
commit 797eda1494
1 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -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 extends GemFireCache> T mockGemFireCache(T mockGemFireCache,
@SuppressWarnings("unchecked")
public static <T extends GemFireCache> 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 <K, V> Region<K, V> mockRegion(String name, DataPolicy dataPolicy) {