1822 lines
56 KiB
JavaScript
1822 lines
56 KiB
JavaScript
'use strict';
|
|
|
|
var qs = require('querystring');
|
|
var step = require('step');
|
|
const { URL } = require('url');
|
|
var PSQL = require('cartodb-psql');
|
|
var _ = require('underscore');
|
|
const mapnik = require('@carto/mapnik');
|
|
|
|
var LayergroupToken = require('../../lib/models/layergroup-token');
|
|
|
|
var assert = require('./assert');
|
|
var helper = require('./test-helper');
|
|
|
|
var CartodbWindshaft = require('../../lib/server');
|
|
var serverOptions = require('../../lib/server-options');
|
|
serverOptions.analysis.batch.inlineExecution = true;
|
|
|
|
const MAPNIK_SUPPORTED_FORMATS = {
|
|
png: true,
|
|
png32: true,
|
|
'grid.json': true,
|
|
mvt: true
|
|
};
|
|
|
|
function TestClient (config, apiKey, extraHeaders = {}, overrideServerOptions = {}) {
|
|
this.mapConfig = isMapConfig(config) ? config : null;
|
|
this.template = isTemplate(config) ? config : null;
|
|
this.apiKey = apiKey;
|
|
this.extraHeaders = extraHeaders;
|
|
this.keysToDelete = {};
|
|
this.serverOptions = Object.assign({}, serverOptions, overrideServerOptions);
|
|
this.server = new CartodbWindshaft(this.serverOptions);
|
|
}
|
|
|
|
module.exports = TestClient;
|
|
|
|
function isMapConfig (config) {
|
|
return config && config.layers;
|
|
}
|
|
|
|
function isTemplate (config) {
|
|
return config && config.layergroup;
|
|
}
|
|
|
|
module.exports.RESPONSE = {
|
|
ERROR: {
|
|
status: 400,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
}
|
|
};
|
|
|
|
module.exports.CARTOCSS = {
|
|
POINTS: [
|
|
'#layer{',
|
|
' marker-placement: point;',
|
|
' marker-allow-overlap: true;',
|
|
' marker-line-opacity: 0.2;',
|
|
' marker-line-width: 0.5;',
|
|
' marker-opacity: 1;',
|
|
' marker-width: 5;',
|
|
' marker-fill: red;',
|
|
'}'
|
|
].join('\n'),
|
|
|
|
LINES: [
|
|
'#lines {',
|
|
' line-color: black;',
|
|
' line-width: 1;',
|
|
' line-opacity: 1;',
|
|
'}'
|
|
].join('\n'),
|
|
|
|
POLYGONS: [
|
|
'#layer {',
|
|
' polygon-fill: red;',
|
|
' polygon-opacity: 0.6;',
|
|
' polygon-opacity: 0.7;',
|
|
' line-color: #FFF;',
|
|
' line-width: 0.5;',
|
|
' line-opacity: 1;',
|
|
'}'
|
|
].join('\n'),
|
|
|
|
TORQUE: [
|
|
'Map {',
|
|
' -torque-frame-count: 256;',
|
|
' -torque-animation-duration: 30;',
|
|
' -torque-time-attribute: "cartodb_id";',
|
|
' -torque-aggregation-function: "count(1)";',
|
|
' -torque-resolution: 4;',
|
|
' -torque-data-aggregation: linear;',
|
|
'}',
|
|
'#layer {',
|
|
' marker-width: 7;',
|
|
' marker-fill: #FFB927;',
|
|
' marker-fill-opacity: 0.9;',
|
|
' marker-line-width: 1;',
|
|
' marker-line-color: #FFF;',
|
|
' marker-line-opacity: 1;',
|
|
' comp-op: lighter;',
|
|
'}',
|
|
'#layer[frame-offset=1] {',
|
|
' marker-width: 9;',
|
|
' marker-fill-opacity: 0.45;',
|
|
'}',
|
|
'#layer[frame-offset=2] {',
|
|
' marker-width: 11;',
|
|
' marker-fill-opacity: 0.225;',
|
|
'}'
|
|
].join('\n')
|
|
};
|
|
|
|
module.exports.SQL = {
|
|
EMPTY: 'select 1 as cartodb_id, null::geometry as the_geom_webmercator',
|
|
ONE_POINT: 'select 1 as cartodb_id, \'SRID=3857;POINT(0 0)\'::geometry the_geom_webmercator'
|
|
};
|
|
|
|
function resErr2errRes (callback) {
|
|
return (res, err) => {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
return callback(err, res);
|
|
};
|
|
}
|
|
|
|
function layergroupidTemplate (layergroupId, params) {
|
|
const { token, signer, cacheBuster } = LayergroupToken.parse(layergroupId);
|
|
|
|
// {user}@{token}:{cache_buster}
|
|
// {token}:{cache_buster}
|
|
return `${signer ? signer + '@' : ''}${token}:${params.cacheBuster ? Date.now() : cacheBuster}`;
|
|
}
|
|
|
|
TestClient.prototype.getWidget = function (widgetName, params, callback) {
|
|
var self = this;
|
|
|
|
if (!callback) {
|
|
callback = params;
|
|
params = {};
|
|
}
|
|
|
|
var url = '/api/v1/map';
|
|
if (params && params.filters) {
|
|
url += '?' + qs.stringify({ filters: JSON.stringify(params.filters) });
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, self.extraHeaders);
|
|
|
|
step(
|
|
function createLayergroup () {
|
|
var next = this;
|
|
assert.response(self.server,
|
|
{
|
|
url: url,
|
|
method: 'POST',
|
|
headers,
|
|
data: JSON.stringify(self.mapConfig)
|
|
},
|
|
{
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
},
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
var parsedBody = JSON.parse(res.body);
|
|
|
|
var expectedWidgetURLS = {
|
|
http: '/api/v1/map/' + parsedBody.layergroupid + '/0/widget/' + widgetName
|
|
};
|
|
assert.ok(parsedBody.metadata.layers[0].widgets[widgetName]);
|
|
assert.ok(
|
|
parsedBody.metadata.layers[0].widgets[widgetName].url.http.match(expectedWidgetURLS.http)
|
|
);
|
|
|
|
var expectedDataviewsURLS = {
|
|
http: '/api/v1/map/' + parsedBody.layergroupid + '/dataview/' + widgetName
|
|
};
|
|
assert.ok(parsedBody.metadata.dataviews[widgetName]);
|
|
assert.ok(
|
|
parsedBody.metadata.dataviews[widgetName].url.http.match(expectedDataviewsURLS.http)
|
|
);
|
|
|
|
return next(null, parsedBody.layergroupid);
|
|
}
|
|
);
|
|
},
|
|
function getWidgetResult (err, layergroupId) {
|
|
assert.ifError(err);
|
|
|
|
var next = this;
|
|
self.keysToDelete['map_cfg|' + LayergroupToken.parse(layergroupId).token] = 0;
|
|
self.keysToDelete['user:localhost:mapviews:global'] = 5;
|
|
|
|
var urlParams = {
|
|
own_filter: Object.prototype.hasOwnProperty.call(params, 'own_filter') ? params.own_filter : 1
|
|
};
|
|
['bbox', 'bins', 'start', 'end'].forEach(function (extraParam) {
|
|
if (Object.prototype.hasOwnProperty.call(params, extraParam)) {
|
|
urlParams[extraParam] = params[extraParam];
|
|
}
|
|
});
|
|
|
|
url = '/api/v1/map/' + layergroupId + '/0/widget/' + widgetName + '?' + qs.stringify(urlParams);
|
|
const headers = Object.assign({ host: 'localhost' }, self.extraHeaders);
|
|
|
|
assert.response(self.server,
|
|
{
|
|
url: url,
|
|
method: 'GET',
|
|
headers
|
|
},
|
|
{
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
},
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
|
|
next(null, res);
|
|
}
|
|
);
|
|
},
|
|
function finish (err, res) {
|
|
var widget;
|
|
if (!err && res.body) {
|
|
widget = JSON.parse(res.body);
|
|
}
|
|
return callback(err, res, widget);
|
|
}
|
|
);
|
|
};
|
|
|
|
TestClient.prototype.widgetSearch = function (widgetName, userQuery, params, callback) {
|
|
var self = this;
|
|
|
|
if (!callback) {
|
|
callback = params;
|
|
params = {};
|
|
}
|
|
|
|
var url = '/api/v1/map';
|
|
if (params && params.filters) {
|
|
url += '?' + qs.stringify({ filters: JSON.stringify(params.filters) });
|
|
}
|
|
|
|
step(
|
|
function createLayergroup () {
|
|
var next = this;
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, self.extraHeaders);
|
|
|
|
assert.response(self.server,
|
|
{
|
|
url: url,
|
|
method: 'POST',
|
|
headers,
|
|
data: JSON.stringify(self.mapConfig)
|
|
},
|
|
{
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
},
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
var parsedBody = JSON.parse(res.body);
|
|
|
|
var expectedWidgetURLS = {
|
|
http: '/api/v1/map/' + parsedBody.layergroupid + '/0/widget/' + widgetName
|
|
};
|
|
assert.ok(parsedBody.metadata.layers[0].widgets[widgetName]);
|
|
assert.ok(
|
|
parsedBody.metadata.layers[0].widgets[widgetName].url.http.match(expectedWidgetURLS.http)
|
|
);
|
|
|
|
var expectedDataviewsURLS = {
|
|
http: '/api/v1/map/' + parsedBody.layergroupid + '/dataview/' + widgetName
|
|
};
|
|
assert.ok(parsedBody.metadata.dataviews[widgetName]);
|
|
assert.ok(
|
|
parsedBody.metadata.dataviews[widgetName].url.http.match(expectedDataviewsURLS.http)
|
|
);
|
|
|
|
return next(null, parsedBody.layergroupid);
|
|
}
|
|
);
|
|
},
|
|
function getWidgetSearchResult (err, layergroupId) {
|
|
assert.ifError(err);
|
|
|
|
var next = this;
|
|
self.keysToDelete['map_cfg|' + LayergroupToken.parse(layergroupId).token] = 0;
|
|
self.keysToDelete['user:localhost:mapviews:global'] = 5;
|
|
|
|
var urlParams = {
|
|
q: userQuery,
|
|
own_filter: Object.prototype.hasOwnProperty.call(params, 'own_filter') ? params.own_filter : 1
|
|
};
|
|
if (params && params.bbox) {
|
|
urlParams.bbox = params.bbox;
|
|
}
|
|
url = '/api/v1/map/' + layergroupId + '/0/widget/' + widgetName + '/search?' + qs.stringify(urlParams);
|
|
const headers = Object.assign({ host: 'localhost' }, self.extraHeaders);
|
|
|
|
assert.response(self.server,
|
|
{
|
|
url: url,
|
|
method: 'GET',
|
|
headers
|
|
},
|
|
{
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
},
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
|
|
next(null, res);
|
|
}
|
|
);
|
|
},
|
|
function finish (err, res) {
|
|
var searchResult;
|
|
if (!err && res.body) {
|
|
searchResult = JSON.parse(res.body);
|
|
}
|
|
return callback(err, res, searchResult);
|
|
}
|
|
);
|
|
};
|
|
|
|
TestClient.prototype.getDataview = function (dataviewName, params, callback) {
|
|
var self = this;
|
|
|
|
if (!callback) {
|
|
callback = params;
|
|
params = {};
|
|
}
|
|
|
|
var extraParams = {};
|
|
if (this.apiKey) {
|
|
extraParams.api_key = this.apiKey;
|
|
}
|
|
if (params && params.filters) {
|
|
extraParams.filters = JSON.stringify(params.filters);
|
|
}
|
|
|
|
var url = '/api/v1/map';
|
|
var urlNamed = url + '/named';
|
|
if (Object.keys(extraParams).length > 0) {
|
|
url += '?' + qs.stringify(extraParams);
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, self.extraHeaders);
|
|
|
|
var expectedResponse = params.response || {
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
};
|
|
|
|
step(
|
|
function createTemplate () {
|
|
var next = this;
|
|
|
|
if (!self.template) {
|
|
return next();
|
|
}
|
|
|
|
if (!self.apiKey) {
|
|
return next(new Error('apiKey param is mandatory to create a new template'));
|
|
}
|
|
|
|
params.placeholders = params.placeholders || {};
|
|
|
|
assert.response(self.server,
|
|
{
|
|
url: urlNamed + '?' + qs.stringify({ api_key: self.apiKey }),
|
|
method: 'POST',
|
|
headers,
|
|
data: JSON.stringify(self.template)
|
|
},
|
|
{
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
},
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
return next(null, JSON.parse(res.body).template_id);
|
|
}
|
|
);
|
|
},
|
|
function createLayergroup (err, templateId) {
|
|
assert.ifError(err);
|
|
|
|
var next = this;
|
|
|
|
var data = templateId ? params.placeholders : self.mapConfig;
|
|
|
|
const queryParams = {};
|
|
|
|
if (self.apiKey) {
|
|
queryParams.api_key = self.apiKey;
|
|
}
|
|
|
|
if (params.filters !== undefined) {
|
|
queryParams.filters = JSON.stringify(params.filters);
|
|
}
|
|
|
|
var path = templateId
|
|
? urlNamed + '/' + templateId + '?' + qs.stringify(queryParams)
|
|
: url;
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, self.extraHeaders);
|
|
|
|
assert.response(self.server,
|
|
{
|
|
url: path,
|
|
method: 'POST',
|
|
headers,
|
|
data: JSON.stringify(data)
|
|
},
|
|
{
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
},
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
var parsedBody = JSON.parse(res.body);
|
|
var expectedDataviewsURLS = {
|
|
http: '/api/v1/map/' + parsedBody.layergroupid + '/dataview/' + dataviewName
|
|
};
|
|
assert.ok(parsedBody.metadata.dataviews[dataviewName]);
|
|
assert.ok(
|
|
parsedBody.metadata.dataviews[dataviewName].url.http.match(expectedDataviewsURLS.http)
|
|
);
|
|
return next(null, parsedBody.layergroupid);
|
|
}
|
|
);
|
|
},
|
|
function getDataviewResult (err, layergroupId) {
|
|
assert.ifError(err);
|
|
|
|
var next = this;
|
|
self.keysToDelete['map_cfg|' + LayergroupToken.parse(layergroupId).token] = 0;
|
|
self.keysToDelete['user:localhost:mapviews:global'] = 5;
|
|
|
|
var urlParams = {};
|
|
if (Object.prototype.hasOwnProperty.call(params, 'no_filters')) {
|
|
urlParams.no_filters = params.no_filters;
|
|
}
|
|
if (Object.prototype.hasOwnProperty.call(params, 'own_filter')) {
|
|
urlParams.own_filter = params.own_filter;
|
|
}
|
|
|
|
['bbox', 'circle', 'polygon', 'bins', 'start', 'end', 'aggregation', 'offset', 'categories'].forEach(function (extraParam) {
|
|
if (Object.prototype.hasOwnProperty.call(params, extraParam)) {
|
|
urlParams[extraParam] = params[extraParam];
|
|
}
|
|
});
|
|
|
|
if (self.apiKey) {
|
|
urlParams.api_key = self.apiKey;
|
|
}
|
|
url = '/api/v1/map/' + layergroupId + '/dataview/' + dataviewName + '?' + qs.stringify(urlParams);
|
|
const headers = Object.assign({ host: 'localhost' }, self.extraHeaders);
|
|
|
|
assert.response(self.server,
|
|
{
|
|
url: url,
|
|
method: 'GET',
|
|
headers
|
|
},
|
|
expectedResponse,
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
next(null, JSON.parse(res.body), res.headers);
|
|
}
|
|
);
|
|
},
|
|
function finish (err, dataview, headers = null) {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
return callback(null, dataview, headers);
|
|
}
|
|
);
|
|
};
|
|
|
|
TestClient.prototype.getFeatureAttributes = function (featureId, layerId, params, callback) {
|
|
var self = this;
|
|
|
|
if (!callback) {
|
|
callback = params;
|
|
params = {};
|
|
}
|
|
|
|
var extraParams = {};
|
|
if (this.apiKey) {
|
|
extraParams.api_key = this.apiKey;
|
|
}
|
|
if (params && params.filters) {
|
|
extraParams.filters = JSON.stringify(params.filters);
|
|
}
|
|
|
|
var url = '/api/v1/map';
|
|
if (Object.keys(extraParams).length > 0) {
|
|
url += '?' + qs.stringify(extraParams);
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, self.extraHeaders);
|
|
|
|
var expectedResponse = params.response || {
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
};
|
|
|
|
step(
|
|
function createLayergroup () {
|
|
var next = this;
|
|
assert.response(self.server,
|
|
{
|
|
url: url,
|
|
method: 'POST',
|
|
headers,
|
|
data: JSON.stringify(self.mapConfig)
|
|
},
|
|
{
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
},
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
|
|
var parsedBody = JSON.parse(res.body);
|
|
|
|
if (parsedBody.layergroupid) {
|
|
self.keysToDelete['map_cfg|' + LayergroupToken.parse(parsedBody.layergroupid).token] = 0;
|
|
self.keysToDelete['user:localhost:mapviews:global'] = 5;
|
|
}
|
|
|
|
return next(null, parsedBody.layergroupid);
|
|
}
|
|
);
|
|
},
|
|
function getFeatureAttributes (err, layergroupId) {
|
|
assert.ifError(err);
|
|
|
|
var next = this;
|
|
|
|
url = '/api/v1/map/' + layergroupId + '/' + layerId + '/attributes/' + featureId;
|
|
const headers = Object.assign({ host: 'localhost' }, self.extraHeaders);
|
|
|
|
assert.response(self.server,
|
|
{
|
|
url: url,
|
|
method: 'GET',
|
|
headers
|
|
},
|
|
expectedResponse,
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
|
|
next(null, JSON.parse(res.body));
|
|
}
|
|
);
|
|
},
|
|
function finish (err, attributes) {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
|
|
return callback(null, attributes);
|
|
}
|
|
);
|
|
};
|
|
|
|
TestClient.prototype.getClusterFeatures = function (zoom, clusterId, layerId, params, callback) {
|
|
var self = this;
|
|
|
|
if (!callback) {
|
|
callback = params;
|
|
params = {};
|
|
}
|
|
|
|
var extraParams = {};
|
|
|
|
if (this.apiKey) {
|
|
extraParams.api_key = this.apiKey;
|
|
}
|
|
|
|
// if (params && params.filters) {
|
|
// extraParams.filters = JSON.stringify(params.filters);
|
|
// }
|
|
|
|
var url = '/api/v1/map';
|
|
if (Object.keys(extraParams).length > 0) {
|
|
url += '?' + qs.stringify(extraParams);
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, self.extraHeaders);
|
|
|
|
var expectedResponse = params.response || {
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
};
|
|
|
|
step(
|
|
function createLayergroup () {
|
|
var next = this;
|
|
assert.response(self.server,
|
|
{
|
|
url: url,
|
|
method: 'POST',
|
|
headers,
|
|
data: JSON.stringify(self.mapConfig)
|
|
},
|
|
{
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
},
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
|
|
var parsedBody = JSON.parse(res.body);
|
|
|
|
if (parsedBody.layergroupid) {
|
|
self.keysToDelete['map_cfg|' + LayergroupToken.parse(parsedBody.layergroupid).token] = 0;
|
|
self.keysToDelete['user:localhost:mapviews:global'] = 5;
|
|
}
|
|
|
|
return next(null, parsedBody.layergroupid);
|
|
}
|
|
);
|
|
},
|
|
function getCLusterFeatures (err, layergroupId) {
|
|
assert.ifError(err);
|
|
|
|
var next = this;
|
|
|
|
let queryParams = '';
|
|
if (params.aggregation) {
|
|
queryParams = qs.stringify({ aggregation: JSON.stringify(params.aggregation) });
|
|
}
|
|
|
|
url = `/api/v1/map/${layergroupId}/${layerId}/${zoom}/cluster/${clusterId}?${queryParams}`;
|
|
const headers = Object.assign({ host: 'localhost' }, self.extraHeaders);
|
|
|
|
assert.response(self.server,
|
|
{
|
|
url: url,
|
|
method: 'GET',
|
|
headers
|
|
},
|
|
expectedResponse,
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
|
|
next(null, JSON.parse(res.body));
|
|
}
|
|
);
|
|
},
|
|
function finish (err, attributes) {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
|
|
return callback(null, attributes);
|
|
}
|
|
);
|
|
};
|
|
|
|
TestClient.prototype.getTile = function (z, x, y, params, callback) {
|
|
var self = this;
|
|
|
|
if (!callback) {
|
|
callback = params;
|
|
params = {};
|
|
}
|
|
|
|
var url = '/api/v1/map';
|
|
var urlNamed = url + '/named';
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, self.extraHeaders);
|
|
|
|
if (this.apiKey) {
|
|
url += '?' + qs.stringify({ api_key: this.apiKey });
|
|
}
|
|
|
|
var layergroupId;
|
|
|
|
if (params.layergroupid) {
|
|
layergroupId = params.layergroupid;
|
|
}
|
|
|
|
step(
|
|
function createTemplate () {
|
|
var next = this;
|
|
|
|
if (!self.template) {
|
|
return next();
|
|
}
|
|
|
|
if (!self.apiKey) {
|
|
return next(new Error('apiKey param is mandatory to create a new template'));
|
|
}
|
|
|
|
params.placeholders = params.placeholders || {};
|
|
|
|
assert.response(self.server,
|
|
{
|
|
url: urlNamed + '?' + qs.stringify({ api_key: self.apiKey }),
|
|
method: 'POST',
|
|
headers,
|
|
data: JSON.stringify(self.template)
|
|
},
|
|
{
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
},
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
return next(null, JSON.parse(res.body).template_id);
|
|
}
|
|
);
|
|
},
|
|
function createLayergroup (err, templateId) {
|
|
var next = this;
|
|
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
|
|
if (layergroupId) {
|
|
return next(null, layergroupId);
|
|
}
|
|
|
|
var data = templateId ? params.placeholders : self.mapConfig;
|
|
|
|
const queryParams = {};
|
|
|
|
if (self.apiKey) {
|
|
queryParams.api_key = self.apiKey;
|
|
}
|
|
|
|
if (params.aggregation !== undefined) {
|
|
queryParams.aggregation = params.aggregation;
|
|
}
|
|
|
|
var path = templateId
|
|
? urlNamed + '/' + templateId + '?' + qs.stringify(queryParams)
|
|
: url;
|
|
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, self.extraHeaders);
|
|
|
|
assert.response(self.server,
|
|
{
|
|
url: path,
|
|
method: 'POST',
|
|
headers,
|
|
data: JSON.stringify(data)
|
|
},
|
|
{
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
},
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
return next(null, JSON.parse(res.body).layergroupid);
|
|
}
|
|
);
|
|
},
|
|
function getTileResult (err, layergroupId) {
|
|
assert.ifError(err);
|
|
|
|
self.keysToDelete['map_cfg|' + LayergroupToken.parse(layergroupId).token] = 0;
|
|
self.keysToDelete['user:localhost:mapviews:global'] = 5;
|
|
|
|
url = `/api/v1/map/${layergroupidTemplate(layergroupId, params)}/`;
|
|
var layers = params.layers;
|
|
|
|
if (layers !== undefined) {
|
|
layers = Array.isArray(layers) ? layers : [layers];
|
|
url += layers.join(',') + '/';
|
|
}
|
|
var format = params.format || 'png';
|
|
|
|
if (layers === undefined && !MAPNIK_SUPPORTED_FORMATS[format]) {
|
|
throw new Error(`Missing layer filter while fetching ${format} tile, review params argument`);
|
|
}
|
|
|
|
url += [z, x, y].join('/');
|
|
url += '.' + format;
|
|
|
|
const headers = Object.assign({ host: 'localhost' }, self.extraHeaders);
|
|
|
|
const queryParams = {};
|
|
|
|
if (self.apiKey) {
|
|
queryParams.api_key = self.apiKey;
|
|
}
|
|
|
|
if (Object.keys(queryParams).length) {
|
|
url += '?' + qs.stringify(queryParams);
|
|
}
|
|
|
|
var request = {
|
|
url: url,
|
|
method: 'GET',
|
|
headers
|
|
};
|
|
|
|
var expectedResponse = Object.assign({}, {
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'image/png'
|
|
}
|
|
}, params.response);
|
|
|
|
var isPng = format.match(/png$/);
|
|
|
|
if (isPng) {
|
|
request.encoding = 'binary';
|
|
}
|
|
|
|
var isMvt = format.match(/mvt$/);
|
|
|
|
if (isMvt) {
|
|
request.encoding = 'binary';
|
|
if (expectedResponse.status === 200) {
|
|
expectedResponse.headers['Content-Type'] = 'application/x-protobuf';
|
|
}
|
|
}
|
|
|
|
var isGeojson = format.match(/geojson$/);
|
|
|
|
if (isGeojson) {
|
|
request.encoding = 'utf-8';
|
|
expectedResponse.headers['Content-Type'] = 'application/json; charset=utf-8';
|
|
}
|
|
|
|
var isGridJSON = format.match(/grid.json$/);
|
|
|
|
if (isGridJSON) {
|
|
request.encoding = 'utf-8';
|
|
expectedResponse.headers['Content-Type'] = 'application/json; charset=utf-8';
|
|
}
|
|
|
|
if (params.contentType) {
|
|
expectedResponse.headers['Content-Type'] = 'application/json; charset=utf-8';
|
|
}
|
|
|
|
assert.response(self.server, request, expectedResponse, resErr2errRes(this));
|
|
},
|
|
function finish (err, res) {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
|
|
var body;
|
|
switch (res.headers['content-type']) {
|
|
case 'image/png':
|
|
body = mapnik.Image.fromBytes(Buffer.from(res.body, 'binary'));
|
|
break;
|
|
case 'application/x-protobuf':
|
|
body = new mapnik.VectorTile(z, x, y);
|
|
body.setDataSync(Buffer.from(res.body, 'binary'));
|
|
break;
|
|
case 'application/json; charset=utf-8':
|
|
body = JSON.parse(res.body);
|
|
break;
|
|
default:
|
|
body = res.body;
|
|
break;
|
|
}
|
|
|
|
return callback(err, res, body);
|
|
}
|
|
);
|
|
};
|
|
|
|
TestClient.prototype.getLayergroup = function (params, callback) {
|
|
var self = this;
|
|
|
|
if (!callback) {
|
|
callback = params;
|
|
params = null;
|
|
}
|
|
|
|
if (!params) {
|
|
params = {};
|
|
}
|
|
|
|
let url = '/api/v1/map';
|
|
const urlNamed = url + '/named';
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, self.extraHeaders);
|
|
|
|
const queryParams = {};
|
|
|
|
if (self.apiKey) {
|
|
queryParams.api_key = self.apiKey;
|
|
}
|
|
|
|
if (params.aggregation !== undefined) {
|
|
queryParams.aggregation = params.aggregation;
|
|
}
|
|
|
|
if (Object.keys(queryParams).length) {
|
|
url += '?' + qs.stringify(queryParams);
|
|
}
|
|
|
|
var layergroupId;
|
|
|
|
if (params.layergroupid) {
|
|
layergroupId = params.layergroupid;
|
|
}
|
|
|
|
step(
|
|
function createTemplate () {
|
|
var next = this;
|
|
|
|
if (!self.template) {
|
|
return next();
|
|
}
|
|
|
|
if (!self.apiKey) {
|
|
return next(new Error('apiKey param is mandatory to create a new template'));
|
|
}
|
|
|
|
params.placeholders = params.placeholders || {};
|
|
|
|
assert.response(self.server,
|
|
{
|
|
url: urlNamed + '?' + qs.stringify({ api_key: self.apiKey }),
|
|
method: 'POST',
|
|
headers,
|
|
data: JSON.stringify(self.template)
|
|
},
|
|
{
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
},
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
return next(null, JSON.parse(res.body).template_id);
|
|
}
|
|
);
|
|
},
|
|
function createLayergroup (err, templateId) {
|
|
var next = this;
|
|
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
|
|
if (layergroupId) {
|
|
return next(null, layergroupId);
|
|
}
|
|
|
|
const data = templateId ? params.placeholders : self.mapConfig;
|
|
|
|
if (!params.response) {
|
|
params.response = {
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
};
|
|
}
|
|
|
|
const queryParams = {};
|
|
|
|
if (self.apiKey) {
|
|
queryParams.api_key = self.apiKey;
|
|
}
|
|
|
|
if (params.aggregation !== undefined) {
|
|
queryParams.aggregation = params.aggregation;
|
|
}
|
|
|
|
if (params.client !== undefined) {
|
|
queryParams.client = params.client;
|
|
}
|
|
|
|
const path = templateId
|
|
? `${urlNamed}/${templateId}?${qs.stringify(queryParams)}`
|
|
: `${url}?${qs.stringify(queryParams)}`;
|
|
|
|
assert.response(self.server,
|
|
{
|
|
url: path,
|
|
method: 'POST',
|
|
headers,
|
|
data: JSON.stringify(data)
|
|
},
|
|
params.response,
|
|
function (res, err) {
|
|
var parsedBody;
|
|
// If there is a response, we are still interested in catching the created keys
|
|
// to be able to delete them on the .drain() method.
|
|
if (res) {
|
|
parsedBody = JSON.parse(res.body);
|
|
if (parsedBody.layergroupid) {
|
|
self.keysToDelete['map_cfg|' + LayergroupToken.parse(parsedBody.layergroupid).token] = 0;
|
|
self.keysToDelete['user:localhost:mapviews:global'] = 5;
|
|
}
|
|
if (res.statusCode === 200 && self.template && self.template.layergroup && self.template.layergroup.stat_tag) {
|
|
self.keysToDelete[`user:localhost:mapviews:stat_tag:${self.template.layergroup.stat_tag}`] = 5;
|
|
}
|
|
if (res.statusCode === 200 && self.mapConfig && self.mapConfig.stat_tag) {
|
|
self.keysToDelete[`user:localhost:mapviews:stat_tag:${self.mapConfig.stat_tag}`] = 5;
|
|
}
|
|
}
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
|
|
return callback(null, parsedBody, res);
|
|
}
|
|
);
|
|
}
|
|
);
|
|
};
|
|
|
|
TestClient.prototype.getStaticCenter = function (params, callback) {
|
|
var self = this;
|
|
|
|
const { layergroupid, zoom, lat, lng, width, height, format } = params;
|
|
|
|
var url = '/api/v1/map/';
|
|
if (this.apiKey) {
|
|
url += '?' + qs.stringify({ api_key: this.apiKey });
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, self.extraHeaders);
|
|
|
|
step(
|
|
function createLayergroup () {
|
|
var next = this;
|
|
|
|
if (layergroupid) {
|
|
return next(null, layergroupid);
|
|
}
|
|
|
|
var data = self.mapConfig;
|
|
var path = url;
|
|
|
|
assert.response(self.server,
|
|
{
|
|
url: path,
|
|
method: 'POST',
|
|
headers,
|
|
data: JSON.stringify(data)
|
|
},
|
|
{
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
},
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
return next(null, JSON.parse(res.body).layergroupid);
|
|
}
|
|
);
|
|
},
|
|
function getStaticResult (err, layergroupId) {
|
|
assert.ifError(err);
|
|
|
|
self.keysToDelete['map_cfg|' + LayergroupToken.parse(layergroupId).token] = 0;
|
|
self.keysToDelete['user:localhost:mapviews:global'] = 5;
|
|
|
|
const layergroupid = layergroupidTemplate(layergroupId, params);
|
|
|
|
url = `/api/v1/map/static/center/${layergroupid}/${zoom}/${lat}/${lng}/${width}/${height}.${format}`;
|
|
|
|
if (self.apiKey) {
|
|
url += '?' + qs.stringify({ api_key: self.apiKey });
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost' }, self.extraHeaders);
|
|
|
|
var request = {
|
|
url: url,
|
|
encoding: 'binary',
|
|
method: 'GET',
|
|
headers
|
|
};
|
|
|
|
var expectedResponse = Object.assign({}, {
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'image/png'
|
|
}
|
|
}, params.response);
|
|
|
|
assert.response(self.server, request, expectedResponse, resErr2errRes(this));
|
|
},
|
|
function (err, res) {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
|
|
var body;
|
|
switch (res.headers['content-type']) {
|
|
case 'image/png':
|
|
body = mapnik.Image.fromBytes(Buffer.from(res.body, 'binary'));
|
|
break;
|
|
case 'application/json; charset=utf-8':
|
|
body = JSON.parse(res.body);
|
|
break;
|
|
default:
|
|
body = res.body;
|
|
break;
|
|
}
|
|
|
|
return callback(err, res, body);
|
|
}
|
|
);
|
|
};
|
|
|
|
TestClient.prototype.getNodeStatus = function (nodeName, callback) {
|
|
var self = this;
|
|
|
|
var url = '/api/v1/map';
|
|
|
|
if (this.apiKey) {
|
|
url += '?' + qs.stringify({ api_key: this.apiKey });
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, this.extraHeaders);
|
|
|
|
var nodes = {};
|
|
step(
|
|
function createLayergroup () {
|
|
var next = this;
|
|
assert.response(self.server,
|
|
{
|
|
url: url,
|
|
method: 'POST',
|
|
headers,
|
|
data: JSON.stringify(self.mapConfig)
|
|
},
|
|
{
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
},
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
var parsedBody = JSON.parse(res.body);
|
|
|
|
nodes = parsedBody.metadata.analyses.reduce(function (nodes, analysis) {
|
|
return Object.keys(analysis.nodes).reduce(function (nodes, nodeName) {
|
|
var node = analysis.nodes[nodeName];
|
|
nodes[nodeName] = node.url.http;
|
|
return nodes;
|
|
}, nodes);
|
|
}, nodes);
|
|
|
|
return next(null, parsedBody.layergroupid);
|
|
}
|
|
);
|
|
},
|
|
function getNodeStatusResult (err, layergroupId) {
|
|
assert.ifError(err);
|
|
|
|
self.keysToDelete['map_cfg|' + LayergroupToken.parse(layergroupId).token] = 0;
|
|
self.keysToDelete['user:localhost:mapviews:global'] = 5;
|
|
|
|
url = new URL(nodes[nodeName]).pathname;
|
|
|
|
if (self.apiKey) {
|
|
url += '?' + qs.stringify({ api_key: self.apiKey });
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost' }, self.extraHeaders);
|
|
|
|
var request = {
|
|
url: url,
|
|
method: 'GET',
|
|
headers
|
|
};
|
|
|
|
var expectedResponse = {
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
};
|
|
|
|
assert.response(self.server, request, expectedResponse, resErr2errRes(this));
|
|
},
|
|
function finish (err, res) {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
return callback(null, res, JSON.parse(res.body));
|
|
}
|
|
);
|
|
};
|
|
|
|
TestClient.prototype.getAttributes = function (params, callback) {
|
|
var self = this;
|
|
|
|
if (!Number.isFinite(params.featureId)) {
|
|
throw new Error('featureId param must be a number');
|
|
}
|
|
|
|
if (!Number.isFinite(params.layer)) {
|
|
throw new Error('layer param must be a number');
|
|
}
|
|
|
|
var url = '/api/v1/map';
|
|
|
|
if (this.apiKey) {
|
|
url += '?' + qs.stringify({ api_key: this.apiKey });
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, this.extraHeaders);
|
|
|
|
var layergroupid;
|
|
|
|
if (params.layergroupid) {
|
|
layergroupid = params.layergroupid;
|
|
}
|
|
|
|
step(
|
|
function createLayergroup () {
|
|
var next = this;
|
|
|
|
if (layergroupid) {
|
|
return next(null, layergroupid);
|
|
}
|
|
|
|
assert.response(self.server,
|
|
{
|
|
url: url,
|
|
method: 'POST',
|
|
headers,
|
|
data: JSON.stringify(self.mapConfig)
|
|
},
|
|
{
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
},
|
|
function (res, err) {
|
|
if (err) {
|
|
return next(err);
|
|
}
|
|
var parsedBody = JSON.parse(res.body);
|
|
|
|
return next(null, parsedBody.layergroupid);
|
|
}
|
|
);
|
|
},
|
|
function getAttributes (err, layergroupId) {
|
|
assert.ifError(err);
|
|
|
|
self.keysToDelete['map_cfg|' + LayergroupToken.parse(layergroupId).token] = 0;
|
|
self.keysToDelete['user:localhost:mapviews:global'] = 5;
|
|
|
|
url = `/api/v1/map/${layergroupId}/${params.layer}/attributes/${params.featureId}`;
|
|
|
|
if (self.apiKey) {
|
|
url += '?' + qs.stringify({ api_key: self.apiKey });
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost' }, self.extraHeaders);
|
|
|
|
var request = {
|
|
url: url,
|
|
method: 'GET',
|
|
headers
|
|
};
|
|
|
|
var expectedResponse = params.response || {
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
};
|
|
|
|
assert.response(self.server, request, expectedResponse, resErr2errRes(this));
|
|
},
|
|
function finish (err, res) {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
var attributes = JSON.parse(res.body);
|
|
return callback(null, res, attributes);
|
|
}
|
|
);
|
|
};
|
|
|
|
TestClient.prototype.drain = function (callback) {
|
|
helper.deleteRedisKeys(this.keysToDelete, callback);
|
|
};
|
|
|
|
module.exports.getStaticMap = function getStaticMap (templateName, params, callback) {
|
|
var self = this;
|
|
|
|
self.server = new CartodbWindshaft(serverOptions);
|
|
|
|
if (!callback) {
|
|
callback = params;
|
|
params = null;
|
|
}
|
|
|
|
var url = '/api/v1/map/static/named/' + templateName + '/640/480.png';
|
|
|
|
if (params !== null) {
|
|
url += '?' + qs.stringify(params);
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost' }, self.extraHeaders);
|
|
|
|
var requestOptions = {
|
|
url: url,
|
|
method: 'GET',
|
|
headers,
|
|
encoding: 'binary'
|
|
};
|
|
|
|
var expectedResponse = {
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'image/png'
|
|
}
|
|
};
|
|
|
|
// this could be removed once named maps are invalidated, otherwise you hits the cache
|
|
var server = new CartodbWindshaft(serverOptions);
|
|
|
|
assert.response(server, requestOptions, expectedResponse, function (res, err) {
|
|
helper.deleteRedisKeys({ 'user:localhost:mapviews:global': 5 }, function () {
|
|
return callback(err, mapnik.Image.fromBytes(Buffer.from(res.body, 'binary')));
|
|
});
|
|
});
|
|
};
|
|
|
|
TestClient.prototype.setUserRenderTimeoutLimit = function (user, userTimeoutLimit, callback) {
|
|
const userTimeoutLimitsKey = `limits:timeout:${user}`;
|
|
const params = [
|
|
userTimeoutLimitsKey,
|
|
'render', userTimeoutLimit,
|
|
'render_public', userTimeoutLimit
|
|
];
|
|
|
|
this.keysToDelete[userTimeoutLimitsKey] = 5;
|
|
|
|
helper.configureMetadata('hmset', params, callback);
|
|
};
|
|
|
|
TestClient.prototype.getDBConnection = function () {
|
|
const dbname = _.template(global.environment.postgres_auth_user, { user_id: 1 }) + '_db';
|
|
|
|
const psql = new PSQL({
|
|
user: 'postgres',
|
|
dbname: dbname,
|
|
host: global.environment.postgres.host,
|
|
port: global.environment.postgres.port
|
|
});
|
|
|
|
return psql;
|
|
};
|
|
|
|
TestClient.prototype.setUserDatabaseTimeoutLimit = function (timeoutLimit, callback) {
|
|
const dbname = _.template(global.environment.postgres_auth_user, { user_id: 1 }) + '_db';
|
|
const dbuser = _.template(global.environment.postgres_auth_user, { user_id: 1 });
|
|
const publicuser = global.environment.postgres.user;
|
|
|
|
// IMPORTANT: node-postgres uses internallly a singleton, to refresh all pull connections
|
|
// you need to ensure that your dependency tree has only one working version of `cartodb-psql` & `node-postgres`
|
|
// if not, test using this function cannot ensure that all connections have the new settings (STATEMENT_TIMEOUT)
|
|
//
|
|
// TODO: upgrade to node-postgres@7.x
|
|
const psql = new PSQL({
|
|
user: 'postgres',
|
|
dbname: dbname,
|
|
host: global.environment.postgres.host,
|
|
port: global.environment.postgres.port
|
|
});
|
|
|
|
step(
|
|
function configureTimeouts () {
|
|
const timeoutSQLs = [
|
|
`ALTER ROLE "${publicuser}" SET STATEMENT_TIMEOUT TO ${timeoutLimit}`,
|
|
`ALTER ROLE "${dbuser}" SET STATEMENT_TIMEOUT TO ${timeoutLimit}`,
|
|
`ALTER DATABASE "${dbname}" SET STATEMENT_TIMEOUT TO ${timeoutLimit}`
|
|
];
|
|
|
|
const group = this.group();
|
|
|
|
timeoutSQLs.forEach(sql => psql.query(sql, group()));
|
|
},
|
|
// we need to guarantee all new connections have the new settings
|
|
function refreshPoolConnection () {
|
|
psql.end(() => callback());
|
|
}
|
|
);
|
|
};
|
|
|
|
TestClient.prototype.getAnalysesCatalog = function (params, callback) {
|
|
var url = '/api/v1/map/analyses/catalog';
|
|
|
|
if (this.apiKey) {
|
|
url += '?' + qs.stringify({ api_key: this.apiKey });
|
|
}
|
|
|
|
if (params.jsonp) {
|
|
url += '&' + qs.stringify({ callback: params.jsonp });
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, this.extraHeaders);
|
|
|
|
assert.response(this.server,
|
|
{
|
|
url: url,
|
|
method: 'GET',
|
|
headers
|
|
},
|
|
{
|
|
status: params.status || 200,
|
|
headers: {
|
|
'Content-Type': params.jsonp
|
|
? 'text/javascript; charset=utf-8'
|
|
: 'application/json; charset=utf-8'
|
|
}
|
|
},
|
|
function (res, err) {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
|
|
var parsedBody = params.jsonp ? res.body : JSON.parse(res.body);
|
|
|
|
return callback(null, parsedBody);
|
|
}
|
|
);
|
|
};
|
|
|
|
TestClient.prototype.getNamedMapList = function (params, callback) {
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, this.extraHeaders);
|
|
|
|
const request = {
|
|
url: `/api/v1/map/named?${qs.stringify({ api_key: this.apiKey })}`,
|
|
method: 'GET',
|
|
headers
|
|
};
|
|
|
|
let expectedResponse = {
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
};
|
|
|
|
if (params.response) {
|
|
expectedResponse = Object.assign(expectedResponse, params.response);
|
|
}
|
|
|
|
assert.response(this.server, request, expectedResponse, (res, err) => {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
const body = JSON.parse(res.body);
|
|
return callback(null, res, body);
|
|
});
|
|
};
|
|
|
|
TestClient.prototype.getNamedTile = function (name, z, x, y, format, options, callback) {
|
|
const { params } = options;
|
|
|
|
if (!this.apiKey) {
|
|
return callback(new Error('apiKey param is mandatory to create a new template'));
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, this.extraHeaders);
|
|
|
|
const createTemplateRequest = {
|
|
url: `/api/v1/map/named?${qs.stringify({ api_key: this.apiKey })}`,
|
|
method: 'POST',
|
|
headers,
|
|
data: JSON.stringify(this.template)
|
|
};
|
|
|
|
const createTemplateResponse = {
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
};
|
|
|
|
assert.response(this.server, createTemplateRequest, createTemplateResponse, (res, err) => {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
|
|
const templateId = JSON.parse(res.body).template_id;
|
|
const queryParams = params ? `?${qs.stringify(params)}` : '';
|
|
const url = `/api/v1/map/named/${templateId}/all/${[z, x, y].join('/')}.${format}${queryParams}`;
|
|
const headers = Object.assign({ host: 'localhost' }, this.extraHeaders);
|
|
const namedTileRequest = {
|
|
url,
|
|
method: 'GET',
|
|
headers,
|
|
encoding: 'binary'
|
|
};
|
|
|
|
let contentType;
|
|
switch (format) {
|
|
case 'png':
|
|
contentType = 'image/png';
|
|
break;
|
|
case 'mvt':
|
|
contentType = 'application/x-protobuf';
|
|
break;
|
|
default:
|
|
contentType = 'application/json';
|
|
break;
|
|
}
|
|
|
|
const namedTileResponse = Object.assign({
|
|
status: 200,
|
|
headers: {
|
|
'content-type': contentType
|
|
}
|
|
}, options.response);
|
|
|
|
assert.response(this.server, namedTileRequest, namedTileResponse, (res, err) => {
|
|
let body;
|
|
switch (res.headers['content-type']) {
|
|
case 'image/png':
|
|
body = mapnik.Image.fromBytes(Buffer.from(res.body, 'binary'));
|
|
break;
|
|
case 'application/x-protobuf':
|
|
body = new mapnik.VectorTile(z, x, y);
|
|
body.setDataSync(Buffer.from(res.body, 'binary'));
|
|
break;
|
|
case 'application/json; charset=utf-8':
|
|
body = JSON.parse(res.body);
|
|
break;
|
|
default:
|
|
body = res.body;
|
|
break;
|
|
}
|
|
|
|
return callback(err, res, body);
|
|
});
|
|
});
|
|
};
|
|
|
|
TestClient.prototype.createTemplate = function (params, callback) {
|
|
if (!this.apiKey) {
|
|
return callback(new Error('apiKey param is mandatory to create a new template'));
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json' }, this.extraHeaders);
|
|
|
|
const createTemplateRequest = {
|
|
url: `/api/v1/map/named?${qs.stringify({ api_key: this.apiKey })}`,
|
|
method: 'POST',
|
|
headers,
|
|
data: JSON.stringify(this.template)
|
|
};
|
|
|
|
let createTemplateResponse = {
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
};
|
|
|
|
if (params.response) {
|
|
createTemplateResponse = Object.assign(createTemplateResponse, params.response);
|
|
}
|
|
|
|
assert.response(this.server, createTemplateRequest, createTemplateResponse, (res, err) => {
|
|
let body;
|
|
switch (res.headers['content-type']) {
|
|
case 'application/json; charset=utf-8':
|
|
body = JSON.parse(res.body);
|
|
break;
|
|
default:
|
|
body = res.body;
|
|
break;
|
|
}
|
|
|
|
return callback(err, res, body);
|
|
});
|
|
};
|
|
|
|
TestClient.prototype.deleteTemplate = function (params, callback) {
|
|
if (!this.apiKey) {
|
|
return callback(new Error('apiKey param is mandatory to create a new template'));
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost' }, this.extraHeaders);
|
|
|
|
const deleteTemplateRequest = {
|
|
url: `/api/v1/map/named/${params.templateId}?${qs.stringify({ api_key: this.apiKey })}`,
|
|
method: 'DELETE',
|
|
headers
|
|
};
|
|
|
|
let deleteTemplateResponse = {
|
|
status: 204,
|
|
headers: {}
|
|
};
|
|
|
|
if (params.response) {
|
|
deleteTemplateResponse = Object.assign(deleteTemplateResponse, params.response);
|
|
}
|
|
|
|
assert.response(this.server, deleteTemplateRequest, deleteTemplateResponse, (res, err) => {
|
|
let body;
|
|
switch (res.headers['content-type']) {
|
|
case 'application/json; charset=utf-8':
|
|
body = JSON.parse(res.body);
|
|
break;
|
|
default:
|
|
body = res.body;
|
|
break;
|
|
}
|
|
|
|
return callback(err, res, body);
|
|
});
|
|
};
|
|
|
|
TestClient.prototype.updateTemplate = function (params, callback) {
|
|
if (!this.apiKey) {
|
|
return callback(new Error('apiKey param is mandatory to create a new template'));
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost', 'Content-Type': 'application/json; charset=utf-8' }, this.extraHeaders);
|
|
|
|
const updateTemplateRequest = {
|
|
url: `/api/v1/map/named/${params.templateId}?${qs.stringify({ api_key: this.apiKey })}`,
|
|
method: 'PUT',
|
|
headers,
|
|
data: JSON.stringify(params.templateData)
|
|
};
|
|
|
|
let updateTemplateResponse = {
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
};
|
|
|
|
if (params.response) {
|
|
updateTemplateResponse = Object.assign(updateTemplateResponse, params.response);
|
|
}
|
|
|
|
assert.response(this.server, updateTemplateRequest, updateTemplateResponse, (res, err) => {
|
|
let body;
|
|
switch (res.headers['content-type']) {
|
|
case 'application/json; charset=utf-8':
|
|
body = JSON.parse(res.body);
|
|
break;
|
|
default:
|
|
body = res.body;
|
|
break;
|
|
}
|
|
|
|
return callback(err, res, body);
|
|
});
|
|
};
|
|
|
|
TestClient.prototype.getTemplate = function (params, callback) {
|
|
if (!this.apiKey) {
|
|
return callback(new Error('apiKey param is mandatory to create a new template'));
|
|
}
|
|
|
|
const headers = Object.assign({ host: 'localhost' }, this.extraHeaders);
|
|
|
|
const getTemplateRequest = {
|
|
url: `/api/v1/map/named/${params.templateId}?${qs.stringify({ api_key: this.apiKey })}`,
|
|
method: 'GET',
|
|
headers
|
|
};
|
|
|
|
let getTemplateResponse = {
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'application/json; charset=utf-8'
|
|
}
|
|
};
|
|
|
|
if (params.response) {
|
|
getTemplateResponse = Object.assign(getTemplateResponse, params.response);
|
|
}
|
|
|
|
assert.response(this.server, getTemplateRequest, getTemplateResponse, (res, err) => {
|
|
let body;
|
|
switch (res.headers['content-type']) {
|
|
case 'application/json; charset=utf-8':
|
|
body = JSON.parse(res.body);
|
|
break;
|
|
default:
|
|
body = res.body;
|
|
break;
|
|
}
|
|
|
|
return callback(err, res, body);
|
|
});
|
|
};
|
|
|
|
TestClient.prototype.getPreview = function (width, height, params = {}, callback) {
|
|
this.createTemplate({}, (err, res, template) => {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
|
|
params = Object.assign({ api_key: this.apiKey }, params);
|
|
const url = `/api/v1/map/static/named/${template.template_id}/${width}/${height}.png?${qs.stringify(params)}`;
|
|
const headers = Object.assign({ host: 'localhost' }, this.extraHeaders);
|
|
|
|
const requestOptions = {
|
|
url: url,
|
|
method: 'GET',
|
|
headers,
|
|
encoding: 'binary'
|
|
};
|
|
|
|
const expectedResponse = Object.assign({
|
|
status: 200,
|
|
headers: {
|
|
'Content-Type': 'image/png'
|
|
}
|
|
}, params.response || {});
|
|
|
|
assert.response(this.server, requestOptions, expectedResponse, (res, err) => {
|
|
if (err) {
|
|
return callback(err);
|
|
}
|
|
|
|
let body;
|
|
switch (res.headers['content-type']) {
|
|
case 'image/png':
|
|
this.keysToDelete['user:localhost:mapviews:global'] = 5;
|
|
if (this.template.layergroup && this.template.layergroup.stat_tag) {
|
|
this.keysToDelete[`user:localhost:mapviews:stat_tag:${this.template.layergroup.stat_tag}`] = 5;
|
|
}
|
|
body = mapnik.Image.fromBytes(Buffer.from(res.body, 'binary'));
|
|
break;
|
|
case 'application/json; charset=utf-8':
|
|
body = JSON.parse(res.body);
|
|
break;
|
|
default:
|
|
body = res.body;
|
|
break;
|
|
}
|
|
|
|
return callback(null, res, body);
|
|
});
|
|
});
|
|
};
|