update JUnit test

thangpt2019 · thangpt2019 · commit e7ad27466dd7 · 2017-10-01T21:03:56.000+07:00
diff --git a/README.md b/README.md
@@ -13,4 +13,5 @@ Find more detail at:
 * [Set Up Eclipse, IntelliJ And NetBeans For Java 9](https://howtoprogram.xyz/2017/09/24/set-up-eclipse-intellij-and-netbeans-for-java-9/)
 * [Create Immutable Lists In Java 9 By Static Factory Methods](https://howtoprogram.xyz/2017/09/24/java-9-create-immutable-lists-static-factory-method/)
 * [Java 9 Example With Maven And JUnit 5](https://howtoprogram.xyz/2017/09/23/java-9-maven-junit-5-example/)
+* [Copy Streams In Java 9 With The InputStream.transferTo() Method](https://howtoprogram.xyz/2017/10/01/java-9-inputstream-transferto-copy-streams/)
 
diff --git a/java9-language-features-examples/README.md b/java9-language-features-examples/README.md
@@ -3,6 +3,7 @@
 Demonstration for:
 
 [Create Immutable Lists In Java 9 By Static Factory Methods](https://howtoprogram.xyz/2017/09/24/java-9-create-immutable-lists-static-factory-method/)
+
 [Copy Streams In Java 9 With The InputStream.transferTo() Method](https://howtoprogram.xyz/2017/10/01/java-9-inputstream-transferto-copy-streams/)
 
 
diff --git a/java9-language-features-examples/src/test/java/com/howtoprogram/io/InputStreamTransferToTest.java b/java9-language-features-examples/src/test/java/com/howtoprogram/io/InputStreamTransferToTest.java
@@ -3,25 +3,26 @@
 import org.junit.jupiter.api.Test;
 
 import java.io.*;
+import java.util.Arrays;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertTrue;
 
 public class InputStreamTransferToTest {
 
     @Test
     public void testTransferTo() throws IOException {
 
-        ByteArrayInputStream bis = new ByteArrayInputStream("Hello Java 9".getBytes());
+        byte [] inBytes = "Hello Java 9".getBytes();
+        ByteArrayInputStream bis = new ByteArrayInputStream(inBytes);
         ByteArrayOutputStream bos = new ByteArrayOutputStream();
         try {
 
+             bis.transferTo(bos);
 
-            long count = bis.transferTo(bos);
-
-            byte[] bytes = bos.toByteArray();
-
-            assertEquals(count, bytes.length);
+            byte[] outBytes = bos.toByteArray();
 
+            assertTrue(Arrays.equals(inBytes,outBytes));
 
         } finally {
             try {
@@ -43,16 +44,16 @@ public void testTransferTo() throws IOException {
 
     @Test
     public void testTransferToJava9Syntax() throws IOException {
-
-        ByteArrayInputStream bis = new ByteArrayInputStream("Hello Java 9".getBytes());
+        byte [] inBytes = "Hello Java 9".getBytes();
+        ByteArrayInputStream bis = new ByteArrayInputStream(inBytes);
         ByteArrayOutputStream bos = new ByteArrayOutputStream();
 
         try (bis; bos) {
 
             long count = bis.transferTo(bos);
 
-            byte[] bytes = bos.toByteArray();
-            assertEquals(count, bytes.length);
+            byte[] outBytes = bos.toByteArray();
+            assertTrue(Arrays.equals(inBytes,outBytes));
         }
     }
 }
