Fix: Ensure stream is flowing using stream-exhaust module (fixes #11)

index.js

@@ -5,6 +5,7 @@ var domain = require('domain');
 var eos = require('end-of-stream');
 var tick = require('next-tick');
 var once = require('once');
+var exhaust = require('stream-exhaust');
 
 function noop(){}
 
@@ -35,7 +36,7 @@ function asyncDone(fn, cb){
     if(result && typeof result.on === 'function'){
       // assume node stream
       d.add(result);
-      eos(result, { error: false }, onSuccess);
+      eos(exhaust(result), { error: false }, onSuccess);
       return;
     }
 

package.json

@@ -22,7 +22,8 @@
   "dependencies": {
     "end-of-stream": "^0.1.4",
     "next-tick": "^0.2.2",
-    "once": "^1.3.0"
+    "once": "^1.3.0",
+    "stream-exhaust": "^1.0.0"
   },
   "devDependencies": {
     "tap": "~0.4.8",

test/streams.js

@@ -29,6 +29,10 @@ function failure(){
   return read.pipe(new EndStream());
 }
 
+function unpiped(){
+  return fs.createReadStream(exists);
+}
+
 test('handle a successful stream', function(t){
   asyncDone(success, function(err){
     t.ok(err == null, 'error should be null or undefined');
@@ -45,11 +49,17 @@ test('handle an errored stream', function(t){
 });
 
 test('handle a returned stream and cb by only calling callback once', function(t){
-  t.plan(1);
-
   asyncDone(function(cb){
     return success().on('end', function(){ cb(); });
   }, function(err){
     t.ok(err == null, 'error should be null or undefined');
+    t.end();
+  });
+});
+
+test('consumes an unpiped readable stream', function(t){
+  asyncDone(unpiped, function(err){
+    t.ok(err == null, 'error should be null or undefined');
+    t.end();
   });
 });