From 99e54378c3b5d5a9414226fd0fdf091eee8954ed Mon Sep 17 00:00:00 2001
From: DiegoMarti2 <150871541+DiegoMarti2@users.noreply.github.com>
Date: Thu, 14 Dec 2023 00:34:19 +0200
Subject: [PATCH] Update OutputStreamToByteArrayUnitTest.java

---
 .../OutputStreamToByteArrayUnitTest.java                 | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/core-java-modules/core-java-io-5/src/test/java/com/baeldung/outputstreamtobytearray/OutputStreamToByteArrayUnitTest.java b/core-java-modules/core-java-io-5/src/test/java/com/baeldung/outputstreamtobytearray/OutputStreamToByteArrayUnitTest.java
index 0be88b1c07..ed0cab250a 100644
--- a/core-java-modules/core-java-io-5/src/test/java/com/baeldung/outputstreamtobytearray/OutputStreamToByteArrayUnitTest.java
+++ b/core-java-modules/core-java-io-5/src/test/java/com/baeldung/outputstreamtobytearray/OutputStreamToByteArrayUnitTest.java
@@ -29,12 +29,10 @@ public class OutputStreamToByteArrayUnitTest {
 
 
     @Test
-    public void givenByteArrayOutputStream_whenUsingDrainableOutputStream_thenReturnByteArray() throws IOException {
-        String data = "Welcome to Baeldung!";
-
-        ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
-        DrainableOutputStream drainableOutputStream = new DrainableOutputStream(byteArrayOutputStream);
+    public void givenSystemOut_whenUsingDrainableOutputStream_thenReturnByteArray() throws IOException {
+        String data = "Welcome to Baeldung!\n";
 
+        DrainableOutputStream drainableOutputStream = new DrainableOutputStream(System.out);
         try (drainableOutputStream) {
             drainableOutputStream.write(data.getBytes(StandardCharsets.UTF_8));
         }
@@ -43,7 +41,6 @@ public class OutputStreamToByteArrayUnitTest {
         assertEquals(data, new String(writtenData, StandardCharsets.UTF_8));
     }
 
-
     public class DrainableOutputStream extends FilterOutputStream {
         private final ByteArrayOutputStream buffer;