koa-lite/test/response/status.js

'use strict';

const response = require('../helpers/context').response;
const request = require('supertest');
const statuses = require('statuses');
const assert = require('assert');
const Koa = require('../..');

describe('res.status=', () => {
  describe('when a status code', () => {
    describe('and valid', () => {
      it('should set the status', () => {
        const res = response();
        res.status = 403;
        assert.equal(res.status, 403);
      });

      it('should not throw', () => {
        response().status = 403;
      });
    });

    describe('and invalid', () => {
      it('should throw', () => {
        assert.throws(() => {
          response().status = 999;
        }, /invalid status code: 999/);
      });
    });

    describe('and custom status', () => {
      beforeEach(() => statuses['700'] = 'custom status');

      it('should set the status', () => {
        const res = response();
        res.status = 700;
        assert.equal(res.status, 700);
      });

      it('should not throw', () => {
        response().status = 700;
      });
    });

    describe('and HTTP/2', () => {
      it('should not set the status message', () => {
        const res = response({
          'httpVersionMajor': 2,
          'httpVersion': '2.0'
        });
        res.status = 200;
        assert(!res.res.statusMessage);
      });
    });
  });

  describe('when a status string', () => {
    it('should throw', () => {
      assert.throws(() => response().status = 'forbidden', /status code must be a number/);
    });
  });

  function strip(status){
    it('should strip content related header fields', async () => {
      const app = new Koa();

      app.use(ctx => {
        ctx.body = { foo: 'bar' };
        ctx.set('Content-Type', 'application/json; charset=utf-8');
        ctx.set('Content-Length', '15');
        ctx.set('Transfer-Encoding', 'chunked');
        ctx.status = status;
        assert(null == ctx.response.header['content-type']);
        assert(null == ctx.response.header['content-length']);
        assert(null == ctx.response.header['transfer-encoding']);
      });

      const res = await request(app.callback())
        .get('/')
        .expect(status);

      assert.equal(res.headers.hasOwnProperty('content-type'), false);
      assert.equal(res.headers.hasOwnProperty('content-length'), false);
      assert.equal(res.headers.hasOwnProperty('content-encoding'), false);
      assert.equal(res.text.length, 0);
    });

    it('should strip content releated header fields after status set', async () => {
      const app = new Koa();

      app.use(ctx => {
        ctx.status = status;
        ctx.body = { foo: 'bar' };
        ctx.set('Content-Type', 'application/json; charset=utf-8');
        ctx.set('Content-Length', '15');
        ctx.set('Transfer-Encoding', 'chunked');
      });

      const res = await request(app.callback())
        .get('/')
        .expect(status);

      assert.equal(res.headers.hasOwnProperty('content-type'), false);
      assert.equal(res.headers.hasOwnProperty('content-length'), false);
      assert.equal(res.headers.hasOwnProperty('content-encoding'), false);
      assert.equal(res.text.length, 0);
    });
  }

  describe('when 204', () => strip(204));

  describe('when 205', () => strip(205));

  describe('when 304', () => strip(304));
});