issue: #33 - Should support undici timeout err.code (#34)

Author: yohayg

index.js

@@ -88,7 +88,7 @@ module.exports = (opts) => {
             if (err.code === 'ECONNREFUSED' || err.code === 'ERR_HTTP2_STREAM_CANCEL') {
               res.statusCode = 503
               res.end('Service Unavailable')
-            } else if (err.code === 'ECONNRESET') {
+            } else if (err.code === 'ECONNRESET' || err.code === 'UND_ERR_REQUEST_TIMEOUT') {
               res.statusCode = 504
               res.end(err.message)
             } else {

test/5.undici.test.js

@@ -1,17 +1,21 @@
 /* global describe, it */
 'use strict'
-
 const request = require('supertest')
 const bodyParser = require('body-parser')
+const { promisify } = require('util')
 const { expect } = require('chai')
+
+const sleep = promisify(setTimeout)
+
 let gateway, service, close, proxy, gHttpServer
 
 describe('undici', () => {
   it('init', async () => {
     const fastProxy = require('../index')({
       base: 'http://127.0.0.1:3000',
       undici: {
-        pipelining: 10
+        pipelining: 10,
+        requestTimeout: 100
       }
     })
     close = fastProxy.close
@@ -45,6 +49,10 @@ describe('undici', () => {
       res.setHeader('x-agent', 'fast-proxy')
       res.send()
     })
+    service.get('/service/timeout', async (req, res) => {
+      await sleep(200)
+      res.send()
+    })
 
     service.start(3000).then(() => done())
   })
@@ -75,6 +83,12 @@ describe('undici', () => {
       })
   })
 
+  it('should 200 on GET /service/timeout', async () => {
+    await request(gHttpServer)
+      .get('/service/timeout')
+      .expect(504)
+  })
+
   it('close all', async () => {
     close()
     await gateway.close()