test: Add comprehensive test coverage for Azure OpenAI chat and embedding options

Enhances test coverage for AzureOpenAiChatOptions and AzureOpenAiEmbeddingModel with edge cases and robustness testing

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/AzureEmbeddingsOptionsTests.java

@@ -16,6 +16,7 @@
 
 package org.springframework.ai.azure.openai;
 
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
@@ -158,4 +159,56 @@ public void shouldHandleConcurrentRequests() {
 		assertThat(options2.getUser()).isEqualTo("USER2");
 	}
 
+	@Test
+	public void shouldHandleVeryLargeInputList() {
+		// Test with a large number of inputs to verify performance and memory handling
+		List<String> largeInputList = Collections.nCopies(1000, "Test input text");
+		var requestOptions = client.toEmbeddingOptions(new EmbeddingRequest(largeInputList, null));
+
+		assertThat(requestOptions.getInput()).hasSize(1000);
+		assertThat(requestOptions.getModel()).isEqualTo("DEFAULT_MODEL");
+		assertThat(requestOptions.getUser()).isEqualTo("USER_TEST");
+	}
+
+	@Test
+	public void shouldHandleEmptyStringInputs() {
+		List<String> inputsWithEmpty = Arrays.asList("", "Valid text", "", "Another valid text");
+		var requestOptions = client.toEmbeddingOptions(new EmbeddingRequest(inputsWithEmpty, null));
+
+		assertThat(requestOptions.getInput()).hasSize(4);
+		assertThat(requestOptions.getInput()).containsExactly("", "Valid text", "", "Another valid text");
+	}
+
+	@Test
+	public void shouldHandleDifferentClientConfigurations() {
+		var clientWithDifferentDefaults = new AzureOpenAiEmbeddingModel(mockClient, MetadataMode.EMBED,
+				AzureOpenAiEmbeddingOptions.builder().deploymentName("DIFFERENT_DEFAULT").build());
+
+		var requestOptions = clientWithDifferentDefaults
+			.toEmbeddingOptions(new EmbeddingRequest(List.of("Test content"), null));
+
+		assertThat(requestOptions.getModel()).isEqualTo("DIFFERENT_DEFAULT");
+		assertThat(requestOptions.getUser()).isNull(); // No default user set
+	}
+
+	@Test
+	public void shouldHandleWhitespaceOnlyInputs() {
+		List<String> whitespaceInputs = Arrays.asList("   ", "\t\t", "\n\n", "  valid text  ");
+		var requestOptions = client.toEmbeddingOptions(new EmbeddingRequest(whitespaceInputs, null));
+
+		assertThat(requestOptions.getInput()).hasSize(4);
+		assertThat(requestOptions.getInput()).containsExactlyElementsOf(whitespaceInputs);
+	}
+
+	@Test
+	public void shouldValidateInputListIsNotModified() {
+		List<String> originalInputs = Arrays.asList("Input 1", "Input 2", "Input 3");
+		List<String> inputsCopy = new ArrayList<>(originalInputs);
+
+		client.toEmbeddingOptions(new EmbeddingRequest(inputsCopy, null));
+
+		// Verify original list wasn't modified
+		assertThat(inputsCopy).isEqualTo(originalInputs);
+	}
+
 }

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

@@ -195,4 +195,49 @@ void testDefaultValues() {
 		assertThat(options.getModel()).isNull();
 	}
 
+	@Test
+	void testModelAndDeploymentNameRelationship() {
+		AzureOpenAiChatOptions options = new AzureOpenAiChatOptions();
+
+		// Test setting deployment name first
+		options.setDeploymentName("deployment-1");
+		assertThat(options.getDeploymentName()).isEqualTo("deployment-1");
+		assertThat(options.getModel()).isEqualTo("deployment-1");
+
+		// Test setting model overwrites deployment name
+		options.setModel("model-1");
+		assertThat(options.getDeploymentName()).isEqualTo("model-1");
+		assertThat(options.getModel()).isEqualTo("model-1");
+	}
+
+	@Test
+	void testResponseFormatVariations() {
+		// Test with JSON response format
+		AzureOpenAiResponseFormat jsonFormat = AzureOpenAiResponseFormat.builder()
+			.type(AzureOpenAiResponseFormat.Type.JSON_OBJECT)
+			.build();
+
+		AzureOpenAiChatOptions options = AzureOpenAiChatOptions.builder().responseFormat(jsonFormat).build();
+
+		assertThat(options.getResponseFormat()).isEqualTo(jsonFormat);
+		assertThat(options.getResponseFormat().getType()).isEqualTo(AzureOpenAiResponseFormat.Type.JSON_OBJECT);
+	}
+
+	@Test
+	void testEnhancementsConfiguration() {
+		AzureChatEnhancementConfiguration enhancements = new AzureChatEnhancementConfiguration();
+		AzureChatOCREnhancementConfiguration ocrConfig = new AzureChatOCREnhancementConfiguration(false);
+		AzureChatGroundingEnhancementConfiguration groundingConfig = new AzureChatGroundingEnhancementConfiguration(
+				false);
+
+		enhancements.setOcr(ocrConfig);
+		enhancements.setGrounding(groundingConfig);
+
+		AzureOpenAiChatOptions options = AzureOpenAiChatOptions.builder().enhancements(enhancements).build();
+
+		assertThat(options.getEnhancements()).isEqualTo(enhancements);
+		assertThat(options.getEnhancements().getOcr()).isEqualTo(ocrConfig);
+		assertThat(options.getEnhancements().getGrounding()).isEqualTo(groundingConfig);
+	}
+
 }