Adds handling for errors after initial response.

Includes a test.
2014-04-06 22:04:20 -07:00
parent 1822f399d2
commit 1db9b3ec3d
2 changed files with 30 additions and 3 deletions
@@ -68,10 +68,14 @@ CopyStreamQuery.prototype._transform = function(chunk, enc, cb) {
  }
  while((chunk.length - offset) > 5) {
    var messageCode = chunk[offset]
-    //complete
-    if(messageCode == code.c) {
+    //complete or error
+    if(messageCode == code.c || messageCode == code.E) {
      this._detach()
-      this.connection.stream.unshift(chunk.slice(offset + 5))
+      if (messageCode == code.c) {
+        this.connection.stream.unshift(chunk.slice(offset + 5))
+      } else {
+        this.connection.stream.unshift(chunk.slice(offset))
+      }
      this.push(null)
      return cb();
    }