BAEL-1609 Updated test to throw exception

iaforek · iaforek · commit 2fe8d8810d90 · 2018-03-18T13:05:54.000Z
diff --git a/guava/src/main/java/org/baeldung/guava/CountingByteWritter.java b/guava/src/main/java/org/baeldung/guava/CountingByteWritter.java
@@ -0,0 +1,5 @@
+package org.baeldung.guava;
+
+public class CountingByteWritter {
+
+}
diff --git a/guava/src/test/java/org/baeldung/guava/GuavaCountingOutputStreamTest.java b/guava/src/test/java/org/baeldung/guava/GuavaCountingOutputStreamTest.java
@@ -1,25 +1,29 @@
 package org.baeldung.guava;
 
-import static org.junit.Assert.assertEquals;
-
+import java.io.ByteArrayInputStream;
 import java.io.ByteArrayOutputStream;
 
 import org.junit.Test;
 
 import com.google.common.io.CountingOutputStream;
 
 public class GuavaCountingOutputStreamTest {
+    public static final int MAX = 5;
 
-    @Test
-    public void givenData_whenWrittenToStream_thenGetCorrectCount() throws Exception {
+    @Test(expected = RuntimeException.class)
+    public void givenData_whenCountReachesLimit_thenThrowException() throws Exception {
         ByteArrayOutputStream out = new ByteArrayOutputStream();
         CountingOutputStream cos = new CountingOutputStream(out);
-        
-        byte[] data = new byte[10];
-        
-        cos.write(data);
-        
-        assertEquals(10, cos.getCount());
-        
+
+        byte[] data = new byte[1024];
+        ByteArrayInputStream in = new ByteArrayInputStream(data);
+
+        while (in.read() != -1) {
+            cos.write(data);
+            cos.flush();
+            if (cos.getCount() >= MAX) {
+                throw new RuntimeException("Write limit reached");
+            }
+        }
     }
 }

-Original file line number
+Diff line change
@@ @@ -0,0 +1,5 @@ @@
 +package org.baeldung.guava;
++
 +public class CountingByteWritter {
++
 +}