diff --git a/spring-context/src/test/java/org/springframework/context/annotation/ConfigurationClassPostProcessorAotContributionTests.java b/spring-context/src/test/java/org/springframework/context/annotation/ConfigurationClassPostProcessorAotContributionTests.java index e8acdc21791..bae0835fc5c 100644 --- a/spring-context/src/test/java/org/springframework/context/annotation/ConfigurationClassPostProcessorAotContributionTests.java +++ b/spring-context/src/test/java/org/springframework/context/annotation/ConfigurationClassPostProcessorAotContributionTests.java @@ -63,16 +63,12 @@ import static org.assertj.core.api.Assertions.entry; */ class ConfigurationClassPostProcessorAotContributionTests { - private final TestGenerationContext generationContext; + private final TestGenerationContext generationContext = new TestGenerationContext(); - private final MockBeanFactoryInitializationCode beanFactoryInitializationCode; + private final MockBeanFactoryInitializationCode beanFactoryInitializationCode = + new MockBeanFactoryInitializationCode(this.generationContext); - ConfigurationClassPostProcessorAotContributionTests() { - this.generationContext = new TestGenerationContext(); - this.beanFactoryInitializationCode = new MockBeanFactoryInitializationCode(this.generationContext); - } - @Test void processAheadOfTimeWhenNoImportAwareConfigurationReturnsNull() { assertThat(getContribution(SimpleConfiguration.class)).isNull(); @@ -80,8 +76,7 @@ class ConfigurationClassPostProcessorAotContributionTests { @Test void applyToWhenHasImportAwareConfigurationRegistersBeanPostProcessorWithMapEntry() { - BeanFactoryInitializationAotContribution contribution = getContribution( - ImportConfiguration.class); + BeanFactoryInitializationAotContribution contribution = getContribution(ImportConfiguration.class); contribution.applyTo(this.generationContext, this.beanFactoryInitializationCode); compile((initializer, compiled) -> { GenericApplicationContext freshContext = new GenericApplicationContext(); @@ -97,8 +92,7 @@ class ConfigurationClassPostProcessorAotContributionTests { @Test void applyToWhenHasImportAwareConfigurationRegistersBeanPostProcessorAfterApplicationContextAwareProcessor() { - BeanFactoryInitializationAotContribution contribution = getContribution( - TestAwareCallbackConfiguration.class); + BeanFactoryInitializationAotContribution contribution = getContribution(TestAwareCallbackConfiguration.class); contribution.applyTo(this.generationContext, this.beanFactoryInitializationCode); compile((initializer, compiled) -> { GenericApplicationContext freshContext = new GenericApplicationContext(); @@ -139,8 +133,7 @@ class ConfigurationClassPostProcessorAotContributionTests { @Test void applyToWhenHasImportAwareConfigurationRegistersHints() { - BeanFactoryInitializationAotContribution contribution = getContribution( - ImportConfiguration.class); + BeanFactoryInitializationAotContribution contribution = getContribution(ImportConfiguration.class); contribution.applyTo(this.generationContext, this.beanFactoryInitializationCode); assertThat(generationContext.getRuntimeHints().resources().resourcePatterns()) .singleElement() @@ -159,8 +152,7 @@ class ConfigurationClassPostProcessorAotContributionTests { return postProcessor.processAheadOfTime(beanFactory); } - private void assertPostProcessorEntry(BeanPostProcessor postProcessor, - Class key, Class value) { + private void assertPostProcessorEntry(BeanPostProcessor postProcessor, Class key, Class value) { assertThat(postProcessor).extracting("importsMapping") .asInstanceOf(InstanceOfAssertFactories.MAP) .containsExactly(entry(key.getName(), value.getName())); @@ -168,8 +160,7 @@ class ConfigurationClassPostProcessorAotContributionTests { @SuppressWarnings("unchecked") private void compile(BiConsumer, Compiled> result) { - MethodReference methodReference = this.beanFactoryInitializationCode - .getInitializers().get(0); + MethodReference methodReference = this.beanFactoryInitializationCode.getInitializers().get(0); this.beanFactoryInitializationCode.getTypeBuilder().set(type -> { type.addModifiers(Modifier.PUBLIC); type.addSuperinterface(ParameterizedTypeName.get(Consumer.class, DefaultListableBeanFactory.class)); @@ -183,10 +174,10 @@ class ConfigurationClassPostProcessorAotContributionTests { result.accept(compiled.getInstance(Consumer.class), compiled)); } + @Configuration(proxyBeanMethods = false) @Import(TestAwareCallbackBean.class) static class TestAwareCallbackConfiguration { - } static class TestAwareCallbackBean implements ImportAware, ApplicationContextAware { @@ -208,7 +199,6 @@ class ConfigurationClassPostProcessorAotContributionTests { @Configuration(proxyBeanMethods = false) @Import(TestImportAwareBeanPostProcessor.class) static class TestImportAwareBeanPostProcessorConfiguration { - } static class TestImportAwareBeanPostProcessor implements BeanPostProcessor, ImportAware,