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

Add capability to unregister all Functions registered with the FunctionService on destruction of the GemFire Mock Objects.

Browse Source
This commit is contained in:
John Blum
2021-09-14 11:00:32 -07:00
parent 87a7bc8e49
commit b1de07519b
1 changed files with 22 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -124,6 +124,7 @@ import org.apache.geode.cache.client.PoolFactory;
import org.apache.geode.cache.client.PoolManager;
import org.apache.geode.cache.client.SocketFactory;
import org.apache.geode.cache.control.ResourceManager;
import org.apache.geode.cache.execute.FunctionService;
import org.apache.geode.cache.execute.RegionFunctionContext;
import org.apache.geode.cache.lucene.LuceneIndex;
import org.apache.geode.cache.lucene.LuceneIndexFactory;
@@ -323,6 +324,7 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport {
regions.clear();
regionAttributes.clear();
unregisterFunctions();
unregisterManagedPools();
closePools();
destroyGemFireObjects();
@@ -331,10 +333,12 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport {
/**
* Clears all {@literal spring.data.gemfire.test.*} {@link System#getProperties() System Properties}.
*
* @see java.lang.System#getProperties()
*/
static void clearSpringDataGeodeTestProperties() {
Arrays.stream(ArrayUtils.nullSafeArray(SPRING_DATA_GEODE_TEST_PROPERTIES, String.class))
.forEach(property -> System.clearProperty(property));
.forEach(System::clearProperty);
}
/**
@@ -354,6 +358,8 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport {
/**
* Destroys all {@link DisposableBean} based {@link Object GemFire objects}.
*
* @see org.springframework.beans.factory.DisposableBean
*/
static synchronized void destroyGemFireObjects() {
@@ -370,8 +376,23 @@ public abstract class GemFireMockObjectsSupport extends MockObjectsSupport {
cachedGemFireObjects.clear();
}
/**
* Unregisters all {@link Function Functions} registered with the {@link FunctionService} by Spring.
*
* @see org.apache.geode.cache.execute.Function
* @see org.apache.geode.cache.execute.FunctionService
*/
static synchronized void unregisterFunctions() {
CollectionUtils.nullSafeMap(FunctionService.getRegisteredFunctions())
.forEach((functionId, function) -> FunctionService.unregisterFunction(functionId));
}
/**
* Unrigsters all {@link Pool Pools} registered with Apache Geode and managed by Spring.
*
* @see org.apache.geode.cache.client.Pool
* @see org.apache.geode.cache.client.PoolManager
*/
static synchronized void unregisterManagedPools() {