yakbak-enhanced/test/record.js

Record and playback HTTP responses

// Copyright 2019 SmugMug, Inc.
// Licensed under the terms of the MIT license. Please see LICENSE file in the project root for terms.

/* eslint-env mocha */

var subject = require('../lib/record');
var createServer = require('./helpers/server');
var createTmpdir = require('./helpers/tmpdir');
var assert = require('assert');
var http = require('http');
var fs = require('fs');

var fixture = require('./fixtures');

describe('record', function () {
  var server, tmpdir, req;

  beforeEach(function (done) {
    server = createServer(done);
  });

  afterEach(function (done) {
    server.teardown(done);
  });

  beforeEach(function () {
    tmpdir = createTmpdir();
  });

  afterEach(function () {
    tmpdir.teardown();
  });

  beforeEach(function () {
    req = http.request({
      host: server.addr,
      port: server.port
    });
    req.setHeader('User-Agent', 'My User Agent/1.0');
    req.setHeader('Connection', 'close');
  });

  it('returns the filename', function (done) {
    req.on('response', function (res) {
      subject(req, res, tmpdir.join('foo.js')).then(function (filename) {
        assert.equal(filename, tmpdir.join('foo.js'));
        done();
      }).catch(function (err) {
        done(err);
      });
    });

    req.end();
  });

  it('records the response to disk', function (done) {
    var expected = fixture.replace('{addr}', server.addr).replace('{port}', server.port);

    req.on('response', function (res) {
      subject(req, res, tmpdir.join('foo.js')).then(function (filename) {
        assert.equal(fs.readFileSync(filename, 'utf8'), expected);
        done();
      }).catch(function (err) {
        done(err);
      });
    });

    req.end();
  });

});