Avoid overriding headers set in onresponse

Currently, overriding a header from the target response in the source
response requires doing it in both ondata_response (when receiving the first
chunk of data) and in onend_response (for when the response doesn't contain
any data), after verifying that res.headersSent is false. That's clumsy at
best.

By filtering the copied headers with those already present in the source
response, it's possible to set them in onresponse once and for all.

Author: fpavageau

lib/plugins-middleware.js

@@ -388,7 +388,9 @@ function handleTargetResponse(targetRequest, targetResponse, options, cb) {
             Object.keys(targetResponse.headers).forEach(function(header) {
                 // skip setting the 'connection: keep-alive' header
                 // setting it causes gateway to not accept any more connections
-                if (header !== 'connection') {
+                // Headers that have been set in onresponse are also skipped to let the plugins override those from the
+                // target response.
+                if (header !== 'connection' && !sourceResponse.hasHeader(header)) {
                     sourceResponse.setHeader(header, targetResponse.headers[header]);
                 }
             });

tests/plugin-lifecycle-tests.js

@@ -301,6 +301,27 @@ describe('test lifecycle events', function() {
       });
     });
   });
+
+  it('should keep the header set on the source response', function(done) {
+    this.timeout(20000);
+    const testPlugin = TestPlugin((type, data, cb) => cb());
+    const handler = testPlugin.init();
+    handler.onresponse = (sourceReq, sourceRes, targetRes, data, next) => {
+      sourceRes.setHeader('content-type', 'application/octet-stream');
+      sourceRes.setHeader('original-content-type', targetRes.headers['content-type']);
+      next();
+    };
+    gateway.addPlugin('test', () => handler);
+
+    gateway.start((err) => {
+      assert(!err, err);
+      request({ method: 'GET', url: 'http://localhost:' + gatewayPort + '/v1/echo/get' }, (err, r) => {
+        assert.equal(r.headers['content-type'], 'application/octet-stream');
+        assert.equal(r.headers['original-content-type'], 'application/json');
+        done();
+      });
+    });
+  });
 });
 
 function _findHeaders(headers, expectedHeaders) {