116 lines
3.3 KiB
JavaScript
116 lines
3.3 KiB
JavaScript
|
|
'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 = 99;
|
|
}, /invalid status code: 99/);
|
|
});
|
|
});
|
|
|
|
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));
|
|
});
|