From 869a4664d5b6aabe56ee31056ca2e0f0c0c15ad6 Mon Sep 17 00:00:00 2001
From: Ryan Seys <ryan@ryanseys.com>
Date: Wed, 4 Mar 2015 16:58:01 -0500
Subject: [PATCH] Emit errors from upstream on file upload

---
 lib/storage/file.js  |  5 ++++-
 test/storage/file.js | 38 ++++++++++++++++++++++++++++++++++++++
 2 files changed, 42 insertions(+), 1 deletion(-)

diff --git a/lib/storage/file.js b/lib/storage/file.js
index 8a0086a5636..0ee400643e7 100644
--- a/lib/storage/file.js
+++ b/lib/storage/file.js
@@ -988,6 +988,10 @@ File.prototype.startResumableUpload_ = function(stream, metadata) {
     var writeStream = request(reqOpts);
     writeStream.callback = util.noop;
 
+    writeStream.on('error', function(err) {
+      handleError(err);
+    });
+
     writeStream.on('complete', function(res) {
       util.handleResp(null, res, res.body, function(err, data) {
         if (err) {
@@ -1056,7 +1060,6 @@ File.prototype.startResumableUpload_ = function(stream, metadata) {
     }
 
     stream.emit('error', err);
-    stream.end();
   }
 };
 
diff --git a/test/storage/file.js b/test/storage/file.js
index c181248e8db..99bd8c1e5c7 100644
--- a/test/storage/file.js
+++ b/test/storage/file.js
@@ -528,6 +528,44 @@ describe('File', function() {
       writable.write('data');
     });
 
+    it('should re-emit errors', function(done) {
+      var error = new Error('Error.');
+      var requestCount = 0;
+      file.bucket.storage.makeAuthorizedRequest_ = function(reqOpts, cb) {
+        requestCount++;
+
+        // respond to creation POST.
+        if (requestCount === 1) {
+          cb(null, null, { headers: { location: 'http://resume' }});
+          return;
+        }
+
+        // create an authorized request for the first PUT.
+        if (requestCount === 2) {
+          cb.onAuthorized(null, { headers: {} });
+        }
+      };
+
+      // respond to first upload PUT request.
+      request_Override = function() {
+        var stream = through();
+        setImmediate(function() {
+          stream.emit('error', error);
+        });
+        return stream;
+      };
+
+      var stream = duplexify();
+
+      stream
+        .on('error', function(err) {
+          assert.equal(err, error);
+          done();
+        });
+
+      file.startResumableUpload_(stream);
+    });
+
     it('should start a simple upload if specified', function(done) {
       var writable = file.createWriteStream({
         metadata: METADATA,