Spring/spring-framework
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Emulate AbstractAotProcessor.process() in AOT endToEndTests()

Browse Source
This commit is contained in:
Sam Brannen
2024-10-04 15:46:23 +02:00
parent 21d80dd849
commit c4c005b82e
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
spring-test/src/test/java/org/springframework/test/context/aot/AotIntegrationTests.java
View File

@@ -97,7 +97,16 @@ class AotIntegrationTests extends AbstractAotTests {
// AOT BUILD-TIME: PROCESSING
InMemoryGeneratedFiles generatedFiles = new InMemoryGeneratedFiles();
TestContextAotGenerator generator = new TestContextAotGenerator(generatedFiles, new RuntimeHints());
generator.processAheadOfTime(testClasses);
try {
// Emulate AbstractAotProcessor.process().
System.setProperty(AbstractAotProcessor.AOT_PROCESSING, "true");
generator.processAheadOfTime(testClasses);
}
finally {
// Emulate AbstractAotProcessor.process().
System.clearProperty(AbstractAotProcessor.AOT_PROCESSING);
}
List<String> sourceFiles = generatedFiles.getGeneratedFiles(Kind.SOURCE).keySet().stream().toList();
assertThat(sourceFiles).containsExactlyInAnyOrder(expectedSourceFilesForBasicSpringTests);
@@ -176,13 +185,13 @@ class AotIntegrationTests extends AbstractAotTests {
private void runEndToEndTests(List<Class<?>> testClasses, boolean failOnError) {
InMemoryGeneratedFiles generatedFiles = new InMemoryGeneratedFiles();
TestContextAotGenerator generator = new TestContextAotGenerator(generatedFiles, new RuntimeHints(), failOnError);
// AOT BUILD-TIME: PROCESSING
try {
// Emulate AbstractAotProcessor.process().
System.setProperty(AbstractAotProcessor.AOT_PROCESSING, "true");
TestContextAotGenerator generator = new TestContextAotGenerator(generatedFiles, new RuntimeHints(), failOnError);
generator.processAheadOfTime(testClasses.stream());
}
finally {