koa-lite/test/response/header.js

45 lines
1.1 KiB
JavaScript

'use strict';
const request = require('supertest');
const response = require('../helpers/context').response;
const Koa = require('../..');
describe('res.header', () => {
it('should return the response header object', () => {
const res = response();
res.set('X-Foo', 'bar');
res.header.should.eql({ 'x-foo': 'bar' });
});
it('should use res.getHeaders() accessor when available', () => {
const res = response();
res.res._headers = null;
res.res.getHeaders = () => ({ 'x-foo': 'baz' });
res.header.should.eql({ 'x-foo': 'baz' });
});
it('should return the response header object when no mocks are in use', async () => {
const app = new Koa();
let header;
app.use(ctx => {
ctx.set('x-foo', '42');
header = Object.assign({}, ctx.response.header);
});
await request(app.listen())
.get('/');
header.should.eql({ 'x-foo': '42' });
});
describe('when res._headers not present', () => {
it('should return empty object', () => {
const res = response();
res.res._headers = null;
res.header.should.eql({});
});
});
});