diff --git a/core/src/test/scala/unit/kafka/tools/ConsumerPerformanceTest.scala b/core/src/test/scala/unit/kafka/tools/ConsumerPerformanceTest.scala index 49521d6bd9a..d872a40bc1d 100644 --- a/core/src/test/scala/unit/kafka/tools/ConsumerPerformanceTest.scala +++ b/core/src/test/scala/unit/kafka/tools/ConsumerPerformanceTest.scala @@ -25,8 +25,6 @@ import org.junit.jupiter.api.Assertions.{assertEquals, assertThrows} import org.junit.jupiter.api.Test class ConsumerPerformanceTest { - - private val outContent = new ByteArrayOutputStream() private val dateFormat = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss:SSS") @Test @@ -148,16 +146,21 @@ class ConsumerPerformanceTest { } private def testHeaderMatchContent(detailed: Boolean, expectedOutputLineCount: Int, fun: () => Unit): Unit = { - Console.withOut(outContent) { - ConsumerPerformance.printHeader(detailed) - fun() + val outContent = new ByteArrayOutputStream + try { + Console.withOut(outContent) { + ConsumerPerformance.printHeader(detailed) + fun() - val contents = outContent.toString.split("\n") - assertEquals(expectedOutputLineCount, contents.length) - val header = contents(0) - val body = contents(1) + val contents = outContent.toString.split("\n") + assertEquals(expectedOutputLineCount, contents.length) + val header = contents(0) + val body = contents(1) - assertEquals(header.split(",").length, body.split(",").length) + assertEquals(header.split(",\\s").length, body.split(",\\s").length) + } + } finally { + outContent.close() } } }