Improve @RegisterReflectionForBinding for enums

Closes spring-projectsgh-31570
wso2 · Dec 5, 2023 · 3f9a809 · 3f9a809
1 parent c74d60b
commit 3f9a809
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/spring-core/src/main/java/org/springframework/aot/hint/BindingReflectionHintsRegistrar.java b/spring-core/src/main/java/org/springframework/aot/hint/BindingReflectionHintsRegistrar.java
@@ -94,6 +94,11 @@ private void registerReflectionHints(ReflectionHints hints, Set<Type> seen, Type
 							registerRecordHints(hints, seen, recordComponent.getAccessor());
 						}
 					}
+					if (clazz.isEnum()) {
+						typeHint.withMembers(
+								MemberCategory.INVOKE_PUBLIC_CONSTRUCTORS,
+								MemberCategory.INVOKE_PUBLIC_METHODS);
+					}
 					typeHint.withMembers(
 							MemberCategory.DECLARED_FIELDS,
 							MemberCategory.INVOKE_DECLARED_CONSTRUCTORS);

diff --git a/...core/src/test/java/org/springframework/aot/hint/BindingReflectionHintsRegistrarTests.java b/...core/src/test/java/org/springframework/aot/hint/BindingReflectionHintsRegistrarTests.java
@@ -229,8 +229,8 @@ void registerTypeForSerializationWithMultipleLevelsAndCollection() {
 	@Test
 	void registerTypeForSerializationWithEnum() {
 		bindingRegistrar.registerReflectionHints(this.hints.reflection(), SampleEnum.class);
-		assertThat(this.hints.reflection().typeHints()).singleElement()
-				.satisfies(typeHint -> assertThat(typeHint.getType()).isEqualTo(TypeReference.of(SampleEnum.class)));
+		assertThat(RuntimeHintsPredicates.reflection().onType(SampleEnum.class).withMemberCategories(
+				MemberCategory.INVOKE_PUBLIC_CONSTRUCTORS, MemberCategory.INVOKE_PUBLIC_METHODS)).accepts(this.hints);
 	}
 
 	@Test