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

---
 .../OutputStreamToByteArrayUnitTest.java         | 16 ++++++++++------
 1 file changed, 10 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 fa55623973..0082e77cfe 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
@@ -22,7 +22,7 @@ public class OutputStreamToByteArrayUnitTest {
             outputStream.write(data.getBytes(StandardCharsets.UTF_8));
         }
 
-        byte[] writtenData = FileUtils.readFileToByteArray(new File(filePath.toUri()));
+        byte[] writtenData = FileUtils.readFileToByteArray(filePath.toFile());
         String result = new String(writtenData, StandardCharsets.UTF_8);
         assertEquals(data, result);
     }
@@ -34,15 +34,19 @@ public class OutputStreamToByteArrayUnitTest {
         String fileName = "file.txt";
         Path filePath = tempDir.resolve(fileName);
 
-        try (DrainableOutputStream drainableOutputStream = new DrainableOutputStream(new FileOutputStream(filePath.toFile()))) {
+        DrainableOutputStream drainableOutputStream = new DrainableOutputStream(new FileOutputStream(filePath.toFile()));
+
+        try {
             drainableOutputStream.write(data.getBytes(StandardCharsets.UTF_8));
-            byte[] writtenData = drainableOutputStream.toByteArray();
-            assertEquals(data, new String(writtenData, StandardCharsets.UTF_8));
-        } catch (IOException e) {
-            e.printStackTrace();
+        } finally {
+            drainableOutputStream.close();
         }
+
+        byte[] writtenData = drainableOutputStream.toByteArray();
+        assertEquals(data, new String(writtenData, StandardCharsets.UTF_8));
     }
 
+
     public class DrainableOutputStream extends FilterOutputStream {
         private final ByteArrayOutputStream buffer;