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

Polish

Browse Source
This commit is contained in:
Stéphane Nicoll
2024-07-03 10:04:51 +02:00
parent 99e8978a91
commit 1ea4eb147a
2 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
spring-core/src/test/java/org/springframework/aot/generate/GeneratedFilesTests.java
View File

@@ -190,6 +190,7 @@ class GeneratedFilesTests {
AtomicBoolean called = new AtomicBoolean(false);
this.generatedFiles.handleFile(Kind.RESOURCE, "META-INF/test", handler -> {
called.set(true);
assertThat(handler.getContent()).isNotNull();
String existing = readSource(handler.getContent());
handler.override(createSource(existing+"-override"));
});
@@ -225,8 +226,10 @@ class GeneratedFilesTests {
static class TestGeneratedFiles implements GeneratedFiles {
@Nullable
private Kind kind;
@Nullable
private String path;
private TestFileHandler fileHandler = new TestFileHandler();
@@ -247,21 +250,22 @@ class GeneratedFilesTests {
assertThat(this.kind).as("kind").isEqualTo(kind);
assertThat(this.path).as("path").isEqualTo(path);
assertThat(this.fileHandler.content).as("content").isNotNull();
return new GeneratedFileAssert(this.fileHandler);
return new GeneratedFileAssert(this.fileHandler.content, this.fileHandler.override);
}
}
private static class GeneratedFileAssert extends AbstractStringAssert<GeneratedFileAssert> {
private final TestFileHandler fileHandler;
@Nullable
private final Boolean override;
GeneratedFileAssert(TestFileHandler fileHandler) throws IOException {
super(readSource(fileHandler.content), GeneratedFileAssert.class);
this.fileHandler = fileHandler;
GeneratedFileAssert(InputStreamSource content, @Nullable Boolean override) throws IOException {
super(readSource(content), GeneratedFileAssert.class);
this.override = override;
}
public GeneratedFileAssert hasOverride(boolean expected) {
assertThat(this.fileHandler.override).isEqualTo(expected);
assertThat(this.override).isEqualTo(expected);
return this.myself;
}
}