CartoDB-SQL-API/test/support/test-client.js

60 lines
1.2 KiB
JavaScript
Raw Normal View History

2016-10-10 21:37:46 +08:00
'use strict';
require('../helper');
var assert = require('assert');
var appServer = require('../../app/server');
function response(code) {
return {
status: code
};
}
var RESPONSE = {
OK: response(200),
CREATED: response(201)
};
function TestClient(config) {
this.config = config || {};
this.server = appServer();
}
module.exports = TestClient;
TestClient.prototype.getResult = function(query, callback) {
assert.response(
this.server,
{
url: this.getUrl(),
headers: {
host: this.getHost(),
'Content-Type': 'application/json'
},
method: 'POST',
data: JSON.stringify({
q: query
})
},
RESPONSE.OK,
function (err, res) {
if (err) {
return callback(err);
}
var result = JSON.parse(res.body);
return callback(null, result.rows || []);
}
);
};
TestClient.prototype.getHost = function() {
return this.config.host || 'vizzuality.cartodb.com';
};
TestClient.prototype.getUrl = function() {
return '/api/v2/sql?api_key=' + (this.config.apiKey || '1234');
};