Keep fallback flag in AOT-generated code

Closes gh-33261
spring-projects · Jul 23, 2024 · d219362 · d219362
1 parent eda21c4
commit d219362
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/...ain/java/org/springframework/beans/factory/aot/BeanDefinitionPropertiesCodeGenerator.java b/...ain/java/org/springframework/beans/factory/aot/BeanDefinitionPropertiesCodeGenerator.java
@@ -115,6 +115,8 @@ CodeBlock generateCode(RootBeanDefinition beanDefinition) {
 		CodeBlock.Builder code = CodeBlock.builder();
 		addStatementForValue(code, beanDefinition, BeanDefinition::isPrimary,
 				"$L.setPrimary($L)");
+		addStatementForValue(code, beanDefinition, BeanDefinition::isFallback,
+				"$L.setFallback($L)");
 		addStatementForValue(code, beanDefinition, BeanDefinition::getScope,
 				this::hasScope, "$L.setScope($S)");
 		addStatementForValue(code, beanDefinition, BeanDefinition::getDependsOn,

diff --git a/...ava/org/springframework/beans/factory/aot/BeanDefinitionPropertiesCodeGeneratorTests.java b/...ava/org/springframework/beans/factory/aot/BeanDefinitionPropertiesCodeGeneratorTests.java
@@ -92,6 +92,21 @@ void setPrimaryWhenTrue() {
 		compile((actual, compiled) -> assertThat(actual.isPrimary()).isTrue());
 	}
 
+	@Test
+	void setFallbackWhenFalse() {
+		this.beanDefinition.setFallback(false);
+		compile((actual, compiled) -> {
+			assertThat(compiled.getSourceFile()).doesNotContain("setFallback");
+			assertThat(actual.isFallback()).isFalse();
+		});
+	}
+
+	@Test
+	void setFallbackWhenTrue() {
+		this.beanDefinition.setFallback(true);
+		compile((actual, compiled) -> assertThat(actual.isFallback()).isTrue());
+	}
+
 	@Test
 	void setScopeWhenEmptyString() {
 		this.beanDefinition.setScope("");