test: Enhanced test coverage for AnthropicRuntimeHintsTests (#3992)

Signed-off-by: Alex Klimenko <[email protected]>

Author: alxkm

models/spring-ai-anthropic/src/test/java/org/springframework/ai/anthropic/aot/AnthropicRuntimeHintsTests.java

@@ -20,20 +20,30 @@
 import java.util.Set;
 
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.BeforeEach;
 
 import org.springframework.ai.anthropic.api.AnthropicApi;
 import org.springframework.aot.hint.RuntimeHints;
 import org.springframework.aot.hint.TypeReference;
+import org.springframework.aot.hint.MemberCategory;
 
 import static org.assertj.core.api.AssertionsForClassTypes.assertThat;
 import static org.springframework.ai.aot.AiRuntimeHints.findJsonAnnotatedClassesInPackage;
 
 class AnthropicRuntimeHintsTests {
 
+	private RuntimeHints runtimeHints;
+
+	private AnthropicRuntimeHints anthropicRuntimeHints;
+
+	@BeforeEach
+	void setUp() {
+		runtimeHints = new RuntimeHints();
+		anthropicRuntimeHints = new AnthropicRuntimeHints();
+	}
+
 	@Test
 	void registerHints() {
-		RuntimeHints runtimeHints = new RuntimeHints();
-		AnthropicRuntimeHints anthropicRuntimeHints = new AnthropicRuntimeHints();
 		anthropicRuntimeHints.registerHints(runtimeHints, null);
 
 		Set<TypeReference> jsonAnnotatedClasses = findJsonAnnotatedClassesInPackage("org.springframework.ai.anthropic");
@@ -55,4 +65,84 @@ void registerHints() {
 		assertThat(registeredTypes.contains(TypeReference.of(AnthropicApi.AnthropicMessage.class))).isTrue();
 	}
 
+	@Test
+	void registerHintsWithNullClassLoader() {
+		// Test that registering hints with null ClassLoader works correctly
+		anthropicRuntimeHints.registerHints(runtimeHints, null);
+
+		Set<TypeReference> registeredTypes = new HashSet<>();
+		runtimeHints.reflection().typeHints().forEach(typeHint -> registeredTypes.add(typeHint.getType()));
+
+		assertThat(registeredTypes.size()).isGreaterThan(0);
+	}
+
+	@Test
+	void registerHintsWithCustomClassLoader() {
+		// Test that registering hints with a custom ClassLoader works correctly
+		ClassLoader customClassLoader = Thread.currentThread().getContextClassLoader();
+		anthropicRuntimeHints.registerHints(runtimeHints, customClassLoader);
+
+		Set<TypeReference> registeredTypes = new HashSet<>();
+		runtimeHints.reflection().typeHints().forEach(typeHint -> registeredTypes.add(typeHint.getType()));
+
+		assertThat(registeredTypes.size()).isGreaterThan(0);
+	}
+
+	@Test
+	void allMemberCategoriesAreRegistered() {
+		anthropicRuntimeHints.registerHints(runtimeHints, null);
+
+		Set<TypeReference> jsonAnnotatedClasses = findJsonAnnotatedClassesInPackage("org.springframework.ai.anthropic");
+
+		// Verify that all MemberCategory values are registered for each type
+		runtimeHints.reflection().typeHints().forEach(typeHint -> {
+			if (jsonAnnotatedClasses.contains(typeHint.getType())) {
+				Set<MemberCategory> expectedCategories = Set.of(MemberCategory.values());
+				Set<MemberCategory> actualCategories = typeHint.getMemberCategories();
+				assertThat(actualCategories.containsAll(expectedCategories)).isTrue();
+			}
+		});
+	}
+
+	@Test
+	void emptyRuntimeHintsInitiallyContainsNoTypes() {
+		// Verify that fresh RuntimeHints instance contains no reflection hints
+		RuntimeHints emptyHints = new RuntimeHints();
+		Set<TypeReference> emptyRegisteredTypes = new HashSet<>();
+		emptyHints.reflection().typeHints().forEach(typeHint -> emptyRegisteredTypes.add(typeHint.getType()));
+
+		assertThat(emptyRegisteredTypes.size()).isEqualTo(0);
+	}
+
+	@Test
+	void multipleRegistrationCallsAreIdempotent() {
+		// Register hints multiple times and verify no duplicates
+		anthropicRuntimeHints.registerHints(runtimeHints, null);
+		int firstRegistrationCount = (int) runtimeHints.reflection().typeHints().count();
+
+		anthropicRuntimeHints.registerHints(runtimeHints, null);
+		int secondRegistrationCount = (int) runtimeHints.reflection().typeHints().count();
+
+		assertThat(firstRegistrationCount).isEqualTo(secondRegistrationCount);
+	}
+
+	@Test
+	void verifyJsonAnnotatedClassesInPackageIsNotEmpty() {
+		Set<TypeReference> jsonAnnotatedClasses = findJsonAnnotatedClassesInPackage("org.springframework.ai.anthropic");
+		assertThat(jsonAnnotatedClasses.isEmpty()).isFalse();
+	}
+
+	@Test
+	void verifyEnumTypesAreRegistered() {
+		anthropicRuntimeHints.registerHints(runtimeHints, null);
+
+		Set<TypeReference> registeredTypes = new HashSet<>();
+		runtimeHints.reflection().typeHints().forEach(typeHint -> registeredTypes.add(typeHint.getType()));
+
+		// Verify enum types are properly registered
+		assertThat(registeredTypes.contains(TypeReference.of(AnthropicApi.Role.class))).isTrue();
+		assertThat(registeredTypes.contains(TypeReference.of(AnthropicApi.ThinkingType.class))).isTrue();
+		assertThat(registeredTypes.contains(TypeReference.of(AnthropicApi.EventType.class))).isTrue();
+	}
+
 }