add new ApplicationContextRunner

Signed-off-by: Alexandros Pappas <[email protected]>

Author: apappascs

auto-configurations/models/spring-ai-autoconfigure-model-openai/src/test/java/org/springframework/ai/model/openai/autoconfigure/OpenAiImageAutoConfigurationIT.java

@@ -81,23 +81,15 @@ void generateImage() {
 
 	@Test
 	void imageModelDisabled() {
-		new ApplicationContextRunner()
-			.withPropertyValues("spring.ai.openai.apiKey=" + System.getenv("OPENAI_API_KEY"),
-					"spring.ai.model.image=none")
-			.withConfiguration(
-					AutoConfigurations.of(SpringAiRetryAutoConfiguration.class, RestClientAutoConfiguration.class,
-							WebClientAutoConfiguration.class, OpenAiImageAutoConfiguration.class))
+		this.contextRunner.withPropertyValues("spring.ai.model.image=none")
+			.withConfiguration(AutoConfigurations.of(OpenAiImageAutoConfiguration.class))
 			.run(context -> assertThat(context.getBeansOfType(OpenAiImageModel.class)).isEmpty());
 	}
 
 	@Test
 	void imageModelExplicitlyEnabled() {
-		new ApplicationContextRunner()
-			.withPropertyValues("spring.ai.openai.apiKey=" + System.getenv("OPENAI_API_KEY"),
-					"spring.ai.model.image=openai")
-			.withConfiguration(
-					AutoConfigurations.of(SpringAiRetryAutoConfiguration.class, RestClientAutoConfiguration.class,
-							WebClientAutoConfiguration.class, OpenAiImageAutoConfiguration.class))
+		this.contextRunner.withPropertyValues("spring.ai.model.image=openai")
+			.withConfiguration(AutoConfigurations.of(OpenAiImageAutoConfiguration.class))
 			.run(context -> assertThat(context.getBeansOfType(OpenAiImageModel.class)).isNotEmpty());
 	}