List view

A

test\utils\helpers.mjs

• 0 Duplication
• 2 Churn
• 62 Lines
• 48 Lines of Code

import should from 'should';
import zlib from 'zlib';
import { createServer, request } from 'http';

const helpers = {
  header:
    (key = null, value = null) =>
    (res) => {
      should(res).be.an.Object();
      should(res.headers).be.an.Object();

      if (value) {
        should(res.headers).with.property(key);
        should(res.headers[key]).be.a.String().and.equal(value);
      } else {
        should.not.exist(res.headers[key]);
      }
    },

  size: (size) => (res) => {
    should(res.text.length).be.a.Number().and.equal(size);
  },

  wrap: (module, handler, options) =>
    createServer((req, res) =>
      module(options)(req, res, (error) => (error ? res.end(error.message) : handler(req, res)))
    ),

  request: (options) => {
    return new Promise((yes) => {
      const req = request(options, (response) => {
        const data = [];
        const { statusCode, headers } = response;
        if (statusCode === 404) {
          // return no();
        }

        /**
         * This logic is mostly used for the `compression` unit tests.
         */

        const brotli = headers['content-encoding'] === 'br';
        if (brotli) {
          response.setEncoding('binary');
        }
        response.on('data', (chunk) => data.push(chunk));
        response.on('end', () => {
          if (brotli) {
            const r = zlib.brotliDecompressSync(Buffer.from(data.join(''), 'binary'));
            return yes({ status: statusCode, headers, data: r.toString() });
          }

          return yes(data.join(''));
        });
      });

      req.end();
    });
  }
};

export default helpers;