test: Add tests for MistralAI retry logic and AOT native image support

Adds essential test coverage for production reliability and native image compatibility

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

Author: alxkm

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

@@ -178,6 +178,19 @@ public void mistralAiEmbeddingNonTransientError() {
 			.call(new org.springframework.ai.embedding.EmbeddingRequest(List.of("text1", "text2"), null)));
 	}
 
+	@Test
+	public void mistralAiChatMixedTransientAndNonTransientErrors() {
+		given(this.mistralAiApi.chatCompletionEntity(isA(ChatCompletionRequest.class)))
+			.willThrow(new TransientAiException("Transient Error"))
+			.willThrow(new RuntimeException("Non Transient Error"));
+
+		// Should fail immediately on non-transient error, no further retries
+		assertThrows(RuntimeException.class, () -> this.chatModel.call(new Prompt("text")));
+
+		// Should have 1 retry attempt before hitting non-transient error
+		assertThat(this.retryListener.onErrorRetryCount).isEqualTo(2);
+	}
+
 	private static class TestRetryListener implements RetryListener {
 
 		int onErrorRetryCount = 0;

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

@@ -118,4 +118,60 @@ void verifyPackageScanningWorks() {
 		assertThat(jsonAnnotatedClasses.size()).isGreaterThan(0);
 	}
 
+	@Test
+	void verifyAllCriticalApiClassesAreRegistered() {
+		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()));
+
+		// Critical API classes that must be registered for runtime
+		String[] criticalClasses = { "MistralAiApi$ChatCompletionRequest", "MistralAiApi$ChatCompletionMessage",
+				"MistralAiApi$EmbeddingRequest", "MistralAiApi$EmbeddingList", "MistralAiApi$Usage" };
+
+		for (String className : criticalClasses) {
+			assertThat(registeredTypes.stream()
+				.anyMatch(tr -> tr.getName().contains(className.replace("$", "."))
+						|| tr.getName().contains(className.replace("$", "$"))))
+				.as("Critical class %s should be registered", className)
+				.isTrue();
+		}
+	}
+
+	@Test
+	void verifyEnumTypesAreRegistered() {
+		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()));
+
+		// Enums are critical for JSON deserialization in native images
+		assertThat(registeredTypes.contains(TypeReference.of(MistralAiApi.ChatModel.class)))
+			.as("ChatModel enum should be registered")
+			.isTrue();
+
+		assertThat(registeredTypes.contains(TypeReference.of(MistralAiApi.EmbeddingModel.class)))
+			.as("EmbeddingModel enum should be registered")
+			.isTrue();
+	}
+
+	@Test
+	void verifyReflectionHintsIncludeConstructors() {
+		RuntimeHints runtimeHints = new RuntimeHints();
+		MistralAiRuntimeHints mistralAiRuntimeHints = new MistralAiRuntimeHints();
+		mistralAiRuntimeHints.registerHints(runtimeHints, null);
+
+		// Verify that reflection hints include constructor access
+		boolean hasConstructorHints = runtimeHints.reflection()
+			.typeHints()
+			.anyMatch(typeHint -> typeHint.constructors().findAny().isPresent() || typeHint.getMemberCategories()
+				.contains(org.springframework.aot.hint.MemberCategory.INVOKE_DECLARED_CONSTRUCTORS));
+
+		assertThat(hasConstructorHints).as("Should register constructor hints for JSON deserialization").isTrue();
+	}
+
 }