res.set.js

'use strict'

var express = require('..');
var request = require('supertest');

describe('res', function(){
  describe('.set(field, value)', function(){
    it('should set the response header field', function(done){
      var app = express();

      app.use(function(req, res){
        res.set('Content-Type', 'text/x-foo; charset=utf-8').end();
      });

      request(app)
      .get('/')
      .expect('Content-Type', 'text/x-foo; charset=utf-8')
      .end(done);
    })

    it('should coerce to a string', function (done) {
      var app = express();

      app.use(function (req, res) {
        res.set('X-Number', 123);
        res.end(typeof res.get('X-Number'));
      });

      request(app)
      .get('/')
      .expect('X-Number', '123')
      .expect(200, 'string', done);
    })
  })

  describe('.set(field, values)', function(){
    it('should set multiple response header fields', function(done){
      var app = express();

      app.use(function(req, res){
        res.set('Set-Cookie', ["type=ninja", "language=javascript"]);
        res.send(res.get('Set-Cookie'));
      });

      request(app)
      .get('/')
      .expect('["type=ninja","language=javascript"]', done);
    })

    it('should coerce to an array of strings', function (done) {
      var app = express();

      app.use(function (req, res) {
        res.set('X-Numbers', [123, 456]);
        res.end(JSON.stringify(res.get('X-Numbers')));
      });

      request(app)
      .get('/')
      .expect('X-Numbers', '123, 456')
      .expect(200, '["123","456"]', done);
    })

    it('should not set a charset of one is already set', function (done) {
      var app = express();

      app.use(function (req, res) {
        res.set('Content-Type', 'text/html; charset=lol');
        res.end();
      });

      request(app)
      .get('/')
      .expect('Content-Type', 'text/html; charset=lol')
      .expect(200, done);
    })

    it('should throw when Content-Type is an array', function (done) {
      var app = express()

      app.use(function (req, res) {
        res.set('Content-Type', ['text/html'])
        res.end()
      });

      request(app)
      .get('/')
      .expect(500, /TypeError: Content-Type cannot be set to an Array/, done)
    })

    it('should set charset for known Content-Type', function (done) {
      var app = express();

      app.use(function (req, res) {
        res.set('Content-Type', 'text/html');
        res.end();
      });

      request(app)
      .get('/')
      .expect('Content-Type', 'text/html; charset=utf-8')
      .expect(200, done);
    })

    it('should add charset for shorthand Content-Type', function (done) {
      var app = express();

      app.use(function (req, res) {
        res.set('Content-Type', 'html');
        res.end();
      });

      request(app)
      .get('/')
      .expect('Content-Type', 'text/html; charset=utf-8')
      .expect(200, done);
    })

    it('should keep original value when Content-Type is unknown', function (done) {
      var app = express();

      app.use(function (req, res) {
        res.set('Content-Type', 'some-custom-type');
        res.end();
      });

      request(app)
      .get('/')
      .expect('Content-Type', 'some-custom-type')
      .expect(200, done);
    })

    it('should not set header to false for unrecognized Content-Type', function (done) {
      var app = express();

      app.use(function (req, res) {
        res.set('Content-Type', 'unknown-type');
        var value = res.get('Content-Type');
        res.end(value);
      });

      request(app)
      .get('/')
      .expect(function (res) {
        if (res.text === 'false') {
          throw new Error('Content-Type was set to literal string "false"');
        }
      })
      .expect(200, done);
    })

    it('should preserve full MIME type with slash in Content-Type', function (done) {
      var app = express();

      app.use(function (req, res) {
        res.set('Content-Type', 'application/vnd.custom+json');
        res.end();
      });

      request(app)
      .get('/')
      .expect('Content-Type', 'application/vnd.custom+json')
      .expect(200, done);
    })
  })

  describe('.set(object)', function(){
    it('should set multiple fields', function(done){
      var app = express();

      app.use(function(req, res){
        res.set({
          'X-Foo': 'bar',
          'X-Bar': 'baz'
        }).end();
      });

      request(app)
      .get('/')
      .expect('X-Foo', 'bar')
      .expect('X-Bar', 'baz')
      .end(done);
    })

    it('should coerce to a string', function (done) {
      var app = express();

      app.use(function (req, res) {
        res.set({ 'X-Number': 123 });
        res.end(typeof res.get('X-Number'));
      });

      request(app)
      .get('/')
      .expect('X-Number', '123')
      .expect(200, 'string', done);
    })
  })
})