Added tests

test/compression.js

@@ -9,6 +9,16 @@ var zlib = require('zlib')
 
 var compression = require('..')
 
+/**
+ * @const
+ * whether current node version has brotli support
+ */
+var hasBrotliSupport = false
+try {
+  hasBrotliSupport = 'brotli' in require('process').versions
+} catch (ignored) {
+}
+
 describe('compression()', function () {
   it('should skip HEAD', function (done) {
     var server = createServer({ threshold: 0 }, function (req, res) {
@@ -465,6 +475,22 @@ describe('compression()', function () {
     })
   })
 
+  if (hasBrotliSupport) {
+    describe('when "Accept-Encoding: br"', function () {
+      it('should respond with br', function (done) {
+        var server = createServer({ threshold: 0 }, function (req, res) {
+          res.setHeader('Content-Type', 'text/plain')
+          res.end('hello, world')
+        })
+
+        request(server)
+          .get('/')
+          .set('Accept-Encoding', 'br')
+          .expect('Content-Encoding', 'br', done)
+      })
+    })
+  }
+
   describe('when "Accept-Encoding: gzip, deflate"', function () {
     it('should respond with gzip', function (done) {
       var server = createServer({ threshold: 0 }, function (req, res) {
@@ -493,6 +519,22 @@ describe('compression()', function () {
     })
   })
 
+  if (hasBrotliSupport) {
+    describe('when "Accept-Encoding: deflate, br"', function () {
+      it('should respond with br', function (done) {
+        var server = createServer({ threshold: 0 }, function (req, res) {
+          res.setHeader('Content-Type', 'text/plain')
+          res.end('hello, world')
+        })
+
+        request(server)
+          .get('/')
+          .set('Accept-Encoding', 'deflate, br')
+          .expect('Content-Encoding', 'br', done)
+      })
+    })
+  }
+
   describe('when "Cache-Control: no-transform" response header', function () {
     it('should not compress response', function (done) {
       var server = createServer({ threshold: 0 }, function (req, res) {
@@ -631,6 +673,34 @@ describe('compression()', function () {
         .end()
     })
 
+    if (hasBrotliSupport) {
+      it('should flush small chunks for brotli', function (done) {
+        var chunks = 0
+        var next
+        var server = createServer({ threshold: 0 }, function (req, res) {
+          next = writeAndFlush(res, 2, Buffer.from('..'))
+          res.setHeader('Content-Type', 'text/plain')
+          next()
+        })
+
+        function onchunk (chunk) {
+          assert.ok(chunks++ < 20)
+          assert.strictEqual(chunk.toString(), '..')
+          next()
+        }
+
+        request(server)
+          .get('/')
+          .set('Accept-Encoding', 'br')
+          .request()
+          .on('response', unchunk('br', onchunk, function (err) {
+            if (err) return done(err)
+            server.close(done)
+          }))
+          .end()
+      })
+    }
+
     it('should flush small chunks for deflate', function (done) {
       var chunks = 0
       var next
@@ -710,6 +780,9 @@ function unchunk (encoding, onchunk, onend) {
       case 'gzip':
         stream = res.pipe(zlib.createGunzip())
         break
+      case 'br':
+        stream = res.pipe(zlib.createBrotliDecompress())
+        break
     }
 
     stream.on('data', onchunk)