Set attribute that helps MockitoPostProcessor replace a MyBatis mapper bean with a mock

mybatis/spring-boot-starter#475
spring-projects/spring-boot@f4985ab

Author: harawata

src/main/java/org/mybatis/spring/mapper/ClassPathMapperScanner.java

@@ -26,6 +26,7 @@
 import org.mybatis.spring.SqlSessionTemplate;
 import org.springframework.aop.scope.ScopedProxyFactoryBean;
 import org.springframework.aop.scope.ScopedProxyUtils;
+import org.springframework.beans.factory.FactoryBean;
 import org.springframework.beans.factory.annotation.AnnotatedBeanDefinition;
 import org.springframework.beans.factory.config.BeanDefinition;
 import org.springframework.beans.factory.config.BeanDefinitionHolder;
@@ -234,6 +235,10 @@ private void processBeanDefinitions(Set<BeanDefinitionHolder> beanDefinitions) {
 
       definition.getPropertyValues().add("addToConfig", this.addToConfig);
 
+      // Attribute for MockitoPostProcessor
+      // https://github.com/mybatis/spring-boot-starter/issues/475
+      definition.setAttribute(FactoryBean.OBJECT_TYPE_ATTRIBUTE, beanClassName);
+
       boolean explicitFactoryUsed = false;
       if (StringUtils.hasText(this.sqlSessionFactoryBeanName)) {
         definition.getPropertyValues().add("sqlSessionFactory",

src/test/java/org/mybatis/spring/mapper/MapperScannerConfigurerTest.java

@@ -37,6 +37,7 @@
 import org.mybatis.spring.SqlSessionTemplate;
 import org.mybatis.spring.mapper.child.MapperChildInterface;
 import org.mybatis.spring.type.DummyMapperFactoryBean;
+import org.springframework.beans.factory.FactoryBean;
 import org.springframework.beans.factory.NoSuchBeanDefinitionException;
 import org.springframework.beans.factory.config.BeanDefinition;
 import org.springframework.beans.factory.config.ConstructorArgumentValues;
@@ -385,6 +386,14 @@ void testScanWithMapperFactoryBeanClass() {
     assertTrue(DummyMapperFactoryBean.getMapperCount() > 0);
   }
 
+  @Test
+  void testMapperBeanAttribute() {
+    startContext();
+
+    assertThat(applicationContext.getBeanDefinition("annotatedMapper").getAttribute(FactoryBean.OBJECT_TYPE_ATTRIBUTE))
+        .isEqualTo(AnnotatedMapper.class.getName());
+  }
+
   private void setupSqlSessionFactory(String name) {
     GenericBeanDefinition definition = new GenericBeanDefinition();
     definition.setBeanClass(SqlSessionFactoryBean.class);