180 lines
5.4 KiB
JavaScript
180 lines
5.4 KiB
JavaScript
'use strict';
|
|
|
|
require('../helper');
|
|
|
|
const server = require('../../app/server')();
|
|
const assert = require('../support/assert');
|
|
const qs = require('querystring');
|
|
const BatchTestClient = require('../support/batch-test-client');
|
|
const { TYPES } = require('../../app/middlewares/log');
|
|
|
|
const QUERY = `SELECT 14 as foo`;
|
|
const API_KEY = 1234;
|
|
|
|
const logQueries = global.settings.logQueries;
|
|
|
|
describe('Log middleware', function() {
|
|
before(function() {
|
|
global.settings.logQueries = true;
|
|
});
|
|
|
|
after(function() {
|
|
global.settings.logQueries = logQueries;
|
|
});
|
|
|
|
describe('regular queries endpoint', function() {
|
|
['GET', 'POST'].forEach(method => {
|
|
it(`${method} query`, function(done) {
|
|
assert.response(server,
|
|
{
|
|
method,
|
|
url: '/api/v1/sql?' + qs.stringify({
|
|
q: QUERY,
|
|
api_key: API_KEY
|
|
}),
|
|
headers: {
|
|
host: 'vizzuality.cartodb.com'
|
|
}
|
|
},
|
|
{ statusCode: 200 },
|
|
function(err, res) {
|
|
assert.ok(!err);
|
|
|
|
assert.ok(res.headers['x-sqlapi-log']);
|
|
const log = JSON.parse(res.headers['x-sqlapi-log']);
|
|
assert.deepEqual(log, {
|
|
request: {
|
|
sql: {
|
|
type: TYPES.QUERY,
|
|
sql: QUERY
|
|
}
|
|
}
|
|
});
|
|
|
|
return done();
|
|
}
|
|
);
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('batch api queries', function() {
|
|
before(function() {
|
|
this.batchTestClient = new BatchTestClient();
|
|
});
|
|
|
|
after(function(done) {
|
|
this.batchTestClient.drain(done);
|
|
});
|
|
|
|
it('one query', function (done) {
|
|
const payload = { query: QUERY };
|
|
this.batchTestClient.createJob(payload, function(err, jobResult, res) {
|
|
assert.ok(!err);
|
|
|
|
assert.ok(res.headers['x-sqlapi-log']);
|
|
const log = JSON.parse(res.headers['x-sqlapi-log']);
|
|
assert.deepEqual(log, {
|
|
request: {
|
|
sql: {
|
|
type: TYPES.JOB,
|
|
sql: QUERY
|
|
}
|
|
}
|
|
});
|
|
|
|
return done();
|
|
});
|
|
});
|
|
|
|
it('multiquery job with two queries', function (done) {
|
|
const payload = { query: [QUERY, QUERY] };
|
|
this.batchTestClient.createJob(payload, function(err, jobResult, res) {
|
|
assert.ok(!err);
|
|
|
|
assert.ok(res.headers['x-sqlapi-log']);
|
|
const log = JSON.parse(res.headers['x-sqlapi-log']);
|
|
assert.deepEqual(log, {
|
|
request: {
|
|
sql: {
|
|
type: TYPES.JOB,
|
|
sql: [QUERY, QUERY]
|
|
}
|
|
}
|
|
});
|
|
|
|
return done();
|
|
});
|
|
});
|
|
|
|
it('queries with fallbacks', function (done) {
|
|
const FALLBACK_QUERY = {
|
|
query: [{
|
|
query: QUERY,
|
|
onsuccess: QUERY,
|
|
onerror: QUERY
|
|
}],
|
|
onsuccess: QUERY,
|
|
onerror: QUERY
|
|
};
|
|
const payload = { query: FALLBACK_QUERY };
|
|
|
|
this.batchTestClient.createJob(payload, function(err, jobResult, res) {
|
|
assert.ok(!err);
|
|
|
|
assert.ok(res.headers['x-sqlapi-log']);
|
|
const log = JSON.parse(res.headers['x-sqlapi-log']);
|
|
assert.deepEqual(log, {
|
|
request: {
|
|
sql: {
|
|
type: TYPES.JOB,
|
|
sql: FALLBACK_QUERY
|
|
}
|
|
}
|
|
});
|
|
|
|
return done();
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('disable queries log', function() {
|
|
before(function() {
|
|
global.settings.logQueries = false;
|
|
});
|
|
|
|
after(function() {
|
|
global.settings.logQueries = true;
|
|
});
|
|
|
|
it(`GET query`, function(done) {
|
|
assert.response(server,
|
|
{
|
|
method: 'GET',
|
|
url: '/api/v1/sql?' + qs.stringify({
|
|
q: QUERY,
|
|
api_key: API_KEY
|
|
}),
|
|
headers: {
|
|
host: 'vizzuality.cartodb.com'
|
|
}
|
|
},
|
|
{ statusCode: 200 },
|
|
function(err, res) {
|
|
assert.ok(!err);
|
|
|
|
assert.ok(res.headers['x-sqlapi-log']);
|
|
const log = JSON.parse(res.headers['x-sqlapi-log']);
|
|
assert.deepEqual(log, {
|
|
request: {
|
|
sql: null
|
|
}
|
|
});
|
|
|
|
return done();
|
|
}
|
|
);
|
|
});
|
|
});
|
|
});
|