cartodb/CartoDB-SQL-API
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

A much cleaner test

Browse Source
This commit is contained in:
Rafa de la Torre 2018-06-11 18:38:18 +02:00
parent 4b6ee133df
commit 62fe8abd0a
1 changed files with 22 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
test/acceptance/copy-endpoints.js
View File

@ -345,44 +345,32 @@ describe('copy-endpoints client disconnection', function() {
}); });
it('copy to returns the connection to the pool if the client disconnects', function(done) { it('copy to returns the connection to the pool if the client disconnects', function(done) {
const request = require('request');
const port = 5555;
const host = '127.0.0.1';
var listen = function() { var assertCanReuseConnection = function () {
var listener = server.listen(port, host); assert.response(server, {
listener.on('listening', onServerListening); url: '/api/v1/sql?' + querystring.stringify({
}; q: 'SELECT 1',
var onServerListening = function() {
const requestParams = {
url: 'http://' + host + ':' + port + '/api/v1/sql/copyto?' + querystring.stringify({
q: 'COPY populated_places_simple_reduced TO STDOUT',
}), }),
headers: { host: 'vizzuality.cartodb.com' }, headers: { host: 'vizzuality.cartodb.com' },
method: 'GET', method: 'GET'
timeout: 1 }, {}, function(err, res) {
}; assert.ifError(err);
request(requestParams, function(err, response, body) { assert.ok(res.statusCode === 200);
// we're expecting a timeout error done();
assert.ok(err.code === 'ETIMEDOUT' || err.code === 'ESOCKETTIMEDOUT');
// check now that we can perform a regular query, cause the connection should be back to the pool
request({
url: 'http://' + host + ':' + port + '/api/v1/sql?' + querystring.stringify({
q: 'SELECT 1',
}),
headers: { host: 'vizzuality.cartodb.com' },
method: 'GET',
timeout: 5000
}, function(err, response, body) {
assert.ifError(err);
assert(response.statusCode == 200);
done();
});
}); });
} };
listen(); assert.response(server, {
url: '/api/v1/sql/copyto?' + querystring.stringify({
q: 'COPY populated_places_simple_reduced TO STDOUT',
}),
headers: { host: 'vizzuality.cartodb.com' },
method: 'GET',
timeout: 1
}, {}, function(err, res) {
// we're expecting a timeout error
assert.ok(err.code === 'ETIMEDOUT' || err.code === 'ESOCKETTIMEDOUT');
assertCanReuseConnection();
});
}); });
}); });