Resolve object mode as an array of objects

Closes #7
bendrucker · Jun 18, 2016 · ff4c10f · ff4c10f
1 parent dae1d7d
commit ff4c10f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/index.js b/index.js
@@ -19,6 +19,9 @@ function fromReadable (stream) {
 
   return promise
     .then(function concat (parts) {
+      if (stream._readableState && stream._readableState.objectMode) {
+        return parts
+      }
       return Buffer.concat(parts.map(bufferize))
     })
 }

diff --git a/test.js b/test.js
@@ -69,6 +69,18 @@ describe('stream-to-promise', function () {
       })
     })
 
+    it('returns an array for object streams', function () {
+      var objectStream = new Stream.Readable({objectMode: true})
+      objectStream._read = function noop () {}
+      var promise = streamToPromise(objectStream)
+      objectStream.emit('data', {foo: 'bar'})
+      objectStream.emit('data', {baz: 'qux'})
+      objectStream.emit('end')
+      return promise.then(function (results) {
+        expect(results).to.deep.equal([{foo: 'bar'}, {baz: 'qux'}])
+      })
+    })
+
     it('rejects on stream errors', function () {
       var err = new Error()
       readable.emit('error', err)