iSharkFly-Docs
/
java-tutorials
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update OutputStreamtoByteArrayUnitTest.java

This commit is contained in:
DiegoMarti2 2023-12-10 20:53:59 +02:00 committed by GitHub
parent 94ddcd153d
commit b56eaf9b1b
1 changed files with 30 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
core-java-modules/core-java-io-5/src/test/java/com/baeldung/outputstreamtobytearray/OutputStreamtoByteArrayUnitTest.java
View File

@ -1,26 +1,30 @@
package com.baeldung.outputstreamtobytearray; package com.baeldung.outputstreamtobytearray;
import org.apache.commons.io.FileUtils;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.io.TempDir;
import java.io.*; import java.io.*;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import static org.junit.Assert.assertEquals;
public class OutputStreamtoByteArrayUnitTest { public class OutputStreamtoByteArrayUnitTest {
@Test @Test
public void givenFileOutputStream_whenUsingFileUtilsToReadTheFile_thenReturnByteArray() throws IOException { public void givenFileOutputStream_whenUsingFileUtilsToReadTheFile_thenReturnByteArray(@TempDir Path tempDir) throws IOException {
String data = "Welcome to Baeldung!"; String data = "Welcome to Baeldung!";
String filePath = "file.txt"; String fileName = "file.txt";
Path filePath = tempDir.resolve(fileName);
try (FileOutputStream outputStream = new FileOutputStream(filePath)) { try (DrainableOutputStream drainableOutputStream = new DrainableOutputStream(new FileOutputStream(filePath.toFile()))) {
outputStream.write(data.getBytes()); drainableOutputStream.write(data.getBytes());
} }
byte[] byteArray = FileUtils.readFileToByteArray(new File(filePath)); byte[] intercepted = Files.readAllBytes(filePath);
String result = new String(byteArray); String result = new String(intercepted);
assertEquals(data, result);
Assertions.assertEquals(data, result);
} }
@ -28,11 +32,21 @@ public class OutputStreamtoByteArrayUnitTest {
public void givenFileOutputStream_whenUsingDrainableOutputStream_thenReturnByteArray() throws IOException { public void givenFileOutputStream_whenUsingDrainableOutputStream_thenReturnByteArray() throws IOException {
String data = "Welcome to Baeldung!"; String data = "Welcome to Baeldung!";
String filePath = "file.txt"; String filePath = "file.txt";
DrainableOutputStream drainableOutputStream = new DrainableOutputStream(new FileOutputStream(filePath));
DrainableOutputStream drainableOutputStream = null;
try {
drainableOutputStream = new DrainableOutputStream(new FileOutputStream(filePath));
drainableOutputStream.write(data.getBytes()); drainableOutputStream.write(data.getBytes());
byte[] intercepted = drainableOutputStream.toByteArray(); } finally {
if (drainableOutputStream != null) {
drainableOutputStream.close();
}
}
byte[] intercepted = Files.readAllBytes(Paths.get(filePath));
String result = new String(intercepted); String result = new String(intercepted);
Assertions.assertEquals(data, result); assertEquals(data, result);
} }
public class DrainableOutputStream extends FilterOutputStream { public class DrainableOutputStream extends FilterOutputStream {
@ -45,12 +59,12 @@ public class OutputStreamtoByteArrayUnitTest {
@Override @Override
public void write(byte b[]) throws IOException { public void write(byte b[]) throws IOException {
this.buffer.write(b); buffer.write(b);
super.write(b); super.write(b);
} }
public byte[] toByteArray() { public byte[] toByteArray() {
return this.buffer.toByteArray(); return buffer.toByteArray();
} }
} }
} }