Support "args" param in "@SpringBootTest"

Reflectively access `SpringBootTestArgs`(package private) to support
`args` param in `@SpringBootTest`.

Signed-off-by: Tadaya Tsuyukubo <tadaya@ttddyy.net>

Author: ttddyy

spring-native-configuration/src/main/java/org/springframework/boot/test/SpringBootTestHints.java

@@ -68,7 +68,8 @@
 		@TypeHint(types = org.springframework.web.context.ConfigurableWebApplicationContext.class, access = {}),
 		@TypeHint(types = org.springframework.web.servlet.DispatcherServlet.class, access = {}),
 		@TypeHint(types = org.springframework.web.reactive.DispatcherHandler.class, access = {}),
-		@TypeHint(typeNames = "org.glassfish.jersey.server.ResourceConfig", access = {})
+		@TypeHint(typeNames = "org.glassfish.jersey.server.ResourceConfig", access = {}),
+		@TypeHint(typeNames = "org.springframework.boot.test.context.SpringBootTestArgs", access = {TypeAccess.DECLARED_METHODS})
 })
 public class SpringBootTestHints implements NativeConfiguration {
 }

spring-native/src/main/java/org/springframework/aot/test/boot/AotSpringBootConfigContextLoader.java

@@ -16,9 +16,11 @@
 
 package org.springframework.aot.test.boot;
 
+import java.lang.reflect.Method;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
+import java.util.Set;
 
 import org.springframework.aot.SpringApplicationAotUtils;
 import org.springframework.boot.ApplicationContextFactory;
@@ -33,11 +35,15 @@
 import org.springframework.context.ApplicationContextInitializer;
 import org.springframework.context.ConfigurableApplicationContext;
 import org.springframework.core.env.ConfigurableEnvironment;
+import org.springframework.test.context.ContextCustomizer;
 import org.springframework.test.context.MergedContextConfiguration;
 import org.springframework.test.context.SmartContextLoader;
 import org.springframework.test.context.support.TestPropertySourceUtils;
 import org.springframework.test.context.web.WebMergedContextConfiguration;
+import org.springframework.util.Assert;
+import org.springframework.util.ClassUtils;
 import org.springframework.util.ObjectUtils;
+import org.springframework.util.ReflectionUtils;
 import org.springframework.web.context.support.GenericWebApplicationContext;
 
 /**
@@ -49,6 +55,23 @@
  */
 public class AotSpringBootConfigContextLoader extends SpringBootContextLoader {
 
+	// "SpringBootTestArgs#get(Set)" static method
+	private static final Method SPRING_BOOT_TEST_ARGS__GET;
+
+	static {
+		Class<?> clazz;
+		try {
+			// SpringBootTestArgs is a package private class
+			clazz = ClassUtils.forName("org.springframework.boot.test.context.SpringBootTestArgs", null);
+		} catch (ClassNotFoundException ex) {
+			throw new IllegalStateException("Failed to load SpringBootTestArgs class", ex);
+		}
+		Method method = ReflectionUtils.findMethod(clazz, "get", Set.class);
+		Assert.notNull(method, "SpringBootTestArgs#get(Set) method must exist");
+		ReflectionUtils.makeAccessible(method);
+		SPRING_BOOT_TEST_ARGS__GET = method;
+	}
+
 	private final Class<? extends ApplicationContextInitializer<?>> testContextInitializer;
 
 	private final WebApplicationType webApplicationType;
@@ -80,9 +103,6 @@ public AotSpringBootConfigContextLoader(Class<? extends ApplicationContextInitia
 
 	@Override
 	public ConfigurableApplicationContext loadContext(MergedContextConfiguration config) {
-		// TODO: handle application arguments
-		String[] args = new String[0];
-
 		SpringApplication application = new AotTestSpringApplication(config.getTestClass().getClassLoader(), testContextInitializer);
 		application.setMainApplicationClass(config.getTestClass());
 		application.setSources(Collections.singleton(testContextInitializer.getName()));
@@ -108,6 +128,8 @@ else if (this.webApplicationType == WebApplicationType.REACTIVE) {
 						ApplicationContextFactory.of(GenericReactiveWebApplicationContext::new));
 			}
 		}
+
+		String[] args = resolveArguments(config.getContextCustomizers());
 		ConfigurableApplicationContext context = application.run(args);
 
 		return context;
@@ -128,6 +150,10 @@ private void setActiveProfiles(ConfigurableEnvironment environment, String[] pro
 		TestPropertyValues.of(pairs).applyTo(environment);
 	}
 
+	private String[] resolveArguments(Set<ContextCustomizer> customizers) {
+		return (String[]) ReflectionUtils.invokeMethod(SPRING_BOOT_TEST_ARGS__GET, null, customizers);
+	}
+
 	private static class WebConfigurer {
 
 		void configure(MergedContextConfiguration configuration, SpringApplication application,

spring-native/src/test/java/org/springframework/aot/test/boot/AotSpringBootConfigContextLoaderTests.java

@@ -22,6 +22,7 @@
 import org.junit.jupiter.api.Test;
 
 import org.springframework.beans.factory.support.BeanDefinitionBuilder;
+import org.springframework.boot.ApplicationArguments;
 import org.springframework.boot.SpringBootConfiguration;
 import org.springframework.boot.WebApplicationType;
 import org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest;
@@ -84,6 +85,16 @@ void loadContextSetActiveProfiles() {
 				assertThat(context.getEnvironment().getActiveProfiles()).containsOnly("profile1", "profile2"));
 	}
 
+	@Test
+	void loadContextUseTestArguments() {
+		AotSpringBootConfigContextLoader loader = new AotSpringBootConfigContextLoader(TestApplicationContextInitializer.class);
+		run(() -> loader.loadContext(createMergedContextConfiguration(SampleArgumentsTest.class)), (context) -> {
+			ApplicationArguments args = context.getBean(ApplicationArguments.class);
+			assertThat(args.getOptionNames()).containsOnly("app.test");
+			assertThat(args.getOptionValues("app.test")).containsOnly("one");
+		});
+	}
+
 	private void run(Supplier<ConfigurableApplicationContext> supplier, Consumer<AssertableApplicationContext> context) {
 		try (ConfigurableApplicationContext ctx = supplier.get()) {
 			context.accept(AssertableApplicationContext.get(() -> ctx));
@@ -108,6 +119,11 @@ static class SampleProfilesTest {
 
 	}
 
+	@SpringBootTest(args = "--app.test=one")
+	static class SampleArgumentsTest {
+
+	}
+
 	@SpringBootConfiguration
 	static class SampleConfiguration {