koa-lite/test/response/redirect.js

119 lines
3.5 KiB
JavaScript
Raw Normal View History

'use strict';
2017-05-11 03:30:32 +00:00
const assert = require('assert');
const context = require('../helpers/context');
describe('ctx.redirect(url)', () => {
it('should redirect to the given url', () => {
const ctx = context();
ctx.redirect('http://google.com');
2017-05-11 03:30:32 +00:00
assert.equal(ctx.response.header.location, 'http://google.com');
assert.equal(ctx.status, 302);
2015-10-12 20:36:41 +00:00
});
describe('with "back"', () => {
it('should redirect to Referrer', () => {
const ctx = context();
ctx.req.headers.referrer = '/login';
ctx.redirect('back');
2017-05-11 03:30:32 +00:00
assert.equal(ctx.response.header.location, '/login');
2015-10-12 20:36:41 +00:00
});
it('should redirect to Referer', () => {
const ctx = context();
ctx.req.headers.referer = '/login';
ctx.redirect('back');
2017-05-11 03:30:32 +00:00
assert.equal(ctx.response.header.location, '/login');
2015-10-12 20:36:41 +00:00
});
it('should default to alt', () => {
const ctx = context();
ctx.redirect('back', '/index.html');
2017-05-11 03:30:32 +00:00
assert.equal(ctx.response.header.location, '/index.html');
2015-10-12 20:36:41 +00:00
});
it('should default redirect to /', () => {
const ctx = context();
ctx.redirect('back');
2017-05-11 03:30:32 +00:00
assert.equal(ctx.response.header.location, '/');
2015-10-12 20:36:41 +00:00
});
});
describe('when html is accepted', () => {
it('should respond with html', () => {
const ctx = context();
const url = 'http://google.com';
ctx.header.accept = 'text/html';
ctx.redirect(url);
2017-05-11 03:30:32 +00:00
assert.equal(ctx.response.header['content-type'], 'text/html; charset=utf-8');
assert.equal(ctx.body, `Redirecting to <a href="${url}">${url}</a>.`);
2015-10-12 20:36:41 +00:00
});
it('should escape the url', () => {
const ctx = context();
2015-10-22 22:46:47 +00:00
let url = '<script>';
ctx.header.accept = 'text/html';
ctx.redirect(url);
url = escape(url);
2017-05-11 03:30:32 +00:00
assert.equal(ctx.response.header['content-type'], 'text/html; charset=utf-8');
assert.equal(ctx.body, `Redirecting to <a href="${url}">${url}</a>.`);
2015-10-12 20:36:41 +00:00
});
});
describe('when text is accepted', () => {
it('should respond with text', () => {
const ctx = context();
const url = 'http://google.com';
ctx.header.accept = 'text/plain';
ctx.redirect(url);
2017-05-11 03:30:32 +00:00
assert.equal(ctx.body, `Redirecting to ${url}.`);
2015-10-12 20:36:41 +00:00
});
});
describe('when status is 301', () => {
it('should not change the status code', () => {
const ctx = context();
const url = 'http://google.com';
ctx.status = 301;
ctx.header.accept = 'text/plain';
ctx.redirect('http://google.com');
2017-05-11 03:30:32 +00:00
assert.equal(ctx.status, 301);
assert.equal(ctx.body, `Redirecting to ${url}.`);
2015-10-12 20:36:41 +00:00
});
});
describe('when status is 304', () => {
it('should change the status code', () => {
const ctx = context();
const url = 'http://google.com';
ctx.status = 304;
ctx.header.accept = 'text/plain';
ctx.redirect('http://google.com');
2017-05-11 03:30:32 +00:00
assert.equal(ctx.status, 302);
assert.equal(ctx.body, `Redirecting to ${url}.`);
2015-10-12 20:36:41 +00:00
});
});
describe('when content-type was present', () => {
it('should overwrite content-type', () => {
const ctx = context();
ctx.body = {};
const url = 'http://google.com';
ctx.header.accept = 'text/plain';
ctx.redirect('http://google.com');
2017-05-11 03:30:32 +00:00
assert.equal(ctx.status, 302);
assert.equal(ctx.body, `Redirecting to ${url}.`);
assert.equal(ctx.type, 'text/plain');
2015-10-12 20:36:41 +00:00
});
});
});
2015-10-12 20:36:41 +00:00
function escape(html){
return String(html)
.replace(/&/g, '&amp;')
.replace(/"/g, '&quot;')
.replace(/</g, '&lt;')
.replace(/>/g, '&gt;');
2014-03-24 18:21:15 +00:00
}