195 lines
6.0 KiB
JavaScript
195 lines
6.0 KiB
JavaScript
const TestClient = require('../../support/test-client');
|
|
|
|
require('../../support/assert');
|
|
|
|
var assert = require('assert');
|
|
var querystring = require('querystring');
|
|
|
|
describe('timeout', function () {
|
|
describe('export database', function () {
|
|
const databaseTimeoutQuery = `
|
|
select
|
|
ST_SetSRID(ST_Point(0, 0), 4326) as the_geom,
|
|
pg_sleep(0.2) as sleep,
|
|
1 as value
|
|
`;
|
|
|
|
const scenarios = [
|
|
{
|
|
desc: 'CSV',
|
|
format: 'csv',
|
|
contentType: 'application/x-www-form-urlencoded',
|
|
parser: querystring.stringify,
|
|
// only: true,
|
|
skip: true
|
|
},
|
|
{
|
|
query: databaseTimeoutQuery,
|
|
desc: 'Geopackage',
|
|
format: 'gpkg'
|
|
},
|
|
{
|
|
query: databaseTimeoutQuery,
|
|
desc: 'KML',
|
|
format: 'kml'
|
|
},
|
|
{
|
|
query: databaseTimeoutQuery,
|
|
desc: 'Shapefile',
|
|
format: 'shp'
|
|
},
|
|
{
|
|
query: databaseTimeoutQuery,
|
|
desc: 'Spatialite',
|
|
format: 'spatialite'
|
|
},
|
|
{
|
|
query: databaseTimeoutQuery,
|
|
desc: 'Array Buffer',
|
|
format: 'arraybuffer'
|
|
},
|
|
{
|
|
query: databaseTimeoutQuery,
|
|
desc: 'GeoJSON',
|
|
format: 'geojson'
|
|
},
|
|
{
|
|
query: databaseTimeoutQuery,
|
|
desc: 'JSON',
|
|
format: 'json'
|
|
},
|
|
{
|
|
query: databaseTimeoutQuery,
|
|
desc: 'SVG',
|
|
format: 'svg'
|
|
},
|
|
{
|
|
query: databaseTimeoutQuery,
|
|
desc: 'TopoJSON',
|
|
format: 'topojson'
|
|
}
|
|
];
|
|
|
|
beforeEach(function (done) {
|
|
this.testClient = new TestClient();
|
|
this.testClient.setUserDatabaseTimeoutLimit('localhost', 100, done);
|
|
});
|
|
|
|
afterEach(function (done) {
|
|
this.testClient.setUserDatabaseTimeoutLimit('localhost', 2000, done);
|
|
});
|
|
|
|
scenarios.forEach((scenario) => {
|
|
const test = scenario.only ? it.only : scenario.skip ? it.skip : it;
|
|
|
|
test(`${scenario.desc} export exceeding statement timeout responds 429 Over Limits`, function (done) {
|
|
const override = {
|
|
'Content-Type': scenario.contentType,
|
|
parser: scenario.parser,
|
|
anonymous: true,
|
|
format: scenario.format,
|
|
response: {
|
|
status: 429
|
|
}
|
|
};
|
|
|
|
this.testClient.getResult(scenario.query, override, (err, res) => {
|
|
assert.ifError(err);
|
|
|
|
assert.deepEqual(res, {
|
|
error: [
|
|
'You are over platform\'s limits. Please contact us to know more details'
|
|
],
|
|
context: 'limit',
|
|
detail: 'datasource'
|
|
});
|
|
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('export ogr command timeout', function () {
|
|
const ogrCommandTimeoutQuery = `
|
|
select
|
|
ST_SetSRID(ST_Point(0, 0), 4326) as the_geom,
|
|
pg_sleep(0.2) as sleep,
|
|
1 as value
|
|
`;
|
|
|
|
const scenarios = [
|
|
{
|
|
query: ogrCommandTimeoutQuery,
|
|
desc: 'CSV',
|
|
format: 'csv',
|
|
contentType: 'application/x-www-form-urlencoded',
|
|
parser: querystring.stringify,
|
|
// only: true,
|
|
// skip: true
|
|
},
|
|
{
|
|
query: ogrCommandTimeoutQuery,
|
|
filename: 'wadus_gpkg_filename',
|
|
desc: 'Geopackage',
|
|
format: 'gpkg'
|
|
},
|
|
{
|
|
query: ogrCommandTimeoutQuery,
|
|
desc: 'KML',
|
|
format: 'kml'
|
|
},
|
|
{
|
|
query: ogrCommandTimeoutQuery,
|
|
desc: 'Shapefile',
|
|
format: 'shp'
|
|
},
|
|
{
|
|
query: ogrCommandTimeoutQuery,
|
|
desc: 'Spatialite',
|
|
format: 'spatialite'
|
|
}
|
|
];
|
|
|
|
beforeEach(function (done) {
|
|
this.testClient = new TestClient();
|
|
this.testClient.setUserRenderTimeoutLimit('vizzuality', 100, done);
|
|
});
|
|
|
|
afterEach(function (done) {
|
|
this.testClient.setUserRenderTimeoutLimit('vizzuality', 0, done);
|
|
});
|
|
|
|
scenarios.forEach((scenario) => {
|
|
const test = scenario.only ? it.only : scenario.skip ? it.skip : it;
|
|
|
|
test(`${scenario.desc} export exceeding statement timeout responds 429 Over Limits`, function (done) {
|
|
const override = {
|
|
'Content-Type': scenario.contentType,
|
|
parser: scenario.parser,
|
|
anonymous: true,
|
|
format: scenario.format,
|
|
filename: scenario.filename,
|
|
response: {
|
|
status: 429
|
|
}
|
|
};
|
|
|
|
this.testClient.getResult(scenario.query, override, (err, res) => {
|
|
assert.ifError(err);
|
|
|
|
assert.deepEqual(res, {
|
|
error: [
|
|
'You are over platform\'s limits. Please contact us to know more details'
|
|
],
|
|
context: 'limit',
|
|
detail: 'datasource'
|
|
});
|
|
|
|
done();
|
|
});
|
|
});
|
|
});
|
|
});
|
|
});
|