Preset origin response code (#32)

preset the origin response code before calling onResponse hooks implementations.

Author: jkyberneees

index.js

@@ -109,10 +109,12 @@ module.exports = (opts) => {
           copyHeaders(rewriteHeaders(headers), res)
         }
 
+        // set origin response code
+        res.statusCode = statusCode
+
         if (onResponse) {
           onResponse(req, res, stream)
         } else {
-          res.statusCode = statusCode
           pump(stream, res)
         }
       })

test/4.opts.test.js

@@ -50,10 +50,15 @@ describe('fast-proxy smoke', () => {
     service = require('restana')()
     service.use(bodyParser.json())
 
-    service.get('/service/headers', (req, res) => {
-      res.setHeader('url', req.url)
-      res.send()
-    })
+    service
+      .get('/service/headers', (req, res) => {
+        res.setHeader('url', req.url)
+        res.send()
+      })
+      .get('/service/302', (req, res) => {
+        res.statusCode = 302
+        res.end()
+      })
 
     await service.start(3000)
   })
@@ -86,6 +91,12 @@ describe('fast-proxy smoke', () => {
       })
   })
 
+  it('should retrieve http status from origin', async () => {
+    await request(gHttpServer)
+      .get('/service/302')
+      .expect(302)
+  })
+
   it('close all', async () => {
     close()
     await gateway.close()