Implement CountingServletOutputStream#close()

gini · Oct 22, 2018 · 5456078 · 5456078
1 parent 0217d57
commit 5456078
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/main/java/net/gini/dropwizard/gelf/filters/CountingServletOutputStream.java b/src/main/java/net/gini/dropwizard/gelf/filters/CountingServletOutputStream.java
@@ -50,6 +50,11 @@ public void write(final byte[] b, final int off, final int len) throws IOExcepti
         outputStream.write(b, off, len);
     }
 
+    @Override
+    public void close() throws IOException {
+        underlyingStream.close();
+    }
+
     @Override
     public void flush() throws IOException {
         underlyingStream.flush();

diff --git a/src/test/java/net/gini/dropwizard/gelf/filters/CountingServletOutputStreamTest.java b/src/test/java/net/gini/dropwizard/gelf/filters/CountingServletOutputStreamTest.java
@@ -31,6 +31,13 @@ public void setUp() {
         outputStream = new CountingServletOutputStream(delegate);
     }
 
+    @Test
+    public void testThatCloseDelegates() throws IOException {
+        outputStream.close();
+
+        verify(delegate).close();
+    }
+
     @Test
     public void testThatFlushDelegates() throws IOException {
         outputStream.flush();