fix streams

Author: deployable

lib/index.js

@@ -8,15 +8,16 @@ module.exports = class RingBuffer{
   constructor ( size ) {
     this.size = size
     this.sizei = size - 1
-    this.buffer = new Array(size) 
+    this.buffer = new Array(size)
     this.pointer = 0
   }
 
   add (item) {
     this.buffer[this.pointer] = item
     this.pointer = ( this.size + this.pointer + 1 ) % this.size
+    return true
   }
-  
+
   get (i) {
     return this.buffer[i]
   }
@@ -25,23 +26,32 @@ module.exports = class RingBuffer{
     let index = ( this.size + this.pointer - 1 ) % this.size
     return this.buffer[index]
   }
-  
+
   toArray(){
     let res = this.buffer.slice(this.pointer, this.size)
     if ( this.pointer > 0 )
       res = res.concat(this.buffer.slice(0, this.pointer))
     return res
-  } 
-  
+  }
+
   clear(){
     this.buffer = new Array(this.size)
     this.pointer = 0
     return true
   }
- 
+
   writeStream(){
     if ( this.stream ) return this.stream
-    return this.stream = new require('stream').Writable({ write: (chunk)=> this.add(chunk) })
+    return this.stream = new require('stream').Writable({
+      write: (chunk, enc, next)=> {
+        try {
+          this.add(chunk)
+        } catch (err) {
+          next(err)
+        }
+        next()
+      }
+    })
   }
 
 }

test/unit_package_spec.js

@@ -1,9 +1,11 @@
+const expect = require('chai').expect
 const RingBuffer = require('../')
 
+
 describe('Unit::deployable-ringbuffer', function(){
 
   describe('Something does something', function(){
-  
+
     it('should create a RingBuffer', function(){
       expect( new RingBuffer(5) ).to.be.ok
     })
@@ -31,6 +33,14 @@ describe('Unit::deployable-ringbuffer', function(){
       expect( rb.last() ).to.equal('two')
     })
 
+    it('should get the last added when added', function(){
+      let rb = new RingBuffer(100)
+      rb.add('first')
+      expect( rb.last() ).to.equal('first')
+      rb.add('second')
+      expect( rb.last() ).to.equal('second')
+    })
+
     it('should get undefined when there is no last', function(){
       let rb = new RingBuffer(3)
       expect( rb.last() ).to.equal( undefined )
@@ -77,6 +87,15 @@ describe('Unit::deployable-ringbuffer', function(){
       expect( rb.last().toString() ).to.equal( 'test' )
     })
 
+    it('should return a write stream', function(){
+      let rb = new RingBuffer(2)
+      let ws = rb.writeStream()
+      ws.write('test')
+      ws.write('testa')
+      ws.write('testb')
+      expect( rb.last().toString() ).to.equal( 'testb' )
+    })
+
   })
 
 })