test/support/utils.js

   1
   2 /**
   3  * Module dependencies.
   4  * @private
   5  */
   6
   7 var assert = require('assert');
   8 var Buffer = require('safe-buffer').Buffer
   9
  10 /**
  11  * Module exports.
  12  * @public
  13  */
  14
  15 exports.shouldHaveBody = shouldHaveBody
  16 exports.shouldHaveHeader = shouldHaveHeader
  17 exports.shouldNotHaveBody = shouldNotHaveBody
  18 exports.shouldNotHaveHeader = shouldNotHaveHeader;
  19
  20 /**
  21  * Assert that a supertest response has a specific body.
  22  *
  23  * @param {Buffer} buf
  24  * @returns {function}
  25  */
  26
  27 function shouldHaveBody (buf) {
  28   return function (res) {
  29     var body = !Buffer.isBuffer(res.body)
  30       ? Buffer.from(res.text)
  31       : res.body
  32     assert.ok(body, 'response has body')
  33     assert.strictEqual(body.toString('hex'), buf.toString('hex'))
  34   }
  35 }
  36
  37 /**
  38  * Assert that a supertest response does have a header.
  39  *
  40  * @param {string} header Header name to check
  41  * @returns {function}
  42  */
  43
  44 function shouldHaveHeader (header) {
  45   return function (res) {
  46     assert.ok((header.toLowerCase() in res.headers), 'should have header ' + header)
  47   }
  48 }
  49
  50 /**
  51  * Assert that a supertest response does not have a body.
  52  *
  53  * @returns {function}
  54  */
  55
  56 function shouldNotHaveBody () {
  57   return function (res) {
  58     assert.ok(res.text === '' || res.text === undefined)
  59   }
  60 }
  61
  62 /**
  63  * Assert that a supertest response does not have a header.
  64  *
  65  * @param {string} header Header name to check
  66  * @returns {function}
  67  */
  68 function shouldNotHaveHeader(header) {
  69   return function (res) {
  70     assert.ok(!(header.toLowerCase() in res.headers), 'should not have header ' + header);
  71   };
  72 }