mirror of https://github.com/pelias/api.git
Stephen Hess
8 years ago
3 changed files with 108 additions and 0 deletions
@ -0,0 +1,19 @@
|
||||
const _ = require('lodash'); |
||||
|
||||
function setup(comparator) { |
||||
function middleware(req, res, next) { |
||||
// do nothing if there's nothing to do
|
||||
if (_.isEmpty(_.get(res, 'data', []))) { |
||||
return next(); |
||||
} |
||||
|
||||
res.data = res.data.sort(comparator); |
||||
|
||||
next(); |
||||
} |
||||
|
||||
return middleware; |
||||
|
||||
} |
||||
|
||||
module.exports = setup; |
@ -0,0 +1,88 @@
|
||||
const sortResponseData = require('../../../middleware/sortResponseData'); |
||||
|
||||
module.exports.tests = {}; |
||||
|
||||
module.exports.tests.doIt = (test, common) => { |
||||
test('undefined res should return without interacting with comparator', (t) => { |
||||
const comparator = () => { |
||||
throw Error('should not have been called'); |
||||
}; |
||||
|
||||
const sort = sortResponseData(comparator); |
||||
|
||||
sort(undefined, undefined, () => { |
||||
t.end(); |
||||
}); |
||||
|
||||
}); |
||||
|
||||
test('undefined res.data should return without interacting with comparator', (t) => { |
||||
const comparator = () => { |
||||
throw Error('should not have been called'); |
||||
}; |
||||
|
||||
const sort = sortResponseData(comparator); |
||||
|
||||
const res = {}; |
||||
|
||||
sort(undefined, res, () => { |
||||
t.deepEquals(res, {}); |
||||
t.end(); |
||||
}); |
||||
|
||||
}); |
||||
|
||||
test('empty res.data should not cause problems', (t) => { |
||||
const comparator = () => { |
||||
throw Error('should not have been called'); |
||||
}; |
||||
|
||||
const sort = sortResponseData(comparator); |
||||
|
||||
const res = { |
||||
data: [] |
||||
}; |
||||
|
||||
sort(undefined, res, () => { |
||||
t.deepEquals(res.data, [], 'res.data should still be empty'); |
||||
t.end(); |
||||
}); |
||||
|
||||
}); |
||||
|
||||
test('comparator should be consulted for sorting res.data when defined', (t) => { |
||||
const comparator = (a, b) => { |
||||
return a.key > b.key; |
||||
}; |
||||
|
||||
const sort = sortResponseData(comparator); |
||||
|
||||
const res = { |
||||
data: [ |
||||
{ key: 3 }, |
||||
{ key: 2 }, |
||||
{ key: 1 }, |
||||
] |
||||
}; |
||||
|
||||
sort(undefined, res, () => { |
||||
t.deepEquals(res.data.shift(), { key: 1 }); |
||||
t.deepEquals(res.data.shift(), { key: 2 }); |
||||
t.deepEquals(res.data.shift(), { key: 3 }); |
||||
t.end(); |
||||
}); |
||||
|
||||
}); |
||||
|
||||
}; |
||||
|
||||
module.exports.all = (tape, common) => { |
||||
|
||||
function test(name, testFunction) { |
||||
return tape(`[middleware] sortResponseData: ${name}`, testFunction); |
||||
} |
||||
|
||||
for( var testCase in module.exports.tests ){ |
||||
module.exports.tests[testCase](test, common); |
||||
} |
||||
}; |
Loading…
Reference in new issue