Merge pull request #2 from rchodava/sf-issue-43

Allow truncation
rchodava · Feb 25, 2016 · 5b15ac0 · 5b15ac0
2 parents de29c63 + 4a88c3d
commit 5b15ac0
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/src/main/java/com/chodavarapu/io/ReadWriteBuffer.java b/src/main/java/com/chodavarapu/io/ReadWriteBuffer.java
@@ -65,6 +65,12 @@ public OutputStream openOutputStream() {
         return new BufferOutputStream();
     }
 
+    public void truncate() {
+        synchronized (bufferMonitor) {
+            writtenPosition = 0;
+        }
+    }
+
     private class BufferInputStream extends InputStream {
         private int readPosition = 0;
         private long waitStart;

diff --git a/src/test/java/com/chodavarapu/io/ReadWriteBufferTest.java b/src/test/java/com/chodavarapu/io/ReadWriteBufferTest.java
@@ -27,6 +27,24 @@ public void basicReadWrite() throws Exception {
         assertEquals("Another String", reader.readLine());
     }
 
+    @Test
+    public void truncation() throws Exception {
+        ReadWriteBuffer buffer = new ReadWriteBuffer();
+
+        OutputStreamWriter writer = new OutputStreamWriter(buffer.openOutputStream());
+        writer.write("Test String\n");
+        writer.flush();
+
+        buffer.truncate();
+
+        BufferedReader reader = new BufferedReader(new InputStreamReader(buffer.openInputStream()));
+
+        writer.write("Another String\n");
+        writer.flush();
+
+        assertEquals("Another String", reader.readLine());
+    }
+
     @Test
     public void outputCompletionNotification() throws Exception {
         ArrayList<byte[]> completionNotificationBytes = new ArrayList<>();