/** * Some tests run so quickly that the rebalance operation doesn't even have time to start before * regions are already being destroyed - GEDOE-4312. */ private static void waitForNoRebalancing() { if (cache != null && !cache.isClosed()) { await().until(() -> { return cache.getResourceManager().getRebalanceOperations().size() == 0; }); } }
public static ResourceManager mockResourceManager() { ResourceManager mockResourceManager = mock(ResourceManager.class); AtomicReference<Float> criticalHeapPercentage = new AtomicReference<>(ResourceManager.DEFAULT_CRITICAL_PERCENTAGE); AtomicReference<Float> criticalOffHeapPercentage = new AtomicReference<>(0.0f); AtomicReference<Float> evictionHeapPercentage = new AtomicReference<>(ResourceManager.DEFAULT_EVICTION_PERCENTAGE); AtomicReference<Float> evictionOffHeapPercentage = new AtomicReference<>(0.0f); doAnswer(newSetter(criticalHeapPercentage, null)) .when(mockResourceManager).setCriticalHeapPercentage(anyFloat()); doAnswer(newSetter(criticalOffHeapPercentage, null)) .when(mockResourceManager).setCriticalOffHeapPercentage(anyFloat()); doAnswer(newSetter(evictionHeapPercentage, null)) .when(mockResourceManager).setEvictionHeapPercentage(anyFloat()); doAnswer(newSetter(evictionOffHeapPercentage, null)) .when(mockResourceManager).setEvictionOffHeapPercentage(anyFloat()); when(mockResourceManager.getCriticalHeapPercentage()).thenAnswer(newGetter(criticalHeapPercentage)); when(mockResourceManager.getCriticalOffHeapPercentage()).thenAnswer(newGetter(criticalOffHeapPercentage)); when(mockResourceManager.getEvictionHeapPercentage()).thenAnswer(newGetter(evictionHeapPercentage)); when(mockResourceManager.getEvictionOffHeapPercentage()).thenAnswer(newGetter(evictionOffHeapPercentage)); when(mockResourceManager.getRebalanceOperations()).thenReturn(Collections.emptySet()); return mockResourceManager; }