node-http-proxy-json/test/deflate.js

for node-http-proxy transform the response json from the proxied server.

'use strict';
/**
 * Test content-encoding for deflate
 */

const assert = require('chai').assert;

const zlib = require('zlib');
const http = require('http');
const httpProxy = require('http-proxy');
const modifyResponse = require('../');

const SERVER_PORT = 5002;
const TARGET_SERVER_PORT = 5003;

const isObject = function (obj) {
  return Object.prototype.toString.call(obj) === '[object Object]';
}

describe('modifyResponse--deflate', () => {
  let proxy, server, targetServer;
  beforeEach(() => {
    // Create a proxy server
    proxy = httpProxy.createProxyServer({
      target: 'http://localhost:' + TARGET_SERVER_PORT,
    });

    // Create your server and then proxies the request
    server = http
      .createServer((req, res) => proxy.web(req, res))
      .listen(SERVER_PORT);

    // Create your target server
    targetServer = http
      .createServer(function (req, res) {
        // Create deflated content
        let deflate = zlib.Deflate();
        let _write = res.write;
        let _end = res.end;

        deflate.on('data', buf => _write.call(res, buf));
        deflate.on('end', () => _end.call(res));

        res.write = data => deflate.write(data);
        res.end = () => deflate.end();

        res.writeHead(200, {
          'Content-Type': 'application/json',
          'Content-Encoding': 'deflate',
        });

        res.write(
          JSON.stringify({
            name: 'node-http-proxy-json',
            age: 1,
            version: '1.0.0',
          })
        );

        res.end();
      })
      .listen(TARGET_SERVER_PORT);
  });

  afterEach(() => {
    proxy.close();
    server.close();
    targetServer.close();
  });

  describe('callback returns data', () => {
    beforeEach(() => {
      // Listen for the `proxyRes` event on `proxy`.
      proxy.on('proxyRes', (proxyRes, req, res) => {
        modifyResponse(res, proxyRes, body => {
          if (isObject(body)) {
            // modify some information
            body.age = 2;
            delete body.version;
          }
          return body;
        });
      });
    });
    it('deflate: modify response json successfully', done => {
      // Test server
      http.get('http://localhost:' + SERVER_PORT, res => {
        let body = '';
        let inflate = zlib.Inflate();
        res.pipe(inflate);

        inflate
          .on('data', function (chunk) {
            body += chunk;
          })
          .on('end', function () {
            assert.equal(
              JSON.stringify({ name: 'node-http-proxy-json', age: 2 }),
              body
            );

            done();
          });
      });
    });
  });

  describe('callback returns a promise', () => {
    beforeEach(() => {
      // Listen for the `proxyRes` event on `proxy`.
      proxy.on('proxyRes', (proxyRes, req, res) => {
        modifyResponse(res, proxyRes, body => {
          if (isObject(body)) {
            // modify some information
            body.age = 2;
            delete body.version;
          }
          return Promise.resolve(body);
        });
      });
    });
    it('deflate: modify response json successfully', done => {
      // Test server
      http.get('http://localhost:' + SERVER_PORT, res => {
        let body = '';
        let inflate = zlib.Inflate();
        res.pipe(inflate);

        inflate
          .on('data', function (chunk) {
            body += chunk;
          })
          .on('end', function () {
            assert.equal(
              JSON.stringify({ name: 'node-http-proxy-json', age: 2 }),
              body
            );

            done();
          });
      });
    });
  });
});