diff --git a/core/internal/test/src/mill/internal/LinePrefixOutputStreamTests.scala b/core/internal/test/src/mill/internal/LinePrefixOutputStreamTests.scala index 17eb83d5010..4b4b653c33a 100644 --- a/core/internal/test/src/mill/internal/LinePrefixOutputStreamTests.scala +++ b/core/internal/test/src/mill/internal/LinePrefixOutputStreamTests.scala @@ -75,32 +75,34 @@ object LinePrefixOutputStreamTests extends TestSuite { } test("colors") { - val baos = new ByteArrayOutputStream() - val lpos = new LinePrefixOutputStream("PREFIX", baos) - lpos.write(fansi.Color.Red("hello").render.getBytes) - lpos.write('\n') - lpos.flush() - lpos.write(fansi.Color.Green("world").render.getBytes) - lpos.write('\n') - lpos.flush() + for (chunkSize <- Seq(1, 2, 3, 4, 5, 6, 7)) { + val baos = new ByteArrayOutputStream() + val lpos = new LinePrefixOutputStream("PREFIX", baos) + lpos.write(fansi.Color.Red("hello").render.getBytes) + lpos.write('\n') + lpos.flush() + lpos.write(fansi.Color.Green("world").render.getBytes) + lpos.write('\n') + lpos.flush() - val blueText = fansi.Color.Blue("one\ntwo\nthree\nfour\nfive").render.getBytes + val blueText = fansi.Color.Blue("one\ntwo\nthree\nfour\nfive").render.getBytes - blueText.grouped(7).foreach { chunk => - lpos.write(chunk) - lpos.flush() - } + blueText.grouped(chunkSize).foreach { chunk => + lpos.write(chunk) + lpos.flush() + } - val expected = - "PREFIX" + fansi.Color.Red("hello").render + "\n" + - "PREFIX" + fansi.Color.Green("world").render + "\n" + - "PREFIX" + fansi.Color.Blue("one\n") + - "PREFIX" + fansi.Color.Blue("two\n") + - "PREFIX" + fansi.Color.Blue("three\n") + - "PREFIX" + fansi.Color.Blue("four\n") + - "PREFIX" + fansi.Color.Blue("five") + val expected = + "PREFIX" + fansi.Color.Red("hello").render + "\n" + + "PREFIX" + fansi.Color.Green("world").render + "\n" + + "PREFIX" + fansi.Color.Blue("one\n") + + "PREFIX" + fansi.Color.Blue("two\n") + + "PREFIX" + fansi.Color.Blue("three\n") + + "PREFIX" + fansi.Color.Blue("four\n") + + "PREFIX" + fansi.Color.Blue("five") - assert(baos.toString == expected) + assert(baos.toString == expected) + } } }