Browse Source

Polishing.

Reformat code.

See #3414
Original pull request: #3415
pull/3426/head
Mark Paluch 3 weeks ago
parent
commit
12d78b7550
No known key found for this signature in database
GPG Key ID: 55BC6374BAA9D973
  1. 16
      src/main/java/org/springframework/data/aot/ManagedTypesBeanRegistrationAotProcessor.java
  2. 15
      src/test/java/org/springframework/data/aot/ManagedTypesBeanRegistrationAotProcessorUnitTests.java

16
src/main/java/org/springframework/data/aot/ManagedTypesBeanRegistrationAotProcessor.java

@ -54,9 +54,11 @@ import org.springframework.util.StringUtils; @@ -54,9 +54,11 @@ import org.springframework.util.StringUtils;
*/
public class ManagedTypesBeanRegistrationAotProcessor implements BeanRegistrationAotProcessor, EnvironmentAware {
private static final Lazy<Environment> DEFAULT_ENVIRONMENT = Lazy.of(StandardEnvironment::new);
private final Log logger = LogFactory.getLog(getClass());
private @Nullable String moduleIdentifier;
private static final Lazy<Environment> DEFAULT_ENVIRONMENT = Lazy.of(StandardEnvironment::new);
private @Nullable Environment environment = null;
public void setModuleIdentifier(@Nullable String moduleIdentifier) {
@ -195,12 +197,14 @@ public class ManagedTypesBeanRegistrationAotProcessor implements BeanRegistratio @@ -195,12 +197,14 @@ public class ManagedTypesBeanRegistrationAotProcessor implements BeanRegistratio
if (registeredBean.getBeanFactory() instanceof EnvironmentCapable ec) {
return ec.getEnvironment();
} else {
String[] beanNamesForType = registeredBean.getBeanFactory().getBeanNamesForType(Environment.class);
if (!ObjectUtils.isEmpty(beanNamesForType)) {
return registeredBean.getBeanFactory().getBean(beanNamesForType[0], Environment.class);
}
}
String[] beanNamesForType = registeredBean.getBeanFactory().getBeanNamesForType(Environment.class);
if (!ObjectUtils.isEmpty(beanNamesForType)) {
return registeredBean.getBeanFactory().getBean(beanNamesForType[0], Environment.class);
}
return DEFAULT_ENVIRONMENT.get();
}
}

15
src/test/java/org/springframework/data/aot/ManagedTypesBeanRegistrationAotProcessorUnitTests.java

@ -16,7 +16,6 @@ @@ -16,7 +16,6 @@
package org.springframework.data.aot;
import static org.assertj.core.api.Assertions.*;
import static org.mockito.ArgumentMatchers.*;
import static org.mockito.Mockito.*;
import java.util.Arrays;
@ -28,6 +27,7 @@ import java.util.function.Consumer; @@ -28,6 +27,7 @@ import java.util.function.Consumer;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
import org.springframework.aot.generate.GenerationContext;
import org.springframework.aot.hint.predicate.RuntimeHintsPredicates;
import org.springframework.aot.test.generate.TestGenerationContext;
@ -91,7 +91,7 @@ class ManagedTypesBeanRegistrationAotProcessorUnitTests { @@ -91,7 +91,7 @@ class ManagedTypesBeanRegistrationAotProcessorUnitTests {
createPostProcessor("commons").processAheadOfTime(RegisteredBean.of(beanFactory, "commons.managed-types"));
verify(beanFactory, never()).getBean(eq("commons.managed-types"), eq(ManagedTypes.class));
verify(beanFactory, never()).getBean("commons.managed-types", ManagedTypes.class);
}
@Test // GH-2593
@ -128,7 +128,7 @@ class ManagedTypesBeanRegistrationAotProcessorUnitTests { @@ -128,7 +128,7 @@ class ManagedTypesBeanRegistrationAotProcessorUnitTests {
createPostProcessor("commons").processAheadOfTime(RegisteredBean.of(beanFactory, "commons.managed-types"));
verify(beanFactory).getBean(eq("commons.managed-types"), eq(ManagedTypes.class));
verify(beanFactory).getBean("commons.managed-types", ManagedTypes.class);
}
@Test // GH-2593
@ -157,8 +157,7 @@ class ManagedTypesBeanRegistrationAotProcessorUnitTests { @@ -157,8 +157,7 @@ class ManagedTypesBeanRegistrationAotProcessorUnitTests {
@Test // GH-2593
void returnsEmptyContributionWhenBeanCannotBeLoaded() {
doThrow(new BeanCreationException("o_O")).when(beanFactory).getBean(eq("commons.managed-types"),
eq(ManagedTypes.class));
doThrow(new BeanCreationException("o_O")).when(beanFactory).getBean("commons.managed-types", ManagedTypes.class);
beanFactory.registerBeanDefinition("commons.managed-types", myManagedTypesDefinition);
@ -170,7 +169,7 @@ class ManagedTypesBeanRegistrationAotProcessorUnitTests { @@ -170,7 +169,7 @@ class ManagedTypesBeanRegistrationAotProcessorUnitTests {
contribution.applyTo(generationContext, null);
assertThat(generationContext.getRuntimeHints().reflection().typeHints()).isEmpty();
verify(beanFactory).getBean(eq("commons.managed-types"), eq(ManagedTypes.class));
verify(beanFactory).getBean("commons.managed-types", ManagedTypes.class);
}
@Test // GH-2680
@ -282,7 +281,7 @@ class ManagedTypesBeanRegistrationAotProcessorUnitTests { @@ -282,7 +281,7 @@ class ManagedTypesBeanRegistrationAotProcessorUnitTests {
contribution.applyTo(new TestGenerationContext(Object.class), null);
verify(env).getProperty(eq("spring.aot.data.accessors.enabled"), eq(Boolean.class), eq(true));
verify(env).getProperty("spring.aot.data.accessors.enabled", Boolean.class, true);
}
@Test // GH-3414
@ -299,7 +298,7 @@ class ManagedTypesBeanRegistrationAotProcessorUnitTests { @@ -299,7 +298,7 @@ class ManagedTypesBeanRegistrationAotProcessorUnitTests {
contribution.applyTo(new TestGenerationContext(Object.class), null);
verify(env).getProperty(eq("spring.aot.data.accessors.enabled"), eq(Boolean.class), eq(true));
verify(env).getProperty("spring.aot.data.accessors.enabled", Boolean.class, true);
}
private ManagedTypesBeanRegistrationAotProcessor createPostProcessor(String moduleIdentifier) {

Loading…
Cancel
Save