Add ControllerAdviceBean#resolveBean() tests

Author: tafjwr

spring-web/src/test/java/org/springframework/web/method/ControllerAdviceBeanTests.java

@@ -89,9 +89,9 @@ void shouldFailWhenControllerAdviceNull() {
 
 	@Test
 	void equalsHashCodeAndToStringForBeanName() {
-		String beanName = SimpleControllerAdvice.class.getSimpleName();
-		ControllerAdviceBean bean1 = createControllerAdviceBean(SimpleControllerAdvice.class);
-		ControllerAdviceBean bean2 = createControllerAdviceBean(SimpleControllerAdvice.class);
+		String beanName = getSingletonBeanName(SimpleControllerAdvice.class);
+		ControllerAdviceBean bean1 = createSingletonControllerAdviceBean(SimpleControllerAdvice.class);
+		ControllerAdviceBean bean2 = createSingletonControllerAdviceBean(SimpleControllerAdvice.class);
 		assertEqualsHashCodeAndToString(bean1, bean2, beanName);
 	}
 
@@ -127,9 +127,28 @@ void orderedViaAnnotationForBeanInstance() {
 		assertOrder(PriorityAnnotationControllerAdvice.class, 200);
 	}
 
+	@Test
+	void resolveBeanForSingletonBean() {
+		ControllerAdviceBean cab = createSingletonControllerAdviceBean(SimpleControllerAdvice.class);
+		Object bean = this.applicationContext.getBean(getSingletonBeanName(SimpleControllerAdvice.class));
+		assertThat(cab).extracting("resolvedBean").isNull();
+		Object resolvedBean = cab.resolveBean();
+		assertThat(cab).extracting("resolvedBean").isEqualTo(bean);
+		assertThat(resolvedBean).isEqualTo(bean);
+	}
+
+	@Test
+	void resolveBeanForNonSingletonBean() {
+		ControllerAdviceBean cab = createPrototypeControllerAdviceBean(SimpleControllerAdvice.class);
+		assertThat(cab).extracting("resolvedBean").isNull();
+		Object resolvedBean = cab.resolveBean();
+		assertThat(cab).extracting("resolvedBean").isNull();
+		assertThat(resolvedBean).isInstanceOf(SimpleControllerAdvice.class);
+	}
+
 	@Test
 	void shouldMatchAll() {
-		ControllerAdviceBean bean = createControllerAdviceBean(SimpleControllerAdvice.class);
+		ControllerAdviceBean bean = createSingletonControllerAdviceBean(SimpleControllerAdvice.class);
 		assertApplicable("should match all", bean, AnnotatedController.class);
 		assertApplicable("should match all", bean, ImplementationController.class);
 		assertApplicable("should match all", bean, InheritanceController.class);
@@ -138,7 +157,7 @@ void shouldMatchAll() {
 
 	@Test
 	void basePackageSupport() {
-		ControllerAdviceBean bean = createControllerAdviceBean(BasePackageSupport.class);
+		ControllerAdviceBean bean = createSingletonControllerAdviceBean(BasePackageSupport.class);
 		assertApplicable("base package support", bean, AnnotatedController.class);
 		assertApplicable("base package support", bean, ImplementationController.class);
 		assertApplicable("base package support", bean, InheritanceController.class);
@@ -147,7 +166,7 @@ void basePackageSupport() {
 
 	@Test
 	void basePackageValueSupport() {
-		ControllerAdviceBean bean = createControllerAdviceBean(BasePackageValueSupport.class);
+		ControllerAdviceBean bean = createSingletonControllerAdviceBean(BasePackageValueSupport.class);
 		assertApplicable("base package support", bean, AnnotatedController.class);
 		assertApplicable("base package support", bean, ImplementationController.class);
 		assertApplicable("base package support", bean, InheritanceController.class);
@@ -156,14 +175,14 @@ void basePackageValueSupport() {
 
 	@Test
 	void annotationSupport() {
-		ControllerAdviceBean bean = createControllerAdviceBean(AnnotationSupport.class);
+		ControllerAdviceBean bean = createSingletonControllerAdviceBean(AnnotationSupport.class);
 		assertApplicable("annotation support", bean, AnnotatedController.class);
 		assertNotApplicable("this bean is not annotated", bean, InheritanceController.class);
 	}
 
 	@Test
 	void markerClassSupport() {
-		ControllerAdviceBean bean = createControllerAdviceBean(MarkerClassSupport.class);
+		ControllerAdviceBean bean = createSingletonControllerAdviceBean(MarkerClassSupport.class);
 		assertApplicable("base package class support", bean, AnnotatedController.class);
 		assertApplicable("base package class support", bean, ImplementationController.class);
 		assertApplicable("base package class support", bean, InheritanceController.class);
@@ -172,7 +191,7 @@ void markerClassSupport() {
 
 	@Test
 	void shouldNotMatch() {
-		ControllerAdviceBean bean = createControllerAdviceBean(ShouldNotMatch.class);
+		ControllerAdviceBean bean = createSingletonControllerAdviceBean(ShouldNotMatch.class);
 		assertNotApplicable("should not match", bean, AnnotatedController.class);
 		assertNotApplicable("should not match", bean, ImplementationController.class);
 		assertNotApplicable("should not match", bean, InheritanceController.class);
@@ -181,7 +200,7 @@ void shouldNotMatch() {
 
 	@Test
 	void assignableTypesSupport() {
-		ControllerAdviceBean bean = createControllerAdviceBean(AssignableTypesSupport.class);
+		ControllerAdviceBean bean = createSingletonControllerAdviceBean(AssignableTypesSupport.class);
 		assertApplicable("controller implements assignable", bean, ImplementationController.class);
 		assertApplicable("controller inherits assignable", bean, InheritanceController.class);
 		assertNotApplicable("not assignable", bean, AnnotatedController.class);
@@ -190,7 +209,7 @@ void assignableTypesSupport() {
 
 	@Test
 	void multipleMatch() {
-		ControllerAdviceBean bean = createControllerAdviceBean(MultipleSelectorsSupport.class);
+		ControllerAdviceBean bean = createSingletonControllerAdviceBean(MultipleSelectorsSupport.class);
 		assertApplicable("controller implements assignable", bean, ImplementationController.class);
 		assertApplicable("controller is annotated", bean, AnnotatedController.class);
 		assertNotApplicable("should not match", bean, InheritanceController.class);
@@ -216,13 +235,28 @@ public void findAnnotatedBeansSortsBeans() {
 		assertThat(adviceBeans).extracting(ControllerAdviceBean::getBeanType).containsExactly(expectedTypes);
 	}
 
-	private ControllerAdviceBean createControllerAdviceBean(Class<?> beanType) {
-		String beanName = beanType.getSimpleName();
+	private ControllerAdviceBean createSingletonControllerAdviceBean(Class<?> beanType) {
+		String beanName = getSingletonBeanName(beanType);
 		this.applicationContext.registerSingleton(beanName, beanType);
 		ControllerAdvice controllerAdvice = AnnotatedElementUtils.findMergedAnnotation(beanType, ControllerAdvice.class);
 		return new ControllerAdviceBean(beanName, this.applicationContext, controllerAdvice);
 	}
 
+	private static String getSingletonBeanName(Class<?> beanType) {
+		return beanType.getSimpleName() + "Singleton";
+	}
+
+	private ControllerAdviceBean createPrototypeControllerAdviceBean(Class<?> beanType) {
+		String beanName = getPrototypeBeanName(beanType);
+		this.applicationContext.registerPrototype(beanName, beanType);
+		ControllerAdvice controllerAdvice = AnnotatedElementUtils.findMergedAnnotation(beanType, ControllerAdvice.class);
+		return new ControllerAdviceBean(beanName, this.applicationContext, controllerAdvice);
+	}
+
+	private static String getPrototypeBeanName(Class<?> beanType) {
+		return beanType.getSimpleName() + "Prototype";
+	}
+
 	private void assertEqualsHashCodeAndToString(ControllerAdviceBean bean1, ControllerAdviceBean bean2, String toString) {
 		assertThat(bean1).isEqualTo(bean2);
 		assertThat(bean2).isEqualTo(bean1);
@@ -232,7 +266,7 @@ private void assertEqualsHashCodeAndToString(ControllerAdviceBean bean1, Control
 	}
 
 	private void assertOrder(Class<?> beanType, int expectedOrder) {
-		assertThat(createControllerAdviceBean(beanType).getOrder()).isEqualTo(expectedOrder);
+		assertThat(createSingletonControllerAdviceBean(beanType).getOrder()).isEqualTo(expectedOrder);
 	}
 
 	private void assertApplicable(String message, ControllerAdviceBean controllerAdvice, Class<?> controllerBeanType) {