Disable BatchTestContextCustomizer after AOT processing

Issue #4286

Author: Aleksandr Arshavskiy

spring-batch-test/src/main/java/org/springframework/batch/test/context/BatchTestContextCustomizer.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2018-2022 the original author or authors.
+ * Copyright 2018-2023 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -15,6 +15,7 @@
  */
 package org.springframework.batch.test.context;
 
+import org.springframework.aot.AotDetector;
 import org.springframework.batch.test.JobLauncherTestUtils;
 import org.springframework.batch.test.JobRepositoryTestUtils;
 import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
@@ -30,7 +31,7 @@
  * ({@link JobLauncherTestUtils} and {@link JobRepositoryTestUtils}) as beans in the test
  * context.
  *
- * @author Mahmoud Ben Hassine
+ * @author Mahmoud Ben Hassine, Alexander Arshavskiy
  * @since 4.1
  */
 public class BatchTestContextCustomizer implements ContextCustomizer {
@@ -43,6 +44,10 @@ public class BatchTestContextCustomizer implements ContextCustomizer {
 
 	@Override
 	public void customizeContext(ConfigurableApplicationContext context, MergedContextConfiguration mergedConfig) {
+		if (AotDetector.useGeneratedArtifacts()) {
+			return;
+		}
+
 		ConfigurableListableBeanFactory beanFactory = context.getBeanFactory();
 		Assert.isInstanceOf(BeanDefinitionRegistry.class, beanFactory,
 				"The bean factory must be an instance of BeanDefinitionRegistry");

spring-batch-test/src/test/java/org/springframework/batch/test/context/BatchTestContextCustomizerTests.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2018-2022 the original author or authors.
+ * Copyright 2018-2023 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -15,25 +15,33 @@
  */
 package org.springframework.batch.test.context;
 
+import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.Test;
 import org.mockito.Mockito;
 
 import org.springframework.context.ConfigurableApplicationContext;
 import org.springframework.context.support.GenericApplicationContext;
+import org.springframework.core.SpringProperties;
 import org.springframework.test.context.MergedContextConfiguration;
 
 import static org.hamcrest.CoreMatchers.containsString;
 import static org.hamcrest.MatcherAssert.assertThat;
+import static org.junit.jupiter.api.Assertions.assertFalse;
 import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
 /**
- * @author Mahmoud Ben Hassine
+ * @author Mahmoud Ben Hassine, Alexander Arshavskiy
  */
 class BatchTestContextCustomizerTests {
 
 	private final BatchTestContextCustomizer contextCustomizer = new BatchTestContextCustomizer();
 
+	@AfterEach
+	void removeSystemProperty() {
+		SpringProperties.setProperty("spring.aot.enabled", null);
+	}
+
 	@Test
 	void testCustomizeContext() {
 		// given
@@ -64,4 +72,20 @@ void testCustomizeContext_whenBeanFactoryIsNotAnInstanceOfBeanDefinitionRegistry
 				containsString("The bean factory must be an instance of BeanDefinitionRegistry"));
 	}
 
+	@Test
+	void testCustomizeContext_whenUsingAotGeneratedArtifactsBatchTestContextIsNotRegistered() {
+		// given
+		SpringProperties.setProperty("spring.aot.enabled", "true");
+		ConfigurableApplicationContext context = new GenericApplicationContext();
+		MergedContextConfiguration mergedConfig = Mockito.mock(MergedContextConfiguration.class);
+
+		// when
+		this.contextCustomizer.customizeContext(context, mergedConfig);
+
+		// then
+		assertFalse(context.containsBean("jobLauncherTestUtils"));
+		assertFalse(context.containsBean("jobRepositoryTestUtils"));
+		assertFalse(context.containsBean("batchTestContextBeanPostProcessor"));
+	}
+
 }