Owen-CH-Leung commented on code in PR #13773:
URL: https://github.com/apache/kafka/pull/13773#discussion_r1265177897
##########
metadata/src/test/java/org/apache/kafka/image/writer/ImageWriterOptionsTest.java:
##########
@@ -44,9 +48,33 @@ public void testSetMetadataVersion() {
setMetadataVersion(version);
if (i < MetadataVersion.MINIMUM_BOOTSTRAP_VERSION.ordinal()) {
assertEquals(MetadataVersion.MINIMUM_KRAFT_VERSION,
options.metadataVersion());
+ assertEquals(version, options.requestedMetadataVersion());
} else {
assertEquals(version, options.metadataVersion());
}
}
}
+
+ @Test
+ public void testHandleLoss() {
+ PrintStream originalOut = System.out;
+ String expectedMessage = "stuff";
+ Consumer<UnwritableMetadataException> customLossHandler = e ->
System.out.println(e.getMessage());
+
+ for (int i = MetadataVersion.MINIMUM_KRAFT_VERSION.ordinal();
+ i < MetadataVersion.VERSIONS.length;
+ i++) {
+ ByteArrayOutputStream outContent = new ByteArrayOutputStream();
+ MetadataVersion version = MetadataVersion.VERSIONS[i];
+ ImageWriterOptions options = new ImageWriterOptions.Builder()
+ .setMetadataVersion(version)
+ .setLossHandler(customLossHandler)
+ .build();
+ System.setOut(new PrintStream(outContent));
+ options.handleLoss(expectedMessage);
+ System.setOut(originalOut);
+ String formattedMessage = String.format("Metadata has been lost
because the following could not be represented in metadata version %s: %s",
version, expectedMessage);
+ assertEquals(formattedMessage, outContent.toString().trim());
+ }
Review Comment:
@showuon Thanks! Yes it's a lot cleaner. I've adopted your suggestion.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: jira-unsubscr...@kafka.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
