Set attribute that helps MockitoPostProcessor replace a MyBatis mappe…

…r bean with a mock mybatis/spring-boot-starter#475 spring-projects/spring-boot@f4985ab
harawata · Jul 11, 2020 · d007089 · d007089
1 parent 2af8093
commit d007089
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/main/java/org/mybatis/spring/mapper/ClassPathMapperScanner.java b/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",

diff --git a/src/test/java/org/mybatis/spring/mapper/MapperScannerConfigurerTest.java b/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);