mirror of https://github.com/pelias/api.git
Stephen K Hess
8 years ago
committed by
GitHub
23 changed files with 3220 additions and 120 deletions
@ -0,0 +1,271 @@
|
||||
const _ = require('lodash'); |
||||
const logger = require('pelias-logger').get('api'); |
||||
const Document = require('pelias-model').Document; |
||||
const geolib = require('geolib'); |
||||
|
||||
// composition of toNumber and isFinite, useful for single call to convert a value
|
||||
// to a number, then checking to see if it's finite
|
||||
function isFiniteNumber(value) { |
||||
return !_.isEmpty(_.trim(value)) && _.isFinite(_.toNumber(value)); |
||||
} |
||||
|
||||
// returns true if value is parseable as finite non-negative number
|
||||
function isNonNegativeFiniteNumber(value) { |
||||
return isFiniteNumber(value) && _.gte(value, 0); |
||||
} |
||||
|
||||
function hasLatLon(result) { |
||||
return _.isFinite(_.get(result.geom, 'lat')) && _.isFinite(_.get(result.geom, 'lon')); |
||||
} |
||||
|
||||
function getLatLon(result) { |
||||
return { |
||||
latitude: result.geom.lat, |
||||
longitude: result.geom.lon |
||||
}; |
||||
} |
||||
|
||||
// if geom.lat/lon are parseable as finite numbers, convert to a finite number
|
||||
// otherwise remove the field
|
||||
function numberifyGeomLatLon(result) { |
||||
['lat', 'lon'].forEach((f) => { |
||||
if (isFiniteNumber(_.get(result.geom, f))) { |
||||
result.geom[f] = _.toFinite(result.geom[f]); |
||||
} else { |
||||
// result.geom may not exist, so use unset instead of delete
|
||||
_.unset(result.geom, f); |
||||
} |
||||
}); |
||||
|
||||
return result; |
||||
|
||||
} |
||||
|
||||
// returns true if all 4 ,-delimited (max) substrings are parseable as finite numbers
|
||||
// '12.12,21.21,13.13,31.31' returns true
|
||||
// '12.12,21.21,13.13,31.31,14.14' returns false
|
||||
// '12.12,21.21,13.13,blah' returns false
|
||||
// '12.12,21.21,13.13,31.31,blah' returns false
|
||||
// '12.12,NaN,13.13,31.31' returns false
|
||||
// '12.12,Infinity,13.13,31.31' returns false
|
||||
function is4CommaDelimitedNumbers(bbox) { |
||||
return _.defaultTo(bbox, ''). |
||||
split(','). |
||||
filter(isFiniteNumber).length === 4; |
||||
} |
||||
|
||||
function hasName(result) { |
||||
return !_.isEmpty(_.trim(result.name)); |
||||
} |
||||
|
||||
// filter that passes only results that match on requested layers
|
||||
// passes everything if req.clean.layers is not found
|
||||
function getLayersFilter(clean) { |
||||
if (_.isEmpty(_.get(clean, 'layers', []))) { |
||||
return _.constant(true); |
||||
} |
||||
|
||||
// otherwise return a function that checks for set inclusion of a result placetype
|
||||
return (result) => { |
||||
return _.includes(clean.layers, result.placetype); |
||||
}; |
||||
|
||||
} |
||||
|
||||
// return true if the hierarchy does not have a country.abbr
|
||||
// OR hierarchy country.abbr matches boundary.country
|
||||
function matchesBoundaryCountry(boundaryCountry, hierarchy) { |
||||
return !boundaryCountry || _.get(hierarchy, 'country.abbr') === boundaryCountry; |
||||
} |
||||
|
||||
// return true if the result does not have a lineage
|
||||
// OR at least one lineage matches the requested boundary.country
|
||||
function atLeastOneLineageMatchesBoundaryCountry(boundaryCountry, result) { |
||||
return !result.lineage || result.lineage.some(_.partial(matchesBoundaryCountry, boundaryCountry)); |
||||
} |
||||
|
||||
// return a function that detects if a result has at least one lineage in boundary.country
|
||||
// if there's no boundary.country, return a function that always returns true
|
||||
function getBoundaryCountryFilter(clean) { |
||||
if (_.has(clean, 'boundary.country')) { |
||||
return _.partial(atLeastOneLineageMatchesBoundaryCountry, clean['boundary.country']); |
||||
} |
||||
|
||||
return _.constant(true); |
||||
|
||||
} |
||||
|
||||
// return a function that detects if a result is inside a bbox if a bbox is available
|
||||
// if there's no bbox, return a function that always returns true
|
||||
function getBoundaryRectangleFilter(clean) { |
||||
if (['min_lat', 'min_lon', 'max_lat', 'max_lon'].every((f) => { |
||||
return _.has(clean, `boundary.rect.${f}`); |
||||
})) { |
||||
const polygon = [ |
||||
{ latitude: clean['boundary.rect.min_lat'], longitude: clean['boundary.rect.min_lon'] }, |
||||
{ latitude: clean['boundary.rect.max_lat'], longitude: clean['boundary.rect.min_lon'] }, |
||||
{ latitude: clean['boundary.rect.max_lat'], longitude: clean['boundary.rect.max_lon'] }, |
||||
{ latitude: clean['boundary.rect.min_lat'], longitude: clean['boundary.rect.max_lon'] } |
||||
]; |
||||
const isPointInsidePolygon = _.partialRight(geolib.isPointInside, polygon); |
||||
|
||||
return _.partial(isInsideGeometry, isPointInsidePolygon); |
||||
|
||||
} |
||||
|
||||
return _.constant(true); |
||||
|
||||
} |
||||
|
||||
// return a function that detects if a result is inside a circle if a circle is available
|
||||
// if there's no circle, return a function that always returns true
|
||||
function getBoundaryCircleFilter(clean) { |
||||
if (['lat', 'lon', 'radius'].every((f) => { |
||||
return _.has(clean, `boundary.circle.${f}`); |
||||
})) { |
||||
const center = { |
||||
latitude: clean['boundary.circle.lat'], |
||||
longitude: clean['boundary.circle.lon'] |
||||
}; |
||||
const radiusInMeters = clean['boundary.circle.radius'] * 1000; |
||||
const isPointInCircle = _.partialRight(geolib.isPointInCircle, center, radiusInMeters); |
||||
|
||||
return _.partial(isInsideGeometry, isPointInCircle); |
||||
|
||||
} |
||||
|
||||
return _.constant(true); |
||||
|
||||
} |
||||
|
||||
// helper that calls an "is inside some geometry" function
|
||||
function isInsideGeometry(f, result) { |
||||
return hasLatLon(result) ? f(getLatLon(result)) : false; |
||||
} |
||||
|
||||
function placetypeHasNameAndId(hierarchyElement) { |
||||
return !_.isEmpty(_.trim(hierarchyElement.name)) && |
||||
!_.isEmpty(_.trim(hierarchyElement.id)); |
||||
} |
||||
|
||||
// synthesize an ES doc from a placeholder result
|
||||
function synthesizeDocs(boundaryCountry, result) { |
||||
const doc = new Document('whosonfirst', result.placetype, result.id.toString()); |
||||
doc.setName('default', result.name); |
||||
|
||||
// only assign centroid if both lat and lon are finite numbers
|
||||
if (hasLatLon(result)) { |
||||
doc.setCentroid( { lat: result.geom.lat, lon: result.geom.lon } ); |
||||
} else { |
||||
logger.error(`could not parse centroid for id ${result.id}`); |
||||
} |
||||
|
||||
// lodash conformsTo verifies that an object has a property with a certain format
|
||||
if (_.conformsTo(result.geom, { 'bbox': is4CommaDelimitedNumbers } )) { |
||||
const parsedBoundingBox = result.geom.bbox.split(',').map(_.toFinite); |
||||
doc.setBoundingBox({ |
||||
upperLeft: { |
||||
lat: parsedBoundingBox[3], |
||||
lon: parsedBoundingBox[0] |
||||
}, |
||||
lowerRight: { |
||||
lat: parsedBoundingBox[1], |
||||
lon: parsedBoundingBox[2] |
||||
} |
||||
}); |
||||
} else { |
||||
logger.error(`could not parse bbox for id ${result.id}: ${_.get(result, 'geom.bbox')}`); |
||||
} |
||||
|
||||
// set population and popularity if parseable as finite number
|
||||
if (isNonNegativeFiniteNumber(result.population)) { |
||||
doc.setPopulation(_.toFinite(result.population)); |
||||
} |
||||
if (isNonNegativeFiniteNumber(result.popularity)) { |
||||
doc.setPopularity(_.toFinite(result.popularity)); |
||||
} |
||||
|
||||
_.defaultTo(result.lineage, []) |
||||
// remove all lineages that don't match an explicit boundary.country
|
||||
.filter(_.partial(matchesBoundaryCountry, boundaryCountry)) |
||||
// add all the lineages to the doc
|
||||
.map((hierarchy) => { |
||||
Object.keys(hierarchy) |
||||
.filter(doc.isSupportedParent) |
||||
.filter((placetype) => { |
||||
return placetypeHasNameAndId(hierarchy[placetype]); |
||||
}) |
||||
.forEach((placetype) => { |
||||
doc.addParent( |
||||
placetype, |
||||
hierarchy[placetype].name, |
||||
hierarchy[placetype].id.toString(), |
||||
hierarchy[placetype].abbr); |
||||
}); |
||||
}); |
||||
|
||||
return buildESDoc(doc); |
||||
|
||||
} |
||||
|
||||
function buildESDoc(doc) { |
||||
const esDoc = doc.toESDocument(); |
||||
return _.extend(esDoc.data, { _id: esDoc._id, _type: esDoc._type }); |
||||
} |
||||
|
||||
function setup(placeholderService, should_execute) { |
||||
function controller( req, res, next ){ |
||||
// bail early if req/res don't pass conditions for execution
|
||||
if (!should_execute(req, res)) { |
||||
return next(); |
||||
} |
||||
|
||||
placeholderService(req, (err, results) => { |
||||
if (err) { |
||||
// bubble up an error if one occurred
|
||||
if (_.isObject(err) && err.message) { |
||||
req.errors.push( err.message ); |
||||
} else { |
||||
req.errors.push( err ); |
||||
} |
||||
|
||||
} else { |
||||
const boundaryCountry = _.get(req, ['clean', 'boundary.country']); |
||||
|
||||
// convert results to ES docs
|
||||
// boundary.country filter must happen after synthesis since multiple
|
||||
// lineages may produce different country docs
|
||||
res.meta = {}; |
||||
res.data = results |
||||
// filter out results that don't have a name
|
||||
.filter(hasName) |
||||
// filter out results that don't match on requested layer(s)
|
||||
.filter(getLayersFilter(req.clean)) |
||||
// filter out results that don't match on any lineage country
|
||||
.filter(getBoundaryCountryFilter(req.clean)) |
||||
// clean up geom.lat/lon for boundary rect/circle checks
|
||||
.map(numberifyGeomLatLon) |
||||
// filter out results that aren't in the boundary.rect
|
||||
.filter(getBoundaryRectangleFilter(req.clean)) |
||||
// filter out results that aren't in the boundary.circle
|
||||
.filter(getBoundaryCircleFilter(req.clean)) |
||||
// convert results to ES docs
|
||||
.map(_.partial(synthesizeDocs, boundaryCountry)); |
||||
|
||||
const messageParts = [ |
||||
'[controller:placeholder]', |
||||
`[result_count:${_.defaultTo(res.data, []).length}]` |
||||
]; |
||||
|
||||
logger.info(messageParts.join(' ')); |
||||
} |
||||
|
||||
return next(); |
||||
}); |
||||
|
||||
} |
||||
|
||||
return controller; |
||||
} |
||||
|
||||
module.exports = setup; |
@ -0,0 +1,19 @@
|
||||
const _ = require('lodash'); |
||||
|
||||
// returns a function that returns true if any result.layer is in any of the
|
||||
// supplied layers using array intersection
|
||||
|
||||
// example usage: determining if the response contains only admin results
|
||||
|
||||
module.exports = (layers) => { |
||||
return (request, response) => { |
||||
return !_.isEmpty( |
||||
_.intersection( |
||||
// convert layers to an array if it isn't already one
|
||||
_.castArray(layers), |
||||
// pull all the layer properties into an array
|
||||
_.map(response.data, _.property('layer')) |
||||
)); |
||||
}; |
||||
|
||||
}; |
@ -0,0 +1,13 @@
|
||||
const _ = require('lodash'); |
||||
|
||||
module.exports = (request, response) => { |
||||
if (!request.clean.hasOwnProperty('parsed_text')) { |
||||
return false; |
||||
} |
||||
|
||||
// return true only if all non-admin properties of parsed_text are empty
|
||||
return ['number', 'street', 'query', 'category'].every((prop) => { |
||||
return _.isEmpty(request.clean.parsed_text[prop]); |
||||
}); |
||||
|
||||
}; |
@ -1,7 +0,0 @@
|
||||
module.exports = (uri) => { |
||||
// this predicate relies upon the fact that the schema has already validated
|
||||
// that api.pipService is a URI-formatted string
|
||||
return (request, response) => { |
||||
return uri !== undefined; |
||||
}; |
||||
}; |
@ -0,0 +1,5 @@
|
||||
module.exports = (uri) => { |
||||
return (request, response) => { |
||||
return uri !== undefined; |
||||
}; |
||||
}; |
@ -0,0 +1,35 @@
|
||||
const _ = require('lodash'); |
||||
|
||||
const logger = require('pelias-logger').get('api'); |
||||
|
||||
function setup(comparator, should_execute) { |
||||
function middleware(req, res, next) { |
||||
// bail early if req/res don't pass conditions for execution or there's no data to sort
|
||||
if (!should_execute(req, res) || _.isEmpty(res.data)) { |
||||
return next(); |
||||
} |
||||
|
||||
// capture the pre-sort order
|
||||
const presort_order = res.data.map(_.property('_id')); |
||||
|
||||
// sort operates on array in place
|
||||
res.data.sort(comparator(req.clean)); |
||||
|
||||
// capture the post-sort order
|
||||
const postsort_order = res.data.map(_.property('_id')); |
||||
|
||||
// log it for debugging purposes
|
||||
logger.debug([ |
||||
`req.clean: ${JSON.stringify(req.clean)}`, |
||||
`pre-sort: [${presort_order}]`, |
||||
`post-sort: [${postsort_order}]` |
||||
].join(', ')); |
||||
|
||||
next(); |
||||
} |
||||
|
||||
return middleware; |
||||
|
||||
} |
||||
|
||||
module.exports = setup; |
@ -0,0 +1,37 @@
|
||||
const _ = require('lodash'); |
||||
|
||||
const non_admin_fields = ['number', 'street', 'query', 'category']; |
||||
|
||||
function hasAnyNonAdminFields(parsed_text) { |
||||
return !_.isEmpty( |
||||
_.intersection( |
||||
_.keys(parsed_text), |
||||
non_admin_fields)); |
||||
} |
||||
|
||||
function sanitize( raw, clean ){ |
||||
// error & warning messages
|
||||
const messages = { errors: [], warnings: [] }; |
||||
|
||||
// bail early if analysis isn't admin-only
|
||||
if (_.isUndefined(clean.parsed_text) || hasAnyNonAdminFields(clean.parsed_text)) { |
||||
return messages; |
||||
} |
||||
|
||||
// the analysis is admin-only, so add errors or warnings if geonames was requested
|
||||
if (_.isEqual(clean.sources, ['geonames'])) { |
||||
// if requested sources is only geonames, return an error
|
||||
messages.errors.push('input contains only administrative area data, ' + |
||||
'no results will be returned when sources=geonames'); |
||||
|
||||
} else if (_.includes(clean.sources, 'geonames')) { |
||||
// if there are other sources besides geonames, return an warning
|
||||
messages.warnings.push('input contains only administrative area data, ' + |
||||
'geonames results will not be returned'); |
||||
|
||||
} |
||||
|
||||
return messages; |
||||
} |
||||
|
||||
module.exports = sanitize; |
@ -0,0 +1,32 @@
|
||||
'use strict'; |
||||
|
||||
const url = require('url'); |
||||
|
||||
const _ = require('lodash'); |
||||
|
||||
const ServiceConfiguration = require('pelias-microservice-wrapper').ServiceConfiguration; |
||||
|
||||
class PlaceHolder extends ServiceConfiguration { |
||||
constructor(o) { |
||||
super('placeholder', o); |
||||
} |
||||
|
||||
getParameters(req) { |
||||
const parameters = { |
||||
text: req.clean.text |
||||
}; |
||||
|
||||
if (_.has(req.clean, 'lang.iso6393')) { |
||||
parameters.lang = req.clean.lang.iso6393; |
||||
} |
||||
|
||||
return parameters; |
||||
} |
||||
|
||||
getUrl(req) { |
||||
return url.resolve(this.baseUrl, 'search'); |
||||
} |
||||
|
||||
} |
||||
|
||||
module.exports = PlaceHolder; |
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,122 @@
|
||||
'use strict'; |
||||
|
||||
const _ = require('lodash'); |
||||
const has_results_at_layers = require('../../../../controller/predicates/has_results_at_layers'); |
||||
|
||||
module.exports.tests = {}; |
||||
|
||||
module.exports.tests.interface = (test, common) => { |
||||
test('valid interface', (t) => { |
||||
t.equal(typeof has_results_at_layers, 'function', 'has_results_at_layers is a function'); |
||||
t.equal(has_results_at_layers.length, 1); |
||||
t.end(); |
||||
}); |
||||
}; |
||||
|
||||
module.exports.tests.true_conditions = (test, common) => { |
||||
test('should return true when any result.layer matches any layer in array', (t) => { |
||||
const req = {}; |
||||
const res = { |
||||
data: [ |
||||
{ |
||||
layer: 'layer 1' |
||||
}, |
||||
{ |
||||
layer: 'layer 2' |
||||
}, |
||||
{ |
||||
layer: 'layer 3' |
||||
} |
||||
] |
||||
}; |
||||
|
||||
t.ok(has_results_at_layers(['layer 2', 'layer 4'])(req, res)); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
test('should return true when any result.layer matches layer string', (t) => { |
||||
const req = {}; |
||||
const res = { |
||||
data: [ |
||||
{ |
||||
layer: 'layer 1' |
||||
}, |
||||
{ |
||||
layer: 'layer 2' |
||||
}, |
||||
{ |
||||
layer: 'layer 3' |
||||
} |
||||
] |
||||
}; |
||||
|
||||
t.ok(has_results_at_layers('layer 2')(req, res)); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
}; |
||||
|
||||
module.exports.tests.false_conditions = (test, common) => { |
||||
test('should return false when response has undefined data', (t) => { |
||||
const req = {}; |
||||
const res = {}; |
||||
|
||||
t.notOk(has_results_at_layers('layer')(req, res)); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
test('should return false when response has empty data array', (t) => { |
||||
const req = {}; |
||||
const res = { |
||||
data: [] |
||||
}; |
||||
|
||||
t.notOk(has_results_at_layers('layer')(req, res)); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
test('should return false when layer is a substring of non-array string layers parameter', (t) => { |
||||
const req = {}; |
||||
const res = { |
||||
data: [ |
||||
{ |
||||
layer: 'aye' |
||||
} |
||||
] |
||||
}; |
||||
|
||||
t.notOk(has_results_at_layers('layer')(req, res)); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
test('should return false when no results have layer in supplied layers', (t) => { |
||||
const req = {}; |
||||
const res = { |
||||
data: [ |
||||
{ |
||||
layer: 'layer 1' |
||||
} |
||||
] |
||||
}; |
||||
|
||||
t.notOk(has_results_at_layers(['layer 2', 'layer 3'])(req, res)); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
}; |
||||
|
||||
module.exports.all = (tape, common) => { |
||||
function test(name, testFunction) { |
||||
return tape(`GET /has_results_at_layers ${name}`, testFunction); |
||||
} |
||||
|
||||
for( const testCase in module.exports.tests ){ |
||||
module.exports.tests[testCase](test, common); |
||||
} |
||||
}; |
@ -0,0 +1,77 @@
|
||||
'use strict'; |
||||
|
||||
const _ = require('lodash'); |
||||
const is_admin_only_analysis = require('../../../../controller/predicates/is_admin_only_analysis'); |
||||
|
||||
module.exports.tests = {}; |
||||
|
||||
module.exports.tests.interface = (test, common) => { |
||||
test('valid interface', (t) => { |
||||
t.equal(typeof is_admin_only_analysis, 'function', 'is_admin_only_analysis is a function'); |
||||
t.end(); |
||||
}); |
||||
}; |
||||
|
||||
module.exports.tests.true_conditions = (test, common) => { |
||||
test('parsed_text with admin-only properties should return true', (t) => { |
||||
['neighbourhood', 'borough', 'city', 'county', 'state', 'postalcode', 'country'].forEach((property) => { |
||||
const req = { |
||||
clean: { |
||||
parsed_text: {} |
||||
} |
||||
}; |
||||
const res = {}; |
||||
|
||||
req.clean.parsed_text[property] = `${property} value`; |
||||
|
||||
t.ok(is_admin_only_analysis(req, res)); |
||||
|
||||
}); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
}; |
||||
|
||||
module.exports.tests.false_conditions = (test, common) => { |
||||
test('req.clean with no parsed_text should return false', (t) => { |
||||
const req = { |
||||
clean: { |
||||
} |
||||
}; |
||||
const res = {}; |
||||
|
||||
t.notOk(is_admin_only_analysis(req, res)); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
test('parsed_text with non-admin properties should return false', (t) => { |
||||
['number', 'street', 'query', 'category'].forEach((property) => { |
||||
const req = { |
||||
clean: { |
||||
parsed_text: {} |
||||
} |
||||
}; |
||||
const res = {}; |
||||
|
||||
req.clean.parsed_text[property] = `${property} value`; |
||||
|
||||
t.notOk(is_admin_only_analysis(req, res)); |
||||
|
||||
}); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
}; |
||||
|
||||
module.exports.all = (tape, common) => { |
||||
function test(name, testFunction) { |
||||
return tape(`GET /is_admin_only_analysis ${name}`, testFunction); |
||||
} |
||||
|
||||
for( const testCase in module.exports.tests ){ |
||||
module.exports.tests[testCase](test, common); |
||||
} |
||||
}; |
@ -0,0 +1,156 @@
|
||||
const _ = require('lodash'); |
||||
const proxyquire = require('proxyquire').noCallThru(); |
||||
const mock_logger = require('pelias-mock-logger'); |
||||
|
||||
const sortResponseData = require('../../../middleware/sortResponseData'); |
||||
|
||||
module.exports.tests = {}; |
||||
|
||||
module.exports.tests.should_execute_failure = (test, common) => { |
||||
test('should_execute returning false should call next w/o invoking comparator', (t) => { |
||||
t.plan(2, 'this ensures that should_execute was invoked'); |
||||
|
||||
const comparator = () => { |
||||
throw Error('should not have been called'); |
||||
}; |
||||
|
||||
const should_execute = (req, res) => { |
||||
t.deepEquals(req, { a: 1 }); |
||||
t.deepEquals(res, { b: 2 }); |
||||
return false; |
||||
}; |
||||
|
||||
const sort = sortResponseData(comparator, should_execute); |
||||
|
||||
const req = { a: 1 }; |
||||
const res = { b: 2 }; |
||||
|
||||
sort(req, res, () => { |
||||
t.end(); |
||||
}); |
||||
|
||||
}); |
||||
|
||||
}; |
||||
|
||||
module.exports.tests.general_tests = (test, common) => { |
||||
test('req.clean should be passed to sort', (t) => { |
||||
t.plan(1, 'this ensures that comparator was invoked'); |
||||
|
||||
const comparator = (clean) => { |
||||
t.deepEquals(clean, { a: 1 }); |
||||
return () => { |
||||
throw Error('should not have been called'); |
||||
}; |
||||
}; |
||||
|
||||
const sort = sortResponseData(comparator, _.constant(true)); |
||||
|
||||
const req = { |
||||
clean: { |
||||
a: 1 |
||||
} |
||||
}; |
||||
|
||||
const res = { |
||||
data: [ {} ] |
||||
}; |
||||
|
||||
sort(req, res, () => { |
||||
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, _.constant(true)); |
||||
|
||||
const req = {}; |
||||
const res = {}; |
||||
|
||||
sort(req, res, () => { |
||||
t.deepEquals(res, {}); |
||||
t.end(); |
||||
}); |
||||
|
||||
}); |
||||
|
||||
test('empty res.data should return without interacting with comparator', (t) => { |
||||
const comparator = () => { |
||||
throw Error('should not have been called'); |
||||
}; |
||||
|
||||
const sort = sortResponseData(comparator, _.constant(true)); |
||||
|
||||
const req = {}; |
||||
const res = { |
||||
data: [] |
||||
}; |
||||
|
||||
sort(req, res, () => { |
||||
t.deepEquals(res.data, [], 'res.data should still be empty'); |
||||
t.end(); |
||||
}); |
||||
|
||||
}); |
||||
|
||||
}; |
||||
|
||||
module.exports.tests.successful_sort = (test, common) => { |
||||
test('comparator should be sort res.data', (t) => { |
||||
const logger = mock_logger(); |
||||
|
||||
const comparator = () => { |
||||
return (a, b) => { |
||||
return a._id > b._id; |
||||
}; |
||||
}; |
||||
|
||||
const sortResponseData = proxyquire('../../../middleware/sortResponseData', { |
||||
'pelias-logger': logger |
||||
}); |
||||
|
||||
const sort = sortResponseData(comparator, _.constant(true)); |
||||
|
||||
const req = { |
||||
clean: { |
||||
field: 'value' |
||||
} |
||||
}; |
||||
const res = { |
||||
data: [ |
||||
{ _id: 3 }, |
||||
{ _id: 2 }, |
||||
{ _id: 1 }, |
||||
] |
||||
}; |
||||
|
||||
sort(req, res, () => { |
||||
t.deepEquals(res.data.shift(), { _id: 1 }); |
||||
t.deepEquals(res.data.shift(), { _id: 2 }); |
||||
t.deepEquals(res.data.shift(), { _id: 3 }); |
||||
|
||||
t.ok(logger.isDebugMessage( |
||||
'req.clean: {"field":"value"}, pre-sort: [3,2,1], post-sort: [1,2,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); |
||||
} |
||||
}; |
@ -0,0 +1,111 @@
|
||||
const _ = require('lodash'); |
||||
|
||||
const geonames_warnings = require('../../../sanitizer/_geonames_warnings'); |
||||
|
||||
const nonAdminProperties = ['number', 'street', 'query', 'category']; |
||||
const adminProperties = ['neighbourhood', 'borough', 'city', 'county', 'state', 'postalcode', 'country']; |
||||
|
||||
module.exports.tests = {}; |
||||
|
||||
module.exports.tests.no_errors = (test, common) => { |
||||
test('undefined clean.parsed_text should exit early', (t) => { |
||||
const clean = { |
||||
sources: ['geonames'], |
||||
}; |
||||
|
||||
const messages = geonames_warnings(undefined, clean); |
||||
|
||||
t.deepEquals(messages, { errors: [], warnings: [] }); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
test('any non-admin analysis field with only geonames sources should exit early', (t) => { |
||||
adminProperties.forEach((adminProperty) => { |
||||
nonAdminProperties.forEach((nonAdminProperty) => { |
||||
const clean = { |
||||
sources: ['geonames'], |
||||
parsed_text: {} |
||||
}; |
||||
clean.parsed_text[nonAdminProperty] = `${nonAdminProperty} value`; |
||||
clean.parsed_text[adminProperty] = `${adminProperty} value`; |
||||
|
||||
const messages = geonames_warnings(undefined, clean); |
||||
|
||||
t.deepEquals(messages, { errors: [], warnings: [] }); |
||||
|
||||
}); |
||||
}); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
test('any non-admin analysis field with non-geonames sources should exit early', (t) => { |
||||
adminProperties.forEach((adminProperty) => { |
||||
nonAdminProperties.forEach((nonAdminProperty) => { |
||||
const clean = { |
||||
sources: ['this is not geonames'], |
||||
parsed_text: {} |
||||
}; |
||||
clean.parsed_text[nonAdminProperty] = `${nonAdminProperty} value`; |
||||
clean.parsed_text[adminProperty] = `${adminProperty} value`; |
||||
|
||||
const messages = geonames_warnings(undefined, clean); |
||||
|
||||
t.deepEquals(messages, { errors: [], warnings: [] }); |
||||
|
||||
}); |
||||
}); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
}; |
||||
|
||||
module.exports.tests.error_conditions = (test, common) => { |
||||
test('any admin analysis field and only geonames sources should return error', (t) => { |
||||
adminProperties.forEach((property) => { |
||||
const clean = _.set({ sources: ['geonames'] }, |
||||
['parsed_text', property], `${property} value`); |
||||
|
||||
const messages = geonames_warnings(undefined, clean); |
||||
|
||||
t.deepEquals(messages.errors, ['input contains only administrative area data, ' + |
||||
'no results will be returned when sources=geonames']); |
||||
t.deepEquals(messages.warnings, []); |
||||
|
||||
}); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
}; |
||||
|
||||
module.exports.tests.warning_conditions = (test, common) => { |
||||
test('any admin analysis field and only geonames sources should return warning', (t) => { |
||||
adminProperties.forEach((property) => { |
||||
const clean = _.set({ sources: ['source 1', 'geonames', 'source 2'] }, |
||||
['parsed_text', property], `${property} value`); |
||||
|
||||
const messages = geonames_warnings(undefined, clean); |
||||
|
||||
t.deepEquals(messages.errors, []); |
||||
t.deepEquals(messages.warnings, ['input contains only administrative area data, ' + |
||||
'geonames results will not be returned']); |
||||
|
||||
}); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
}; |
||||
|
||||
module.exports.all = (tape, common) => { |
||||
function test(name, testFunction) { |
||||
return tape(`SANTIZE _geonames_warnings ${name}`, testFunction); |
||||
} |
||||
|
||||
for( var testCase in module.exports.tests ){ |
||||
module.exports.tests[testCase](test, common); |
||||
} |
||||
}; |
@ -0,0 +1,139 @@
|
||||
module.exports.tests = {}; |
||||
|
||||
const PlaceHolder = require('../../../../service/configurations/PlaceHolder'); |
||||
|
||||
module.exports.tests.all = (test, common) => { |
||||
test('getName should return \'placeholder\'', (t) => { |
||||
const configBlob = { |
||||
url: 'http://localhost:1234', |
||||
timeout: 17, |
||||
retries: 19 |
||||
}; |
||||
|
||||
const placeholder = new PlaceHolder(configBlob); |
||||
|
||||
t.equals(placeholder.getName(), 'placeholder'); |
||||
t.equals(placeholder.getBaseUrl(), 'http://localhost:1234/'); |
||||
t.equals(placeholder.getTimeout(), 17); |
||||
t.equals(placeholder.getRetries(), 19); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
test('getUrl should return value passed to constructor', (t) => { |
||||
const configBlob = { |
||||
url: 'http://localhost:1234', |
||||
timeout: 17, |
||||
retries: 19 |
||||
}; |
||||
|
||||
const placeholder = new PlaceHolder(configBlob); |
||||
|
||||
t.equals(placeholder.getUrl(), 'http://localhost:1234/search'); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
test('getParameters should return object with text and lang from req', (t) => { |
||||
const configBlob = { |
||||
url: 'http://localhost:1234', |
||||
timeout: 17, |
||||
retries: 19 |
||||
}; |
||||
|
||||
const placeholder = new PlaceHolder(configBlob); |
||||
|
||||
const req = { |
||||
clean: { |
||||
text: 'text value', |
||||
lang: { |
||||
iso6393: 'lang value' |
||||
} |
||||
} |
||||
}; |
||||
|
||||
t.deepEquals(placeholder.getParameters(req), { text: 'text value', lang: 'lang value' }); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
test('getHeaders should return empty object', (t) => { |
||||
const configBlob = { |
||||
url: 'base url', |
||||
timeout: 17, |
||||
retries: 19 |
||||
}; |
||||
|
||||
const placeholder = new PlaceHolder(configBlob); |
||||
|
||||
t.deepEquals(placeholder.getHeaders(), {}); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
test('getParameters should not include lang if req.clean.lang is unavailable', (t) => { |
||||
const configBlob = { |
||||
url: 'base url', |
||||
timeout: 17, |
||||
retries: 19 |
||||
}; |
||||
|
||||
const placeholder = new PlaceHolder(configBlob); |
||||
|
||||
const req = { |
||||
clean: { |
||||
text: 'text value' |
||||
} |
||||
}; |
||||
|
||||
t.deepEquals(placeholder.getParameters(req), { text: 'text value' }); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
test('getParameters should not include lang if req.clean.lang.iso6393 is unavailable', (t) => { |
||||
const configBlob = { |
||||
url: 'base url', |
||||
timeout: 17, |
||||
retries: 19 |
||||
}; |
||||
|
||||
const placeholder = new PlaceHolder(configBlob); |
||||
|
||||
const req = { |
||||
clean: { |
||||
text: 'text value', |
||||
lang: {} |
||||
} |
||||
}; |
||||
|
||||
t.deepEquals(placeholder.getParameters(req), { text: 'text value' }); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
test('baseUrl ending in / should not have double /\'s return by getUrl', (t) => { |
||||
const configBlob = { |
||||
url: 'http://localhost:1234/blah', |
||||
timeout: 17, |
||||
retries: 19 |
||||
}; |
||||
|
||||
const placeholder = new PlaceHolder(configBlob); |
||||
|
||||
t.deepEquals(placeholder.getUrl(), 'http://localhost:1234/blah/search'); |
||||
t.end(); |
||||
|
||||
}); |
||||
|
||||
}; |
||||
|
||||
module.exports.all = (tape, common) => { |
||||
function test(name, testFunction) { |
||||
return tape(`SERVICE CONFIGURATION /PlaceHolder ${name}`, testFunction); |
||||
} |
||||
|
||||
for( var testCase in module.exports.tests ){ |
||||
module.exports.tests[testCase](test, common); |
||||
} |
||||
}; |
Loading…
Reference in new issue