Override Response.setTimeout to make response timeout handler work (#69)

* Override Response,setTimeout to make response timeout handler work

* Fixes after review

* Response.setTimeout emits timeout event on response instead of response.socket

Author: gyszalai

lib/response.js

@@ -46,6 +46,14 @@ function Response (req, onEnd, reject) {
 
 util.inherits(Response, http.ServerResponse)
 
+Response.prototype.setTimeout = function (msecs, callback) {
+  this.timeoutHandle = setTimeout(() => {
+    this.emit('timeout')
+  }, msecs)
+  this.on('timeout', callback)
+  return this
+}
+
 Response.prototype.writeHead = function () {
   const result = http.ServerResponse.prototype.writeHead.apply(this, arguments)
 
@@ -65,6 +73,9 @@ Response.prototype.writeHead = function () {
 }
 
 Response.prototype.write = function (data, encoding, callback) {
+  if (this.timeoutHandle) {
+    clearTimeout(this.timeoutHandle)
+  }
   http.ServerResponse.prototype.write.call(this, data, encoding, callback)
   this._lightMyRequest.payloadChunks.push(Buffer.from(data, encoding))
   return true

test/test.js

@@ -847,6 +847,28 @@ test('should handle response errors (promises)', (t) => {
     .catch(err => t.ok(err))
 })
 
+test('should handle response timeout handler', (t) => {
+  t.plan(3)
+  const dispatch = function (req, res) {
+    const handle = setTimeout(() => {
+      res.writeHead(200, { 'Content-Type': 'text/plain' })
+      res.end('incorrect')
+    }, 200)
+    res.setTimeout(100, () => {
+      clearTimeout(handle)
+      res.writeHead(200, { 'Content-Type': 'text/plain' })
+      res.end('correct')
+    })
+    res.on('timeout', () => {
+      t.ok(true, 'Response timeout event not emitted')
+    })
+  }
+  inject(dispatch, { method: 'GET', url: '/test' }, (err, res) => {
+    t.error(err)
+    t.equal(res.payload, 'correct')
+  })
+})
+
 test('should throw on unknown HTTP method', (t) => {
   t.plan(1)
   const dispatch = function (req, res) { }