Stephane Nicoll
2 years ago
parent
51df7813a5
commit
e2dd0ef9a4
@ -1,237 +0,0 @@
|
||||
/*
|
||||
* Copyright 2012-2022 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.boot;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.lang.reflect.Method;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Path;
|
||||
import java.nio.file.Paths;
|
||||
import java.util.ArrayList;
|
||||
import java.util.Arrays;
|
||||
import java.util.Collections;
|
||||
import java.util.List;
|
||||
|
||||
import org.springframework.aot.generate.ClassNameGenerator;
|
||||
import org.springframework.aot.generate.DefaultGenerationContext;
|
||||
import org.springframework.aot.generate.FileSystemGeneratedFiles;
|
||||
import org.springframework.aot.generate.GeneratedFiles.Kind;
|
||||
import org.springframework.aot.hint.ExecutableMode;
|
||||
import org.springframework.aot.hint.ReflectionHints;
|
||||
import org.springframework.aot.hint.RuntimeHints;
|
||||
import org.springframework.aot.hint.TypeReference;
|
||||
import org.springframework.aot.nativex.FileNativeConfigurationWriter;
|
||||
import org.springframework.boot.SpringApplication.AbandonedRunException;
|
||||
import org.springframework.context.ApplicationContext;
|
||||
import org.springframework.context.ConfigurableApplicationContext;
|
||||
import org.springframework.context.aot.ApplicationContextAotGenerator;
|
||||
import org.springframework.context.support.GenericApplicationContext;
|
||||
import org.springframework.javapoet.ClassName;
|
||||
import org.springframework.util.Assert;
|
||||
import org.springframework.util.FileSystemUtils;
|
||||
import org.springframework.util.ReflectionUtils;
|
||||
import org.springframework.util.function.ThrowingSupplier;
|
||||
|
||||
/**
|
||||
* Entry point for AOT processing of a {@link SpringApplication}.
|
||||
* <p>
|
||||
* <strong>For internal use only.</strong>
|
||||
*
|
||||
* @author Stephane Nicoll
|
||||
* @author Andy Wilkinson
|
||||
* @author Phillip Webb
|
||||
* @since 3.0.0
|
||||
*/
|
||||
public class AotProcessor {
|
||||
|
||||
private final Class<?> application;
|
||||
|
||||
private final String[] applicationArgs;
|
||||
|
||||
private final Path sourceOutput;
|
||||
|
||||
private final Path resourceOutput;
|
||||
|
||||
private final Path classOutput;
|
||||
|
||||
private final String groupId;
|
||||
|
||||
private final String artifactId;
|
||||
|
||||
/**
|
||||
* Create a new processor for the specified application and settings.
|
||||
* @param application the application main class
|
||||
* @param applicationArgs the arguments to provide to the main method
|
||||
* @param sourceOutput the location of generated sources
|
||||
* @param resourceOutput the location of generated resources
|
||||
* @param classOutput the location of generated classes
|
||||
* @param groupId the group ID of the application, used to locate
|
||||
* native-image.properties
|
||||
* @param artifactId the artifact ID of the application, used to locate
|
||||
* native-image.properties
|
||||
*/
|
||||
public AotProcessor(Class<?> application, String[] applicationArgs, Path sourceOutput, Path resourceOutput,
|
||||
Path classOutput, String groupId, String artifactId) {
|
||||
this.application = application;
|
||||
this.applicationArgs = applicationArgs;
|
||||
this.sourceOutput = sourceOutput;
|
||||
this.resourceOutput = resourceOutput;
|
||||
this.classOutput = classOutput;
|
||||
this.groupId = groupId;
|
||||
this.artifactId = artifactId;
|
||||
}
|
||||
|
||||
/**
|
||||
* Trigger the processing of the application managed by this instance.
|
||||
*/
|
||||
public void process() {
|
||||
deleteExistingOutput();
|
||||
GenericApplicationContext applicationContext = new AotProcessorHook().run(() -> {
|
||||
Method mainMethod = this.application.getMethod("main", String[].class);
|
||||
return ReflectionUtils.invokeMethod(mainMethod, null, new Object[] { this.applicationArgs });
|
||||
});
|
||||
performAotProcessing(applicationContext);
|
||||
}
|
||||
|
||||
private void deleteExistingOutput() {
|
||||
deleteExistingOutput(this.sourceOutput, this.resourceOutput, this.classOutput);
|
||||
}
|
||||
|
||||
private void deleteExistingOutput(Path... paths) {
|
||||
for (Path path : paths) {
|
||||
try {
|
||||
FileSystemUtils.deleteRecursively(path);
|
||||
}
|
||||
catch (IOException ex) {
|
||||
throw new RuntimeException("Failed to delete existing output in '" + path + "'");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private void performAotProcessing(GenericApplicationContext applicationContext) {
|
||||
FileSystemGeneratedFiles generatedFiles = new FileSystemGeneratedFiles(this::getRoot);
|
||||
DefaultGenerationContext generationContext = new DefaultGenerationContext(
|
||||
new ClassNameGenerator(ClassName.get(this.application)), generatedFiles);
|
||||
ApplicationContextAotGenerator generator = new ApplicationContextAotGenerator();
|
||||
ClassName generatedInitializerClassName = generator.processAheadOfTime(applicationContext, generationContext);
|
||||
registerEntryPointHint(generationContext, generatedInitializerClassName);
|
||||
generationContext.writeGeneratedContent();
|
||||
writeHints(generationContext.getRuntimeHints());
|
||||
writeNativeImageProperties();
|
||||
}
|
||||
|
||||
private void registerEntryPointHint(DefaultGenerationContext generationContext,
|
||||
ClassName generatedInitializerClassName) {
|
||||
TypeReference generatedType = TypeReference.of(generatedInitializerClassName.canonicalName());
|
||||
TypeReference applicationType = TypeReference.of(this.application);
|
||||
ReflectionHints reflection = generationContext.getRuntimeHints().reflection();
|
||||
reflection.registerType(applicationType);
|
||||
reflection.registerType(generatedType, (typeHint) -> typeHint.onReachableType(applicationType)
|
||||
.withConstructor(Collections.emptyList(), ExecutableMode.INVOKE));
|
||||
}
|
||||
|
||||
private Path getRoot(Kind kind) {
|
||||
return switch (kind) {
|
||||
case SOURCE -> this.sourceOutput;
|
||||
case RESOURCE -> this.resourceOutput;
|
||||
case CLASS -> this.classOutput;
|
||||
};
|
||||
}
|
||||
|
||||
private void writeHints(RuntimeHints hints) {
|
||||
FileNativeConfigurationWriter writer = new FileNativeConfigurationWriter(this.resourceOutput, this.groupId,
|
||||
this.artifactId);
|
||||
writer.write(hints);
|
||||
}
|
||||
|
||||
private void writeNativeImageProperties() {
|
||||
List<String> args = new ArrayList<>();
|
||||
args.add("-H:Class=" + this.application.getName());
|
||||
args.add("--report-unsupported-elements-at-runtime");
|
||||
args.add("--no-fallback");
|
||||
args.add("--install-exit-handlers");
|
||||
StringBuilder sb = new StringBuilder();
|
||||
sb.append("Args = ");
|
||||
sb.append(String.join(String.format(" \\%n"), args));
|
||||
Path file = this.resourceOutput
|
||||
.resolve("META-INF/native-image/" + this.groupId + "/" + this.artifactId + "/native-image.properties");
|
||||
try {
|
||||
if (!Files.exists(file)) {
|
||||
Files.createDirectories(file.getParent());
|
||||
Files.createFile(file);
|
||||
}
|
||||
Files.writeString(file, sb.toString());
|
||||
}
|
||||
catch (IOException ex) {
|
||||
throw new IllegalStateException("Failed to write native-image properties", ex);
|
||||
}
|
||||
}
|
||||
|
||||
public static void main(String[] args) throws Exception {
|
||||
int requiredArgs = 6;
|
||||
Assert.isTrue(args.length >= requiredArgs, () -> "Usage: " + AotProcessor.class.getName()
|
||||
+ " <applicationName> <sourceOutput> <resourceOutput> <classOutput> <groupId> <artifactId> <originalArgs...>");
|
||||
String applicationName = args[0];
|
||||
Path sourceOutput = Paths.get(args[1]);
|
||||
Path resourceOutput = Paths.get(args[2]);
|
||||
Path classOutput = Paths.get(args[3]);
|
||||
String groupId = args[4];
|
||||
String artifactId = args[5];
|
||||
String[] applicationArgs = (args.length > requiredArgs) ? Arrays.copyOfRange(args, requiredArgs, args.length)
|
||||
: new String[0];
|
||||
Class<?> application = Class.forName(applicationName);
|
||||
new AotProcessor(application, applicationArgs, sourceOutput, resourceOutput, classOutput, groupId, artifactId)
|
||||
.process();
|
||||
}
|
||||
|
||||
/**
|
||||
* {@link SpringApplicationHook} used to capture the {@link ApplicationContext} and
|
||||
* trigger early exit of main method.
|
||||
*/
|
||||
private class AotProcessorHook implements SpringApplicationHook {
|
||||
|
||||
@Override
|
||||
public SpringApplicationRunListener getRunListener(SpringApplication application) {
|
||||
return new SpringApplicationRunListener() {
|
||||
|
||||
@Override
|
||||
public void contextLoaded(ConfigurableApplicationContext context) {
|
||||
throw new AbandonedRunException(context);
|
||||
}
|
||||
|
||||
};
|
||||
}
|
||||
|
||||
private <T> GenericApplicationContext run(ThrowingSupplier<T> action) {
|
||||
try {
|
||||
SpringApplication.withHook(this, action);
|
||||
}
|
||||
catch (AbandonedRunException ex) {
|
||||
ApplicationContext context = ex.getApplicationContext();
|
||||
Assert.isInstanceOf(GenericApplicationContext.class, context,
|
||||
() -> "AOT processing requires a GenericApplicationContext but got a "
|
||||
+ context.getClass().getName());
|
||||
return (GenericApplicationContext) context;
|
||||
}
|
||||
throw new IllegalStateException(
|
||||
"No application context available after calling main method of '%s'. Does it run a SpringApplication?"
|
||||
.formatted(AotProcessor.this.application.getName()));
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,132 @@
|
||||
/*
|
||||
* Copyright 2012-2022 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.boot;
|
||||
|
||||
import java.lang.reflect.Method;
|
||||
import java.nio.file.Path;
|
||||
import java.nio.file.Paths;
|
||||
import java.util.Arrays;
|
||||
|
||||
import org.springframework.boot.SpringApplication.AbandonedRunException;
|
||||
import org.springframework.context.ApplicationContext;
|
||||
import org.springframework.context.ConfigurableApplicationContext;
|
||||
import org.springframework.context.aot.AotProcessor;
|
||||
import org.springframework.context.support.GenericApplicationContext;
|
||||
import org.springframework.util.Assert;
|
||||
import org.springframework.util.ReflectionUtils;
|
||||
import org.springframework.util.function.ThrowingSupplier;
|
||||
|
||||
/**
|
||||
* Entry point for AOT processing of a {@link SpringApplication}.
|
||||
* <p>
|
||||
* <strong>For internal use only.</strong>
|
||||
*
|
||||
* @author Stephane Nicoll
|
||||
* @author Andy Wilkinson
|
||||
* @author Phillip Webb
|
||||
* @since 3.0.0
|
||||
*/
|
||||
public class SpringApplicationAotProcessor extends AotProcessor {
|
||||
|
||||
private final String[] applicationArgs;
|
||||
|
||||
/**
|
||||
* Create a new processor for the specified application and settings.
|
||||
* @param application the application main class
|
||||
* @param applicationArgs the arguments to provide to the main method
|
||||
* @param sourceOutput the location of generated sources
|
||||
* @param resourceOutput the location of generated resources
|
||||
* @param classOutput the location of generated classes
|
||||
* @param groupId the group ID of the application, used to locate
|
||||
* native-image.properties
|
||||
* @param artifactId the artifact ID of the application, used to locate
|
||||
* native-image.properties
|
||||
*/
|
||||
public SpringApplicationAotProcessor(Class<?> application, String[] applicationArgs, Path sourceOutput,
|
||||
Path resourceOutput, Path classOutput, String groupId, String artifactId) {
|
||||
super(application, sourceOutput, resourceOutput, classOutput, groupId, artifactId);
|
||||
this.applicationArgs = applicationArgs;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected GenericApplicationContext prepareApplicationContext(Class<?> application) {
|
||||
return new AotProcessorHook(application).run(() -> {
|
||||
Method mainMethod = application.getMethod("main", String[].class);
|
||||
return ReflectionUtils.invokeMethod(mainMethod, null, new Object[] { this.applicationArgs });
|
||||
});
|
||||
}
|
||||
|
||||
public static void main(String[] args) throws Exception {
|
||||
int requiredArgs = 6;
|
||||
Assert.isTrue(args.length >= requiredArgs, () -> "Usage: " + SpringApplicationAotProcessor.class.getName()
|
||||
+ " <applicationName> <sourceOutput> <resourceOutput> <classOutput> <groupId> <artifactId> <originalArgs...>");
|
||||
String applicationName = args[0];
|
||||
Path sourceOutput = Paths.get(args[1]);
|
||||
Path resourceOutput = Paths.get(args[2]);
|
||||
Path classOutput = Paths.get(args[3]);
|
||||
String groupId = args[4];
|
||||
String artifactId = args[5];
|
||||
String[] applicationArgs = (args.length > requiredArgs) ? Arrays.copyOfRange(args, requiredArgs, args.length)
|
||||
: new String[0];
|
||||
Class<?> application = Class.forName(applicationName);
|
||||
new SpringApplicationAotProcessor(application, applicationArgs, sourceOutput, resourceOutput, classOutput,
|
||||
groupId, artifactId).process();
|
||||
}
|
||||
|
||||
/**
|
||||
* {@link SpringApplicationHook} used to capture the {@link ApplicationContext} and
|
||||
* trigger early exit of main method.
|
||||
*/
|
||||
private static final class AotProcessorHook implements SpringApplicationHook {
|
||||
|
||||
private final Class<?> application;
|
||||
|
||||
private AotProcessorHook(Class<?> application) {
|
||||
this.application = application;
|
||||
}
|
||||
|
||||
@Override
|
||||
public SpringApplicationRunListener getRunListener(SpringApplication application) {
|
||||
return new SpringApplicationRunListener() {
|
||||
|
||||
@Override
|
||||
public void contextLoaded(ConfigurableApplicationContext context) {
|
||||
throw new AbandonedRunException(context);
|
||||
}
|
||||
|
||||
};
|
||||
}
|
||||
|
||||
private <T> GenericApplicationContext run(ThrowingSupplier<T> action) {
|
||||
try {
|
||||
SpringApplication.withHook(this, action);
|
||||
}
|
||||
catch (AbandonedRunException ex) {
|
||||
ApplicationContext context = ex.getApplicationContext();
|
||||
Assert.isInstanceOf(GenericApplicationContext.class, context,
|
||||
() -> "AOT processing requires a GenericApplicationContext but got a "
|
||||
+ context.getClass().getName());
|
||||
return (GenericApplicationContext) context;
|
||||
}
|
||||
throw new IllegalStateException(
|
||||
"No application context available after calling main method of '%s'. Does it run a SpringApplication?"
|
||||
.formatted(this.application.getName()));
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
@ -1,154 +0,0 @@
|
||||
/*
|
||||
* Copyright 2012-2022 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.boot;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Path;
|
||||
import java.util.function.Consumer;
|
||||
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.junit.jupiter.api.io.TempDir;
|
||||
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
|
||||
import static org.assertj.core.api.Assertions.assertThatIllegalStateException;
|
||||
|
||||
/**
|
||||
* Tests for {@link AotProcessor}.
|
||||
*
|
||||
* @author Stephane Nicoll
|
||||
* @author Andy Wilkinson
|
||||
*/
|
||||
class AotProcessorTests {
|
||||
|
||||
@BeforeEach
|
||||
void setup() {
|
||||
SampleApplication.argsHolder = null;
|
||||
SampleApplication.postRunInvoked = false;
|
||||
}
|
||||
|
||||
@Test
|
||||
void processApplicationInvokesRunMethod(@TempDir Path directory) throws IOException {
|
||||
String[] arguments = new String[] { "1", "2" };
|
||||
AotProcessor processor = new AotProcessor(SampleApplication.class, arguments, directory.resolve("source"),
|
||||
directory.resolve("resource"), directory.resolve("class"), "com.example", "example");
|
||||
processor.process();
|
||||
assertThat(SampleApplication.argsHolder).isEqualTo(arguments);
|
||||
assertThat(SampleApplication.postRunInvoked).isFalse();
|
||||
assertThat(directory).satisfies(hasGeneratedAssetsForSampleApplication());
|
||||
}
|
||||
|
||||
@Test
|
||||
void processApplicationWithMainMethodThatDoesNotRun(@TempDir Path directory) {
|
||||
AotProcessor processor = new AotProcessor(BrokenApplication.class, new String[0], directory.resolve("source"),
|
||||
directory.resolve("resource"), directory.resolve("class"), "com.example", "example");
|
||||
assertThatIllegalStateException().isThrownBy(processor::process)
|
||||
.withMessageContaining("Does it run a SpringApplication?");
|
||||
assertThat(directory).isEmptyDirectory();
|
||||
}
|
||||
|
||||
@Test
|
||||
void invokeMainParseArgumentsAndInvokesRunMethod(@TempDir Path directory) throws Exception {
|
||||
String[] mainArguments = new String[] { SampleApplication.class.getName(),
|
||||
directory.resolve("source").toString(), directory.resolve("resource").toString(),
|
||||
directory.resolve("class").toString(), "com.example", "example", "1", "2" };
|
||||
AotProcessor.main(mainArguments);
|
||||
assertThat(SampleApplication.argsHolder).containsExactly("1", "2");
|
||||
assertThat(SampleApplication.postRunInvoked).isFalse();
|
||||
assertThat(directory).satisfies(hasGeneratedAssetsForSampleApplication());
|
||||
}
|
||||
|
||||
@Test
|
||||
void invokeMainWithMissingArguments() {
|
||||
assertThatIllegalArgumentException().isThrownBy(() -> AotProcessor.main(new String[] { "Test" }))
|
||||
.withMessageContaining("Usage:");
|
||||
}
|
||||
|
||||
@Test
|
||||
void processingDeletesExistingOutput(@TempDir Path directory) throws IOException {
|
||||
Path sourceOutput = directory.resolve("source");
|
||||
Path resourceOutput = directory.resolve("resource");
|
||||
Path classOutput = directory.resolve("class");
|
||||
Path existingSourceOutput = createExisting(sourceOutput);
|
||||
Path existingResourceOutput = createExisting(resourceOutput);
|
||||
Path existingClassOutput = createExisting(classOutput);
|
||||
AotProcessor processor = new AotProcessor(SampleApplication.class, new String[0], sourceOutput, resourceOutput,
|
||||
classOutput, "com.example", "example");
|
||||
processor.process();
|
||||
assertThat(existingSourceOutput).doesNotExist();
|
||||
assertThat(existingResourceOutput).doesNotExist();
|
||||
assertThat(existingClassOutput).doesNotExist();
|
||||
}
|
||||
|
||||
private Path createExisting(Path directory) throws IOException {
|
||||
Path existing = directory.resolve("existing");
|
||||
Files.createDirectories(directory);
|
||||
Files.createFile(existing);
|
||||
return existing;
|
||||
}
|
||||
|
||||
private Consumer<Path> hasGeneratedAssetsForSampleApplication() {
|
||||
return (directory) -> {
|
||||
assertThat(directory.resolve(
|
||||
"source/org/springframework/boot/AotProcessorTests_SampleApplication__ApplicationContextInitializer.java"))
|
||||
.exists().isRegularFile();
|
||||
assertThat(directory.resolve("source/org/springframework/boot/AotProcessorTests__BeanDefinitions.java"))
|
||||
.exists().isRegularFile();
|
||||
assertThat(directory.resolve(
|
||||
"source/org/springframework/boot/AotProcessorTests_SampleApplication__BeanFactoryRegistrations.java"))
|
||||
.exists().isRegularFile();
|
||||
assertThat(directory.resolve("resource/META-INF/native-image/com.example/example/reflect-config.json"))
|
||||
.exists().isRegularFile();
|
||||
Path nativeImagePropertiesFile = directory
|
||||
.resolve("resource/META-INF/native-image/com.example/example/native-image.properties");
|
||||
assertThat(nativeImagePropertiesFile).exists().isRegularFile().hasContent("""
|
||||
Args = -H:Class=org.springframework.boot.AotProcessorTests$SampleApplication \\
|
||||
--report-unsupported-elements-at-runtime \\
|
||||
--no-fallback \\
|
||||
--install-exit-handlers
|
||||
""");
|
||||
};
|
||||
}
|
||||
|
||||
@Configuration(proxyBeanMethods = false)
|
||||
public static class SampleApplication {
|
||||
|
||||
public static String[] argsHolder;
|
||||
|
||||
public static boolean postRunInvoked;
|
||||
|
||||
public static void main(String[] args) {
|
||||
argsHolder = args;
|
||||
SpringApplication.run(SampleApplication.class, args);
|
||||
postRunInvoked = true;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
public static class BrokenApplication {
|
||||
|
||||
public static void main(String[] args) {
|
||||
// Does not run an application
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,106 @@
|
||||
/*
|
||||
* Copyright 2012-2022 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* https://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package org.springframework.boot;
|
||||
|
||||
import java.nio.file.Path;
|
||||
|
||||
import org.junit.jupiter.api.BeforeEach;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.junit.jupiter.api.io.TempDir;
|
||||
|
||||
import org.springframework.context.annotation.Configuration;
|
||||
|
||||
import static org.assertj.core.api.Assertions.assertThat;
|
||||
import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
|
||||
import static org.assertj.core.api.Assertions.assertThatIllegalStateException;
|
||||
|
||||
/**
|
||||
* Tests for {@link SpringApplicationAotProcessor}.
|
||||
*
|
||||
* @author Stephane Nicoll
|
||||
* @author Andy Wilkinson
|
||||
*/
|
||||
class SpringApplicationAotProcessorTests {
|
||||
|
||||
@BeforeEach
|
||||
void setup() {
|
||||
SampleApplication.argsHolder = null;
|
||||
SampleApplication.postRunInvoked = false;
|
||||
}
|
||||
|
||||
@Test
|
||||
void processApplicationInvokesRunMethod(@TempDir Path directory) {
|
||||
String[] arguments = new String[] { "1", "2" };
|
||||
SpringApplicationAotProcessor processor = new SpringApplicationAotProcessor(SampleApplication.class, arguments,
|
||||
directory.resolve("source"), directory.resolve("resource"), directory.resolve("class"), "com.example",
|
||||
"example");
|
||||
processor.process();
|
||||
assertThat(SampleApplication.argsHolder).isEqualTo(arguments);
|
||||
assertThat(SampleApplication.postRunInvoked).isFalse();
|
||||
}
|
||||
|
||||
@Test
|
||||
void processApplicationWithMainMethodThatDoesNotRun(@TempDir Path directory) {
|
||||
SpringApplicationAotProcessor processor = new SpringApplicationAotProcessor(BrokenApplication.class,
|
||||
new String[0], directory.resolve("source"), directory.resolve("resource"), directory.resolve("class"),
|
||||
"com.example", "example");
|
||||
assertThatIllegalStateException().isThrownBy(processor::process)
|
||||
.withMessageContaining("Does it run a SpringApplication?");
|
||||
assertThat(directory).isEmptyDirectory();
|
||||
}
|
||||
|
||||
@Test
|
||||
void invokeMainParseArgumentsAndInvokesRunMethod(@TempDir Path directory) throws Exception {
|
||||
String[] mainArguments = new String[] { SampleApplication.class.getName(),
|
||||
directory.resolve("source").toString(), directory.resolve("resource").toString(),
|
||||
directory.resolve("class").toString(), "com.example", "example", "1", "2" };
|
||||
SpringApplicationAotProcessor.main(mainArguments);
|
||||
assertThat(SampleApplication.argsHolder).containsExactly("1", "2");
|
||||
assertThat(SampleApplication.postRunInvoked).isFalse();
|
||||
}
|
||||
|
||||
@Test
|
||||
void invokeMainWithMissingArguments() {
|
||||
assertThatIllegalArgumentException()
|
||||
.isThrownBy(() -> SpringApplicationAotProcessor.main(new String[] { "Test" }))
|
||||
.withMessageContaining("Usage:");
|
||||
}
|
||||
|
||||
@Configuration(proxyBeanMethods = false)
|
||||
public static class SampleApplication {
|
||||
|
||||
public static String[] argsHolder;
|
||||
|
||||
public static boolean postRunInvoked;
|
||||
|
||||
public static void main(String[] args) {
|
||||
argsHolder = args;
|
||||
SpringApplication.run(SampleApplication.class, args);
|
||||
postRunInvoked = true;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
public static class BrokenApplication {
|
||||
|
||||
public static void main(String[] args) {
|
||||
// Does not run an application
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Reference in New Issue