Evict context from ContextCache before loading a new context
Since Spring Framework 4.2, DefaultContextCache supported an LRU (least recently used) eviction policy via a custom LruCache which extended LinkedHashMap. The LruCache reacted to LinkedHashMap's removeEldestEntry() callback to remove the LRU context if the maxSize of the cache was exceeded. Due to the nature of the implementation in LinkedHashMap, the removeEldestEntry() callback is invoked after a new entry has been stored to the map. Consequently, a Spring ApplicationContext (C1) was evicted from the cache after a new context (C2) was loaded and added to the cache, leading to failure scenarios such as the following. - C1 and C2 share an external resource -- for example, a database. - C2 initializes the external resource with test data when C2 is loaded. - C1 cleans up the external resource when C1 is closed. - C1 is loaded and added to the cache. - C2 is loaded and added to the cache before C1 is evicted. - C1 is evicted and closed. - C2 tests fail, because C1 removed test data required for C2. To address such scenarios, this commit replaces the custom LruCache with custom LRU eviction logic in DefaultContextCache and revises the put(MergedContextConfiguration, ApplicationContext) method to delegate to a new evictLruContextIfNecessary() method. This commit also introduces a new put(MergedContextConfiguration, LoadFunction) method in the ContextCache API which is overridden by DefaultContextCache to ensure that an evicted context is removed and closed before a new context is loaded to take its place in the cache. In addition, DefaultCacheAwareContextLoaderDelegate has been revised to make use of the new put(MergedContextConfiguration, LoadFunction) API. Closes gh-21007
This commit is contained in:
		
							parent
							
								
									8eca3a3eaf
								
							
						
					
					
						commit
						ca62119cb3
					
				|  | @ -21,6 +21,7 @@ import org.jspecify.annotations.Nullable; | ||||||
| import org.springframework.context.ApplicationContext; | import org.springframework.context.ApplicationContext; | ||||||
| import org.springframework.test.annotation.DirtiesContext.HierarchyMode; | import org.springframework.test.annotation.DirtiesContext.HierarchyMode; | ||||||
| import org.springframework.test.context.MergedContextConfiguration; | import org.springframework.test.context.MergedContextConfiguration; | ||||||
|  | import org.springframework.util.Assert; | ||||||
| 
 | 
 | ||||||
| /** | /** | ||||||
|  * {@code ContextCache} defines the SPI for caching Spring |  * {@code ContextCache} defines the SPI for caching Spring | ||||||
|  | @ -102,6 +103,7 @@ public interface ContextCache { | ||||||
| 	 * @param key the context key; never {@code null} | 	 * @param key the context key; never {@code null} | ||||||
| 	 * @return the corresponding {@code ApplicationContext} instance, or {@code null} | 	 * @return the corresponding {@code ApplicationContext} instance, or {@code null} | ||||||
| 	 * if not found in the cache | 	 * if not found in the cache | ||||||
|  | 	 * @see #put(MergedContextConfiguration, LoadFunction) | ||||||
| 	 * @see #unregisterContextUsage(MergedContextConfiguration, Class) | 	 * @see #unregisterContextUsage(MergedContextConfiguration, Class) | ||||||
| 	 * @see #remove(MergedContextConfiguration, HierarchyMode) | 	 * @see #remove(MergedContextConfiguration, HierarchyMode) | ||||||
| 	 */ | 	 */ | ||||||
|  | @ -113,9 +115,37 @@ public interface ContextCache { | ||||||
| 	 * @param key the context key; never {@code null} | 	 * @param key the context key; never {@code null} | ||||||
| 	 * @param context the {@code ApplicationContext}; never {@code null} | 	 * @param context the {@code ApplicationContext}; never {@code null} | ||||||
| 	 * @see #get(MergedContextConfiguration) | 	 * @see #get(MergedContextConfiguration) | ||||||
|  | 	 * @see #put(MergedContextConfiguration, LoadFunction) | ||||||
| 	 */ | 	 */ | ||||||
| 	void put(MergedContextConfiguration key, ApplicationContext context); | 	void put(MergedContextConfiguration key, ApplicationContext context); | ||||||
| 
 | 
 | ||||||
|  | 	/** | ||||||
|  | 	 * Explicitly add an {@link ApplicationContext} to the cache under the given | ||||||
|  | 	 * key, potentially honoring a custom eviction policy. | ||||||
|  | 	 * <p>The supplied {@link LoadFunction} will be invoked to load the | ||||||
|  | 	 * {@code ApplicationContext}. | ||||||
|  | 	 * <p>Concrete implementations which honor a custom eviction policy must | ||||||
|  | 	 * override this method to ensure that an evicted context is removed from the | ||||||
|  | 	 * cache and closed before a new context is loaded via the supplied | ||||||
|  | 	 * {@code LoadFunction}. | ||||||
|  | 	 * @param key the context key; never {@code null} | ||||||
|  | 	 * @param loadFunction a function which loads the context for the supplied key; | ||||||
|  | 	 * never {@code null} | ||||||
|  | 	 * @return the {@code ApplicationContext}; never {@code null} | ||||||
|  | 	 * @since 7.0 | ||||||
|  | 	 * @see #get(MergedContextConfiguration) | ||||||
|  | 	 * @see #put(MergedContextConfiguration, ApplicationContext) | ||||||
|  | 	 */ | ||||||
|  | 	default ApplicationContext put(MergedContextConfiguration key, LoadFunction loadFunction) { | ||||||
|  | 		Assert.notNull(key, "Key must not be null"); | ||||||
|  | 		Assert.notNull(loadFunction, "LoadFunction must not be null"); | ||||||
|  | 
 | ||||||
|  | 		ApplicationContext applicationContext = loadFunction.loadContext(key); | ||||||
|  | 		Assert.state(applicationContext != null, "LoadFunction must return a non-null ApplicationContext"); | ||||||
|  | 		put(key, applicationContext); | ||||||
|  | 		return applicationContext; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
| 	/** | 	/** | ||||||
| 	 * Remove the context with the given key from the cache and explicitly | 	 * Remove the context with the given key from the cache and explicitly | ||||||
| 	 * {@linkplain org.springframework.context.ConfigurableApplicationContext#close() close} | 	 * {@linkplain org.springframework.context.ConfigurableApplicationContext#close() close} | ||||||
|  | @ -281,4 +311,26 @@ public interface ContextCache { | ||||||
| 	 */ | 	 */ | ||||||
| 	void logStatistics(); | 	void logStatistics(); | ||||||
| 
 | 
 | ||||||
|  | 
 | ||||||
|  | 	/** | ||||||
|  | 	 * Represents a function that loads an {@link ApplicationContext}. | ||||||
|  | 	 * | ||||||
|  | 	 * @since 7.0 | ||||||
|  | 	 */ | ||||||
|  | 	@FunctionalInterface | ||||||
|  | 	interface LoadFunction { | ||||||
|  | 
 | ||||||
|  | 		/** | ||||||
|  | 		 * Load a new {@link ApplicationContext} based on the supplied | ||||||
|  | 		 * {@link MergedContextConfiguration} and return the context in a fully | ||||||
|  | 		 * <em>refreshed</em> state. | ||||||
|  | 		 * @param mergedConfig the merged context configuration to use to load the | ||||||
|  | 		 * application context | ||||||
|  | 		 * @return a new application context; never {@code null} | ||||||
|  | 		 * @see org.springframework.test.context.SmartContextLoader#loadContext(MergedContextConfiguration) | ||||||
|  | 		 */ | ||||||
|  | 		ApplicationContext loadContext(MergedContextConfiguration mergedConfig); | ||||||
|  | 
 | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
| } | } | ||||||
|  |  | ||||||
|  | @ -139,9 +139,16 @@ public class DefaultCacheAwareContextLoaderDelegate implements CacheAwareContext | ||||||
| 	public ApplicationContext loadContext(MergedContextConfiguration mergedConfig) { | 	public ApplicationContext loadContext(MergedContextConfiguration mergedConfig) { | ||||||
| 		mergedConfig = replaceIfNecessary(mergedConfig); | 		mergedConfig = replaceIfNecessary(mergedConfig); | ||||||
| 		synchronized (this.contextCache) { | 		synchronized (this.contextCache) { | ||||||
| 			ApplicationContext context = this.contextCache.get(mergedConfig); |  | ||||||
| 			try { | 			try { | ||||||
| 				if (context == null) { | 				ApplicationContext context = this.contextCache.get(mergedConfig); | ||||||
|  | 				if (context != null) { | ||||||
|  | 					if (logger.isTraceEnabled()) { | ||||||
|  | 						logger.trace("Retrieved ApplicationContext [%s] from cache with key %s".formatted( | ||||||
|  | 								System.identityHashCode(context), mergedConfig)); | ||||||
|  | 					} | ||||||
|  | 					return context; | ||||||
|  | 				} | ||||||
|  | 
 | ||||||
| 				int failureCount = this.contextCache.getFailureCount(mergedConfig); | 				int failureCount = this.contextCache.getFailureCount(mergedConfig); | ||||||
| 				if (failureCount >= this.failureThreshold) { | 				if (failureCount >= this.failureThreshold) { | ||||||
| 					throw new IllegalStateException(""" | 					throw new IllegalStateException(""" | ||||||
|  | @ -149,24 +156,27 @@ public class DefaultCacheAwareContextLoaderDelegate implements CacheAwareContext | ||||||
| 							skipping repeated attempt to load context for %s""" | 							skipping repeated attempt to load context for %s""" | ||||||
| 								.formatted(this.failureThreshold, mergedConfig)); | 								.formatted(this.failureThreshold, mergedConfig)); | ||||||
| 				} | 				} | ||||||
|  | 
 | ||||||
|  | 				return this.contextCache.put(mergedConfig, key -> { | ||||||
| 					try { | 					try { | ||||||
| 						if (mergedConfig instanceof AotMergedContextConfiguration aotMergedConfig) { | 						ApplicationContext newContext; | ||||||
| 							context = loadContextInAotMode(aotMergedConfig); | 						if (key instanceof AotMergedContextConfiguration aotMergedConfig) { | ||||||
|  | 							newContext = loadContextInAotMode(aotMergedConfig); | ||||||
| 						} | 						} | ||||||
| 						else { | 						else { | ||||||
| 							context = loadContextInternal(mergedConfig); | 							newContext = loadContextInternal(key); | ||||||
| 						} | 						} | ||||||
| 						if (logger.isTraceEnabled()) { | 						if (logger.isTraceEnabled()) { | ||||||
| 							logger.trace("Storing ApplicationContext [%s] in cache under key %s".formatted( | 							logger.trace("Storing ApplicationContext [%s] in cache under key %s".formatted( | ||||||
| 									System.identityHashCode(context), mergedConfig)); | 									System.identityHashCode(newContext), key)); | ||||||
| 						} | 						} | ||||||
| 						this.contextCache.put(mergedConfig, context); | 						return newContext; | ||||||
| 					} | 					} | ||||||
| 					catch (Exception ex) { | 					catch (Exception ex) { | ||||||
| 						if (logger.isTraceEnabled()) { | 						if (logger.isTraceEnabled()) { | ||||||
| 							logger.trace("Incrementing ApplicationContext failure count for " + mergedConfig); | 							logger.trace("Incrementing ApplicationContext failure count for " + key); | ||||||
| 						} | 						} | ||||||
| 						this.contextCache.incrementFailureCount(mergedConfig); | 						this.contextCache.incrementFailureCount(key); | ||||||
| 						Throwable cause = ex; | 						Throwable cause = ex; | ||||||
| 						if (ex instanceof ContextLoadException cle) { | 						if (ex instanceof ContextLoadException cle) { | ||||||
| 							cause = cle.getCause(); | 							cause = cle.getCause(); | ||||||
|  | @ -182,22 +192,13 @@ public class DefaultCacheAwareContextLoaderDelegate implements CacheAwareContext | ||||||
| 								} | 								} | ||||||
| 							} | 							} | ||||||
| 						} | 						} | ||||||
| 						throw new IllegalStateException( | 						throw new IllegalStateException("Failed to load ApplicationContext for " + key, cause); | ||||||
| 								"Failed to load ApplicationContext for " + mergedConfig, cause); |  | ||||||
| 					} |  | ||||||
| 				} |  | ||||||
| 				else { |  | ||||||
| 					if (logger.isTraceEnabled()) { |  | ||||||
| 						logger.trace("Retrieved ApplicationContext [%s] from cache with key %s".formatted( |  | ||||||
| 								System.identityHashCode(context), mergedConfig)); |  | ||||||
| 					} |  | ||||||
| 					} | 					} | ||||||
|  | 				}); | ||||||
| 			} | 			} | ||||||
| 			finally { | 			finally { | ||||||
| 				this.contextCache.logStatistics(); | 				this.contextCache.logStatistics(); | ||||||
| 			} | 			} | ||||||
| 
 |  | ||||||
| 			return context; |  | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
|  |  | ||||||
|  | @ -19,6 +19,7 @@ package org.springframework.test.context.cache; | ||||||
| import java.util.ArrayList; | import java.util.ArrayList; | ||||||
| import java.util.Collections; | import java.util.Collections; | ||||||
| import java.util.HashSet; | import java.util.HashSet; | ||||||
|  | import java.util.Iterator; | ||||||
| import java.util.LinkedHashMap; | import java.util.LinkedHashMap; | ||||||
| import java.util.List; | import java.util.List; | ||||||
| import java.util.Map; | import java.util.Map; | ||||||
|  | @ -62,7 +63,7 @@ public class DefaultContextCache implements ContextCache { | ||||||
| 	 * Map of context keys to Spring {@code ApplicationContext} instances. | 	 * Map of context keys to Spring {@code ApplicationContext} instances. | ||||||
| 	 */ | 	 */ | ||||||
| 	private final Map<MergedContextConfiguration, ApplicationContext> contextMap = | 	private final Map<MergedContextConfiguration, ApplicationContext> contextMap = | ||||||
| 			Collections.synchronizedMap(new LruCache(32, 0.75f)); | 			Collections.synchronizedMap(new LinkedHashMap<>(32, 0.75f, true)); | ||||||
| 
 | 
 | ||||||
| 	/** | 	/** | ||||||
| 	 * Map of parent keys to sets of children keys, representing a top-down <em>tree</em> | 	 * Map of parent keys to sets of children keys, representing a top-down <em>tree</em> | ||||||
|  | @ -157,7 +158,41 @@ public class DefaultContextCache implements ContextCache { | ||||||
| 		Assert.notNull(key, "Key must not be null"); | 		Assert.notNull(key, "Key must not be null"); | ||||||
| 		Assert.notNull(context, "ApplicationContext must not be null"); | 		Assert.notNull(context, "ApplicationContext must not be null"); | ||||||
| 
 | 
 | ||||||
|  | 		evictLruContextIfNecessary(); | ||||||
|  | 		putInternal(key, context); | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	@Override | ||||||
|  | 	public ApplicationContext put(MergedContextConfiguration key, LoadFunction loadFunction) { | ||||||
|  | 		Assert.notNull(key, "Key must not be null"); | ||||||
|  | 		Assert.notNull(loadFunction, "LoadFunction must not be null"); | ||||||
|  | 
 | ||||||
|  | 		evictLruContextIfNecessary(); | ||||||
|  | 		ApplicationContext context = loadFunction.loadContext(key); | ||||||
|  | 		Assert.state(context != null, "LoadFunction must return a non-null ApplicationContext"); | ||||||
|  | 		putInternal(key, context); | ||||||
|  | 		return context; | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	/** | ||||||
|  | 	 * Evict the least recently used (LRU) context if necessary. | ||||||
|  | 	 * @since 7.0 | ||||||
|  | 	 */ | ||||||
|  | 	private void evictLruContextIfNecessary() { | ||||||
|  | 		if (this.contextMap.size() >= this.maxSize) { | ||||||
|  | 			Iterator<MergedContextConfiguration> iterator = this.contextMap.keySet().iterator(); | ||||||
|  | 			Assert.state(iterator.hasNext(), "Failed to retrieve LRU context"); | ||||||
|  | 			// The least recently used (LRU) key is the first/head in a LinkedHashMap | ||||||
|  | 			// configured for access-order iteration order. | ||||||
|  | 			MergedContextConfiguration lruKey = iterator.next(); | ||||||
|  | 			remove(lruKey, HierarchyMode.CURRENT_LEVEL); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	private void putInternal(MergedContextConfiguration key, ApplicationContext context) { | ||||||
| 		this.contextMap.put(key, context); | 		this.contextMap.put(key, context); | ||||||
|  | 
 | ||||||
|  | 		// Update context hierarchy map. | ||||||
| 		MergedContextConfiguration child = key; | 		MergedContextConfiguration child = key; | ||||||
| 		MergedContextConfiguration parent = child.getParent(); | 		MergedContextConfiguration parent = child.getParent(); | ||||||
| 		while (parent != null) { | 		while (parent != null) { | ||||||
|  | @ -357,37 +392,4 @@ public class DefaultContextCache implements ContextCache { | ||||||
| 				.toString(); | 				.toString(); | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
| 
 |  | ||||||
| 	/** |  | ||||||
| 	 * Simple cache implementation based on {@link LinkedHashMap} with a maximum |  | ||||||
| 	 * size and a <em>least recently used</em> (LRU) eviction policy that |  | ||||||
| 	 * properly closes application contexts. |  | ||||||
| 	 * @since 4.3 |  | ||||||
| 	 */ |  | ||||||
| 	@SuppressWarnings("serial") |  | ||||||
| 	private class LruCache extends LinkedHashMap<MergedContextConfiguration, ApplicationContext> { |  | ||||||
| 
 |  | ||||||
| 		/** |  | ||||||
| 		 * Create a new {@code LruCache} with the supplied initial capacity |  | ||||||
| 		 * and load factor. |  | ||||||
| 		 * @param initialCapacity the initial capacity |  | ||||||
| 		 * @param loadFactor the load factor |  | ||||||
| 		 */ |  | ||||||
| 		LruCache(int initialCapacity, float loadFactor) { |  | ||||||
| 			super(initialCapacity, loadFactor, true); |  | ||||||
| 		} |  | ||||||
| 
 |  | ||||||
| 		@Override |  | ||||||
| 		protected boolean removeEldestEntry(Map.Entry<MergedContextConfiguration, ApplicationContext> eldest) { |  | ||||||
| 			if (this.size() > DefaultContextCache.this.getMaxSize()) { |  | ||||||
| 				// Do NOT delete "DefaultContextCache.this."; otherwise, we accidentally |  | ||||||
| 				// invoke java.util.Map.remove(Object, Object). |  | ||||||
| 				DefaultContextCache.this.remove(eldest.getKey(), HierarchyMode.CURRENT_LEVEL); |  | ||||||
| 			} |  | ||||||
| 
 |  | ||||||
| 			// Return false since we invoke a custom eviction algorithm. |  | ||||||
| 			return false; |  | ||||||
| 		} |  | ||||||
| 	} |  | ||||||
| 
 |  | ||||||
| } | } | ||||||
|  |  | ||||||
|  | @ -16,14 +16,29 @@ | ||||||
| 
 | 
 | ||||||
| package org.springframework.test.context.cache; | package org.springframework.test.context.cache; | ||||||
| 
 | 
 | ||||||
|  | import java.util.ArrayList; | ||||||
| import java.util.Arrays; | import java.util.Arrays; | ||||||
|  | import java.util.HashSet; | ||||||
| import java.util.List; | import java.util.List; | ||||||
|  | import java.util.Set; | ||||||
|  | import java.util.concurrent.atomic.AtomicInteger; | ||||||
| 
 | 
 | ||||||
|  | import org.junit.jupiter.api.AfterEach; | ||||||
|  | import org.junit.jupiter.api.BeforeEach; | ||||||
|  | import org.junit.jupiter.api.Nested; | ||||||
| import org.junit.jupiter.api.Test; | import org.junit.jupiter.api.Test; | ||||||
| 
 | 
 | ||||||
|  | import org.springframework.beans.factory.DisposableBean; | ||||||
|  | import org.springframework.beans.factory.InitializingBean; | ||||||
| import org.springframework.context.ApplicationContext; | import org.springframework.context.ApplicationContext; | ||||||
| import org.springframework.context.ConfigurableApplicationContext; | import org.springframework.context.ConfigurableApplicationContext; | ||||||
|  | import org.springframework.context.annotation.Configuration; | ||||||
|  | import org.springframework.context.annotation.Import; | ||||||
|  | import org.springframework.test.annotation.DirtiesContext.HierarchyMode; | ||||||
| import org.springframework.test.context.MergedContextConfiguration; | import org.springframework.test.context.MergedContextConfiguration; | ||||||
|  | import org.springframework.test.context.TestContext; | ||||||
|  | import org.springframework.test.context.TestContextTestUtils; | ||||||
|  | import org.springframework.test.context.junit.jupiter.SpringJUnitConfig; | ||||||
| 
 | 
 | ||||||
| import static org.assertj.core.api.Assertions.as; | import static org.assertj.core.api.Assertions.as; | ||||||
| import static org.assertj.core.api.Assertions.assertThat; | import static org.assertj.core.api.Assertions.assertThat; | ||||||
|  | @ -33,6 +48,7 @@ import static org.mockito.Mockito.mock; | ||||||
| import static org.mockito.Mockito.never; | import static org.mockito.Mockito.never; | ||||||
| import static org.mockito.Mockito.times; | import static org.mockito.Mockito.times; | ||||||
| import static org.mockito.Mockito.verify; | import static org.mockito.Mockito.verify; | ||||||
|  | import static org.springframework.test.context.cache.ContextCacheTestUtils.assertContextCacheStatistics; | ||||||
| 
 | 
 | ||||||
| /** | /** | ||||||
|  * Tests for the LRU eviction policy in {@link DefaultContextCache}. |  * Tests for the LRU eviction policy in {@link DefaultContextCache}. | ||||||
|  | @ -65,6 +81,10 @@ class LruContextCacheTests { | ||||||
| 		assertThatIllegalArgumentException().isThrownBy(() -> new DefaultContextCache(0)); | 		assertThatIllegalArgumentException().isThrownBy(() -> new DefaultContextCache(0)); | ||||||
| 	} | 	} | ||||||
| 
 | 
 | ||||||
|  | 
 | ||||||
|  | 	@Nested | ||||||
|  | 	class PutUnitTests { | ||||||
|  | 
 | ||||||
| 		@Test | 		@Test | ||||||
| 		void maxCacheSizeOne() { | 		void maxCacheSizeOne() { | ||||||
| 			DefaultContextCache cache = new DefaultContextCache(1); | 			DefaultContextCache cache = new DefaultContextCache(1); | ||||||
|  | @ -149,6 +169,316 @@ class LruContextCacheTests { | ||||||
| 			verify(abcContext, never()).close(); | 			verify(abcContext, never()).close(); | ||||||
| 			verify(bazContext, never()).close(); | 			verify(bazContext, never()).close(); | ||||||
| 		} | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	/** | ||||||
|  | 	 * @since 7.0 | ||||||
|  | 	 */ | ||||||
|  | 	@Nested | ||||||
|  | 	class PutWithLoadFunctionUnitTests { | ||||||
|  | 
 | ||||||
|  | 		@Test | ||||||
|  | 		void maxCacheSizeOne() { | ||||||
|  | 			DefaultContextCache cache = new DefaultContextCache(1); | ||||||
|  | 			assertThat(cache.size()).isEqualTo(0); | ||||||
|  | 			assertThat(cache.getMaxSize()).isEqualTo(1); | ||||||
|  | 
 | ||||||
|  | 			cache.put(fooConfig, key -> fooContext); | ||||||
|  | 			assertCacheContents(cache, "Foo"); | ||||||
|  | 
 | ||||||
|  | 			cache.put(fooConfig, key -> fooContext); | ||||||
|  | 			assertCacheContents(cache, "Foo"); | ||||||
|  | 
 | ||||||
|  | 			cache.put(barConfig, key -> barContext); | ||||||
|  | 			assertCacheContents(cache, "Bar"); | ||||||
|  | 
 | ||||||
|  | 			cache.put(fooConfig, key -> fooContext); | ||||||
|  | 			assertCacheContents(cache, "Foo"); | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		@Test | ||||||
|  | 		void maxCacheSizeThree() { | ||||||
|  | 			DefaultContextCache cache = new DefaultContextCache(3); | ||||||
|  | 			assertThat(cache.size()).isEqualTo(0); | ||||||
|  | 			assertThat(cache.getMaxSize()).isEqualTo(3); | ||||||
|  | 
 | ||||||
|  | 			cache.put(fooConfig, key -> fooContext); | ||||||
|  | 			assertCacheContents(cache, "Foo"); | ||||||
|  | 
 | ||||||
|  | 			cache.put(fooConfig, key -> fooContext); | ||||||
|  | 			assertCacheContents(cache, "Foo"); | ||||||
|  | 
 | ||||||
|  | 			cache.put(barConfig, key -> barContext); | ||||||
|  | 			assertCacheContents(cache, "Foo", "Bar"); | ||||||
|  | 
 | ||||||
|  | 			cache.put(bazConfig, key -> bazContext); | ||||||
|  | 			assertCacheContents(cache, "Foo", "Bar", "Baz"); | ||||||
|  | 
 | ||||||
|  | 			cache.put(abcConfig, key -> abcContext); | ||||||
|  | 			assertCacheContents(cache, "Bar", "Baz", "Abc"); | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		@Test | ||||||
|  | 		void ensureLruOrderingIsUpdated() { | ||||||
|  | 			DefaultContextCache cache = new DefaultContextCache(3); | ||||||
|  | 
 | ||||||
|  | 			// Note: when a new entry is added it is considered the MRU entry and inserted at the tail. | ||||||
|  | 			cache.put(fooConfig, key -> fooContext); | ||||||
|  | 			cache.put(barConfig, key -> barContext); | ||||||
|  | 			cache.put(bazConfig, key -> bazContext); | ||||||
|  | 			assertCacheContents(cache, "Foo", "Bar", "Baz"); | ||||||
|  | 
 | ||||||
|  | 			// Note: the MRU entry is moved to the tail when accessed. | ||||||
|  | 			cache.get(fooConfig); | ||||||
|  | 			assertCacheContents(cache, "Bar", "Baz", "Foo"); | ||||||
|  | 
 | ||||||
|  | 			cache.get(barConfig); | ||||||
|  | 			assertCacheContents(cache, "Baz", "Foo", "Bar"); | ||||||
|  | 
 | ||||||
|  | 			cache.get(bazConfig); | ||||||
|  | 			assertCacheContents(cache, "Foo", "Bar", "Baz"); | ||||||
|  | 
 | ||||||
|  | 			cache.get(barConfig); | ||||||
|  | 			assertCacheContents(cache, "Foo", "Baz", "Bar"); | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		@Test | ||||||
|  | 		void ensureEvictedContextsAreClosed() { | ||||||
|  | 			DefaultContextCache cache = new DefaultContextCache(2); | ||||||
|  | 
 | ||||||
|  | 			cache.put(fooConfig, key -> fooContext); | ||||||
|  | 			cache.put(barConfig, key -> barContext); | ||||||
|  | 			assertCacheContents(cache, "Foo", "Bar"); | ||||||
|  | 
 | ||||||
|  | 			cache.put(bazConfig, key -> bazContext); | ||||||
|  | 			assertCacheContents(cache, "Bar", "Baz"); | ||||||
|  | 			verify(fooContext, times(1)).close(); | ||||||
|  | 
 | ||||||
|  | 			cache.put(abcConfig, key -> abcContext); | ||||||
|  | 			assertCacheContents(cache, "Baz", "Abc"); | ||||||
|  | 			verify(barContext, times(1)).close(); | ||||||
|  | 
 | ||||||
|  | 			verify(abcContext, never()).close(); | ||||||
|  | 			verify(bazContext, never()).close(); | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	/** | ||||||
|  | 	 * @since 7.0 | ||||||
|  | 	 */ | ||||||
|  | 	@Nested | ||||||
|  | 	class PutWithLoadFunctionIntegrationTests { | ||||||
|  | 
 | ||||||
|  | 		/** | ||||||
|  | 		 * Mimics a database shared across application contexts. | ||||||
|  | 		 */ | ||||||
|  | 		private static final Set<String> database = new HashSet<>(); | ||||||
|  | 
 | ||||||
|  | 		private static final List<String> events = new ArrayList<>(); | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | 		@BeforeEach | ||||||
|  | 		@AfterEach | ||||||
|  | 		void resetTracking() { | ||||||
|  | 			resetEvents(); | ||||||
|  | 			DatabaseInitializer.counter.set(0); | ||||||
|  | 			database.clear(); | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		@Test | ||||||
|  | 		void maxCacheSizeOne() { | ||||||
|  | 			DefaultContextCache contextCache = new DefaultContextCache(1); | ||||||
|  | 
 | ||||||
|  | 			// ----------------------------------------------------------------- | ||||||
|  | 
 | ||||||
|  | 			// Get ApplicationContext for TestCase1. | ||||||
|  | 			Class<?> testClass1 = TestCase1.class; | ||||||
|  | 			TestContext testContext1 = TestContextTestUtils.buildTestContext(testClass1, contextCache); | ||||||
|  | 			testContext1.getApplicationContext(); | ||||||
|  | 			assertContextCacheStatistics(contextCache, testClass1.getSimpleName(), 1, 1, 0, 1); | ||||||
|  | 			assertCacheContents(contextCache, "Config1"); | ||||||
|  | 			assertThat(database).containsExactly("enigma1"); | ||||||
|  | 			assertThat(events).containsExactly("START 1"); | ||||||
|  | 			resetEvents(); | ||||||
|  | 
 | ||||||
|  | 			// ----------------------------------------------------------------- | ||||||
|  | 
 | ||||||
|  | 			// Get ApplicationContext for TestCase2. | ||||||
|  | 			Class<?> testClass2 = TestCase2.class; | ||||||
|  | 			TestContext testContext2 = TestContextTestUtils.buildTestContext(testClass2, contextCache); | ||||||
|  | 			testContext2.getApplicationContext(); | ||||||
|  | 			assertContextCacheStatistics(contextCache, testClass2.getSimpleName(), 1, 1, 0, 2); | ||||||
|  | 			assertCacheContents(contextCache, "Config2"); | ||||||
|  | 			assertThat(database).containsExactly("enigma2"); | ||||||
|  | 			assertThat(events).containsExactly("CLOSE 1", "START 2"); | ||||||
|  | 			resetEvents(); | ||||||
|  | 
 | ||||||
|  | 			// ----------------------------------------------------------------- | ||||||
|  | 
 | ||||||
|  | 			// Get ApplicationContext for TestCase3. | ||||||
|  | 			Class<?> testClass3 = TestCase3.class; | ||||||
|  | 			TestContext testContext3 = TestContextTestUtils.buildTestContext(testClass3, contextCache); | ||||||
|  | 			testContext3.getApplicationContext(); | ||||||
|  | 			assertContextCacheStatistics(contextCache, testClass3.getSimpleName(), 1, 1, 0, 3); | ||||||
|  | 			assertCacheContents(contextCache, "Config3"); | ||||||
|  | 			assertThat(database).containsExactly("enigma3"); | ||||||
|  | 			assertThat(events).containsExactly("CLOSE 2", "START 3"); | ||||||
|  | 			resetEvents(); | ||||||
|  | 
 | ||||||
|  | 			// ----------------------------------------------------------------- | ||||||
|  | 
 | ||||||
|  | 			// Get ApplicationContext for TestCase1 again. | ||||||
|  | 			testContext1.getApplicationContext(); | ||||||
|  | 			assertContextCacheStatistics(contextCache, testClass1.getSimpleName(), 1, 1, 0, 4); | ||||||
|  | 			assertCacheContents(contextCache, "Config1"); | ||||||
|  | 			assertThat(database).containsExactly("enigma4"); | ||||||
|  | 			assertThat(events).containsExactly("CLOSE 3", "START 4"); | ||||||
|  | 			resetEvents(); | ||||||
|  | 
 | ||||||
|  | 			// ----------------------------------------------------------------- | ||||||
|  | 
 | ||||||
|  | 			testContext1.markApplicationContextDirty(HierarchyMode.EXHAUSTIVE); | ||||||
|  | 			assertThat(events).containsExactly("CLOSE 4"); | ||||||
|  | 			assertThat(database).isEmpty(); | ||||||
|  | 			assertThat(contextCache.size()).isZero(); | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		@Test | ||||||
|  | 		void maxCacheSizeTwo() { | ||||||
|  | 			DefaultContextCache contextCache = new DefaultContextCache(2); | ||||||
|  | 
 | ||||||
|  | 			// ----------------------------------------------------------------- | ||||||
|  | 
 | ||||||
|  | 			// Get ApplicationContext for TestCase1. | ||||||
|  | 			Class<?> testClass1 = TestCase1.class; | ||||||
|  | 			TestContext testContext1 = TestContextTestUtils.buildTestContext(testClass1, contextCache); | ||||||
|  | 			testContext1.getApplicationContext(); | ||||||
|  | 			assertContextCacheStatistics(contextCache, testClass1.getSimpleName(), 1, 1, 0, 1); | ||||||
|  | 			testContext1.markApplicationContextUnused(); | ||||||
|  | 			assertContextCacheStatistics(contextCache, testClass1.getSimpleName(), 1, 0, 0, 1); | ||||||
|  | 			assertCacheContents(contextCache, "Config1"); | ||||||
|  | 			assertThat(events).containsExactly("START 1"); | ||||||
|  | 			assertThat(database).containsExactly("enigma1"); | ||||||
|  | 			resetEvents(); | ||||||
|  | 
 | ||||||
|  | 			// ----------------------------------------------------------------- | ||||||
|  | 
 | ||||||
|  | 			// Get ApplicationContext for TestCase2. | ||||||
|  | 			Class<?> testClass2 = TestCase2.class; | ||||||
|  | 			TestContext testContext2 = TestContextTestUtils.buildTestContext(testClass2, contextCache); | ||||||
|  | 			testContext2.getApplicationContext(); | ||||||
|  | 			assertContextCacheStatistics(contextCache, testClass2.getSimpleName(), 2, 1, 0, 2); | ||||||
|  | 			testContext2.markApplicationContextUnused(); | ||||||
|  | 			assertContextCacheStatistics(contextCache, testClass2.getSimpleName(), 2, 0, 0, 2); | ||||||
|  | 			assertCacheContents(contextCache, "Config1", "Config2"); | ||||||
|  | 			assertThat(events).containsExactly("START 2"); | ||||||
|  | 			assertThat(database).containsExactly("enigma1", "enigma2"); | ||||||
|  | 			resetEvents(); | ||||||
|  | 
 | ||||||
|  | 			// ----------------------------------------------------------------- | ||||||
|  | 
 | ||||||
|  | 			// Get ApplicationContext for TestCase3. | ||||||
|  | 			Class<?> testClass3 = TestCase3.class; | ||||||
|  | 			TestContext testContext3 = TestContextTestUtils.buildTestContext(testClass3, contextCache); | ||||||
|  | 			testContext3.getApplicationContext(); | ||||||
|  | 			assertContextCacheStatistics(contextCache, testClass3.getSimpleName(), 2, 1, 0, 3); | ||||||
|  | 			testContext3.markApplicationContextUnused(); | ||||||
|  | 			assertContextCacheStatistics(contextCache, testClass3.getSimpleName(), 2, 0, 0, 3); | ||||||
|  | 			assertCacheContents(contextCache, "Config2", "Config3"); | ||||||
|  | 			assertThat(events).containsExactly("CLOSE 1", "START 3"); | ||||||
|  | 			// Closing App #1 removed "enigma1" and "enigma2" from the database. | ||||||
|  | 			assertThat(database).containsExactly("enigma3"); | ||||||
|  | 			resetEvents(); | ||||||
|  | 
 | ||||||
|  | 			// ----------------------------------------------------------------- | ||||||
|  | 
 | ||||||
|  | 			// Get ApplicationContext for TestCase1 again. | ||||||
|  | 			testContext1.getApplicationContext(); | ||||||
|  | 			assertContextCacheStatistics(contextCache, testClass1.getSimpleName(), 2, 1, 0, 4); | ||||||
|  | 			testContext1.markApplicationContextUnused(); | ||||||
|  | 			assertContextCacheStatistics(contextCache, testClass1.getSimpleName(), 2, 0, 0, 4); | ||||||
|  | 			assertCacheContents(contextCache, "Config3", "Config1"); | ||||||
|  | 			assertThat(events).containsExactly("CLOSE 2", "START 4"); | ||||||
|  | 			// Closing App #2 removed "enigma3" from the database. | ||||||
|  | 			assertThat(database).containsExactly("enigma4"); | ||||||
|  | 			resetEvents(); | ||||||
|  | 
 | ||||||
|  | 			// ----------------------------------------------------------------- | ||||||
|  | 
 | ||||||
|  | 			testContext3.markApplicationContextDirty(HierarchyMode.EXHAUSTIVE); | ||||||
|  | 			assertThat(events).containsExactly("CLOSE 3"); | ||||||
|  | 			resetEvents(); | ||||||
|  | 
 | ||||||
|  | 			testContext1.markApplicationContextDirty(HierarchyMode.EXHAUSTIVE); | ||||||
|  | 			assertThat(events).containsExactly("CLOSE 4"); | ||||||
|  | 			assertThat(database).isEmpty(); | ||||||
|  | 			assertThat(contextCache.size()).isZero(); | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | 		private static void resetEvents() { | ||||||
|  | 			events.clear(); | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | 		/** | ||||||
|  | 		 * Mimics a Spring component that inserts data into the database when the | ||||||
|  | 		 * application context is started and drops data from a database when the | ||||||
|  | 		 * application context is closed. | ||||||
|  | 		 * | ||||||
|  | 		 * @see org.springframework.jdbc.datasource.init.DataSourceInitializer | ||||||
|  | 		 */ | ||||||
|  | 		static class DatabaseInitializer implements InitializingBean, DisposableBean { | ||||||
|  | 
 | ||||||
|  | 			static final AtomicInteger counter = new AtomicInteger(); | ||||||
|  | 
 | ||||||
|  | 			private final int count; | ||||||
|  | 
 | ||||||
|  | 
 | ||||||
|  | 			DatabaseInitializer() { | ||||||
|  | 				this.count = counter.incrementAndGet(); | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			@Override | ||||||
|  | 			public void afterPropertiesSet() { | ||||||
|  | 				events.add("START " + this.count); | ||||||
|  | 				database.add("enigma" + this.count); | ||||||
|  | 			} | ||||||
|  | 
 | ||||||
|  | 			@Override | ||||||
|  | 			public void destroy() { | ||||||
|  | 				events.add("CLOSE " + this.count); | ||||||
|  | 				database.clear(); | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		@SpringJUnitConfig | ||||||
|  | 		static class TestCase1 { | ||||||
|  | 
 | ||||||
|  | 			@Configuration | ||||||
|  | 			@Import(DatabaseInitializer.class) | ||||||
|  | 			static class Config1 { | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		@SpringJUnitConfig | ||||||
|  | 		static class TestCase2 { | ||||||
|  | 
 | ||||||
|  | 			@Configuration | ||||||
|  | 			@Import(DatabaseInitializer.class) | ||||||
|  | 			static class Config2 { | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 
 | ||||||
|  | 		@SpringJUnitConfig | ||||||
|  | 		static class TestCase3 { | ||||||
|  | 
 | ||||||
|  | 			@Configuration | ||||||
|  | 			@Import(DatabaseInitializer.class) | ||||||
|  | 			static class Config3 { | ||||||
|  | 			} | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
| 
 | 
 | ||||||
| 
 | 
 | ||||||
| 	private static MergedContextConfiguration config(Class<?> clazz) { | 	private static MergedContextConfiguration config(Class<?> clazz) { | ||||||
|  |  | ||||||
		Loading…
	
		Reference in New Issue