update stream clearing to be compatible with node 0.12

src/lib/utils.js

@@ -397,11 +397,13 @@ _.createArray = function (input, transform) {
  * @return {string} - the remaining test to be written to the stream
  */
 _.getUnwrittenFromStream = function (stream) {
-  if (stream && stream._writableState && stream._writableState.buffer) {
+  var writeBuffer = _.getStreamWriteBuffer(stream);
-    // flush the write buffer to disk
+  if (!writeBuffer || !writeBuffer.length) {
-    var writeBuffer = stream._writableState.buffer;
+    return '';
+  }
+
+  // flush the write buffer
   var out = '';
-    if (writeBuffer.length) {
   _.each(writeBuffer, function (writeReq) {
     if (writeReq.chunk) {
       // 0.9.12+ uses WriteReq objects with a chunk prop
@@ -413,11 +415,26 @@ _.getUnwrittenFromStream = function (stream) {
       return false;
     }
   });
-    }
   return out;
+};
+
+_.getStreamWriteBuffer = function (stream) {
+  if (!stream || !stream._writableState) return;
+
+  var writeState = stream._writableState;
+
+  if (writeState.getBuffer) {
+    return writeState.getBuffer();
+  } else if (writeState.buffer) {
+    return writeState.buffer;
   }
 };
 
+_.clearWriteStreamBuffer = function (stream) {
+  var buffer = _.emptyWriteStreamBuffer(stream);
+  return buffer && buffer.splice(0);
+};
+
 /**
  * return the current time in milliseconds since epoch
  */

test/unit/specs/connection_pool.js

@@ -247,7 +247,6 @@ describe('Connection Pool', function () {
       var clock = sinon.useFakeTimers();
       stub.autoRelease(clock);
 
-      debugger;
       connection.setStatus('dead');
       expect(_.size(clock.timers)).to.eql(1);
       var id = _(clock.timers).keys().first();

test/unit/specs/file_logger.js

@@ -15,15 +15,7 @@ describe('File Logger', function () {
 
   afterEach(function () {
     parentLog.close();
-
+    logger && _.clearWriteStreamBuffer(logger.stream);
-    if (logger
-      && logger.stream
-      && logger.stream._writableState
-      && logger.stream._writableState.buffer.length
-    ) {
-      // empty the buffer manually
-      logger.stream._writableState.buffer.splice(0);
-    }
   });
 
   function makeLogger(parent, levels) {

test/unit/specs/stream_logger.js

@@ -19,11 +19,7 @@ describe('Stream Logger', function () {
 
   afterEach(function () {
     parentLog.close();
-
+    _.clearWriteStreamBuffer(stream);
-    if (stream._writableState && stream._writableState.buffer.length) {
-      // empty the buffer manually
-      stream._writableState.buffer.splice(0);
-    }
   });
 
   function makeLogger(parent, levels) {