Merge pull request #494 from harawata/fix-spring-boot-starter-475

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

Author: harawata

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

@@ -57,6 +57,9 @@ public class ClassPathMapperScanner extends ClassPathBeanDefinitionScanner {
 
   private static final Logger LOGGER = LoggerFactory.getLogger(ClassPathMapperScanner.class);
 
+  // Copy of FactoryBean#OBJECT_TYPE_ATTRIBUTE which was added in Spring 5.2
+  static final String FACTORY_BEAN_OBJECT_TYPE = "factoryBeanObjectType";
+
   private boolean addToConfig = true;
 
   private boolean lazyInitialization;
@@ -234,6 +237,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(FACTORY_BEAN_OBJECT_TYPE, beanClassName);
+
       boolean explicitFactoryUsed = false;
       if (StringUtils.hasText(this.sqlSessionFactoryBeanName)) {
         definition.getPropertyValues().add("sqlSessionFactory",

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

@@ -385,6 +385,14 @@ void testScanWithMapperFactoryBeanClass() {
     assertTrue(DummyMapperFactoryBean.getMapperCount() > 0);
   }
 
+  @Test
+  void testMapperBeanAttribute() {
+    startContext();
+
+    assertThat(applicationContext.getBeanDefinition("annotatedMapper").getAttribute(ClassPathMapperScanner.FACTORY_BEAN_OBJECT_TYPE))
+        .isEqualTo(AnnotatedMapper.class.getName());
+  }
+
   private void setupSqlSessionFactory(String name) {
     GenericBeanDefinition definition = new GenericBeanDefinition();
     definition.setBeanClass(SqlSessionFactoryBean.class);