test: Add comprehensive test coverage for runtime hints and bindings processors (#4402)

* test: Add comprehensive test coverage for runtime hints and bindings processors

Co-authored-by: Oleksandr Klymenko <[email protected]>
Signed-off-by: Oleksandr Klymenko <[email protected]>

Author: alxkm

models/spring-ai-azure-openai/src/test/java/org/springframework/ai/azure/openai/aot/AzureOpenAiRuntimeHintsTests.java

@@ -169,4 +169,44 @@ void verifyClientTypesAreRegistered() {
 		assertThat(this.runtimeHints).matches(reflection().onType(OpenAIAsyncClient.class));
 	}
 
+	@Test
+	void verifyNoSerializationHintsAreRegistered() {
+		this.azureOpenAiRuntimeHints.registerHints(this.runtimeHints, null);
+
+		// Azure OpenAI should only register reflection and resource hints, not
+		// serialization hints
+		assertThat(this.runtimeHints.serialization().javaSerializationHints().count()).isEqualTo(0);
+	}
+
+	@Test
+	void verifyRegistrationWithDifferentRuntimeHintsInstances() {
+		RuntimeHints hints1 = new RuntimeHints();
+		RuntimeHints hints2 = new RuntimeHints();
+
+		this.azureOpenAiRuntimeHints.registerHints(hints1, null);
+		this.azureOpenAiRuntimeHints.registerHints(hints2, null);
+
+		// Both instances should have same number of reflection hints
+		long count1 = hints1.reflection().typeHints().count();
+		long count2 = hints2.reflection().typeHints().count();
+
+		assertThat(count1).isEqualTo(count2);
+		assertThat(count1).isGreaterThan(0);
+	}
+
+	@Test
+	void verifyEnumTypesInAzurePackageAreRegistered() {
+		this.azureOpenAiRuntimeHints.registerHints(this.runtimeHints, null);
+
+		Set<TypeReference> registeredTypes = new HashSet<>();
+		this.runtimeHints.reflection().typeHints().forEach(typeHint -> registeredTypes.add(typeHint.getType()));
+
+		// Verify that enum types from Azure OpenAI package are registered
+		boolean hasEnumTypes = registeredTypes.stream()
+			.anyMatch(tr -> tr.getName().contains("com.azure.ai.openai.models")
+					&& tr.getName().toLowerCase().contains("choice"));
+
+		assertThat(hasEnumTypes).as("Azure OpenAI enum types should be registered").isTrue();
+	}
+
 }

models/spring-ai-mistral-ai/src/test/java/org/springframework/ai/mistralai/aot/MistralAiRuntimeHintsTests.java

@@ -28,6 +28,7 @@
 import org.springframework.aot.hint.TypeReference;
 
 import static org.assertj.core.api.AssertionsForClassTypes.assertThat;
+import static org.assertj.core.api.AssertionsForClassTypes.assertThatCode;
 import static org.springframework.ai.aot.AiRuntimeHints.findJsonAnnotatedClassesInPackage;
 
 class MistralAiRuntimeHintsTests {
@@ -174,4 +175,71 @@ void verifyReflectionHintsIncludeConstructors() {
 		assertThat(hasConstructorHints).as("Should register constructor hints for JSON deserialization").isTrue();
 	}
 
+	@Test
+	void verifyNoExceptionThrownWithEmptyRuntimeHints() {
+		RuntimeHints emptyRuntimeHints = new RuntimeHints();
+		MistralAiRuntimeHints mistralAiRuntimeHints = new MistralAiRuntimeHints();
+
+		// Should not throw any exception even with empty runtime hints
+		assertThatCode(() -> mistralAiRuntimeHints.registerHints(emptyRuntimeHints, null)).doesNotThrowAnyException();
+
+		assertThat(emptyRuntimeHints.reflection().typeHints().count()).isGreaterThan(0);
+	}
+
+	@Test
+	void verifyProxyHintsAreNotRegistered() {
+		RuntimeHints runtimeHints = new RuntimeHints();
+		MistralAiRuntimeHints mistralAiRuntimeHints = new MistralAiRuntimeHints();
+		mistralAiRuntimeHints.registerHints(runtimeHints, null);
+
+		// MistralAi should only register reflection hints, not proxy hints
+		assertThat(runtimeHints.proxies().jdkProxyHints().count()).isEqualTo(0);
+	}
+
+	@Test
+	void verifySerializationHintsAreNotRegistered() {
+		RuntimeHints runtimeHints = new RuntimeHints();
+		MistralAiRuntimeHints mistralAiRuntimeHints = new MistralAiRuntimeHints();
+		mistralAiRuntimeHints.registerHints(runtimeHints, null);
+
+		// MistralAi should only register reflection hints, not serialization hints
+		assertThat(runtimeHints.serialization().javaSerializationHints().count()).isEqualTo(0);
+	}
+
+	@Test
+	void verifyResponseTypesAreRegistered() {
+		RuntimeHints runtimeHints = new RuntimeHints();
+		MistralAiRuntimeHints mistralAiRuntimeHints = new MistralAiRuntimeHints();
+		mistralAiRuntimeHints.registerHints(runtimeHints, null);
+
+		Set<TypeReference> registeredTypes = new HashSet<>();
+		runtimeHints.reflection().typeHints().forEach(typeHint -> registeredTypes.add(typeHint.getType()));
+
+		// Verify response wrapper types are registered
+		assertThat(registeredTypes.stream().anyMatch(tr -> tr.getName().contains("EmbeddingList")))
+			.as("EmbeddingList response type should be registered")
+			.isTrue();
+
+		assertThat(registeredTypes.stream().anyMatch(tr -> tr.getName().contains("ChatCompletion")))
+			.as("ChatCompletion response type should be registered")
+			.isTrue();
+	}
+
+	@Test
+	void verifyMultipleInstancesRegisterSameHints() {
+		RuntimeHints runtimeHints1 = new RuntimeHints();
+		RuntimeHints runtimeHints2 = new RuntimeHints();
+
+		MistralAiRuntimeHints hints1 = new MistralAiRuntimeHints();
+		MistralAiRuntimeHints hints2 = new MistralAiRuntimeHints();
+
+		hints1.registerHints(runtimeHints1, null);
+		hints2.registerHints(runtimeHints2, null);
+
+		long count1 = runtimeHints1.reflection().typeHints().count();
+		long count2 = runtimeHints2.reflection().typeHints().count();
+
+		assertThat(count1).isEqualTo(count2);
+	}
+
 }

spring-ai-spring-cloud-bindings/src/test/java/org/springframework/ai/bindings/MistralAiBindingsPropertiesProcessorTests.java

@@ -120,4 +120,35 @@ void wrongBindingTypeShouldBeIgnored() {
 		assertThat(this.properties).isEmpty();
 	}
 
+	@Test
+	void emptyBindingsShouldNotThrowException() {
+		Bindings emptyBindings = new Bindings();
+
+		new MistralAiBindingsPropertiesProcessor().process(this.environment, emptyBindings, this.properties);
+
+		assertThat(this.properties).isEmpty();
+	}
+
+	@Test
+	void onlyUriWithoutApiKeyShouldSetBothProperties() {
+		Bindings bindingsWithOnlyUri = new Bindings(new Binding("test-name", Paths.get("test-path"), Map
+			.of(Binding.TYPE, MistralAiBindingsPropertiesProcessor.TYPE, "uri", "https://custom.mistralai.com")));
+
+		new MistralAiBindingsPropertiesProcessor().process(this.environment, bindingsWithOnlyUri, this.properties);
+
+		assertThat(this.properties).containsEntry("spring.ai.mistralai.base-url", "https://custom.mistralai.com");
+		assertThat(this.properties).containsEntry("spring.ai.mistralai.api-key", null);
+	}
+
+	@Test
+	void onlyApiKeyWithoutUriShouldSetBothProperties() {
+		Bindings bindingsWithOnlyApiKey = new Bindings(new Binding("test-name", Paths.get("test-path"),
+				Map.of(Binding.TYPE, MistralAiBindingsPropertiesProcessor.TYPE, "api-key", "secret-key")));
+
+		new MistralAiBindingsPropertiesProcessor().process(this.environment, bindingsWithOnlyApiKey, this.properties);
+
+		assertThat(this.properties).containsEntry("spring.ai.mistralai.api-key", "secret-key");
+		assertThat(this.properties).containsEntry("spring.ai.mistralai.base-url", null);
+	}
+
 }