2018-10-24 21:42:33 +08:00
|
|
|
'use strict';
|
|
|
|
|
2016-12-09 18:10:08 +08:00
|
|
|
require('../helper');
|
|
|
|
|
2019-10-04 00:24:39 +08:00
|
|
|
var server = require('../../lib/server')();
|
2016-12-09 18:10:08 +08:00
|
|
|
var assert = require('../support/assert');
|
|
|
|
var querystring = require('querystring');
|
|
|
|
|
|
|
|
|
|
|
|
describe('copy-statements', function() {
|
|
|
|
|
|
|
|
var RESPONSE_OK = {
|
|
|
|
statusCode: 200
|
|
|
|
};
|
|
|
|
|
|
|
|
before(function(done) {
|
|
|
|
assert.response(server, {
|
|
|
|
url: "/api/v1/sql?" + querystring.stringify({
|
|
|
|
q: 'CREATE TABLE copy_test_table(a int)',
|
|
|
|
api_key: 1234
|
|
|
|
}),
|
|
|
|
headers: {host: 'vizzuality.cartodb.com'},
|
|
|
|
method: 'GET'
|
|
|
|
}, RESPONSE_OK, done);
|
|
|
|
});
|
|
|
|
|
|
|
|
after(function(done) {
|
|
|
|
assert.response(server, {
|
|
|
|
url: "/api/v1/sql?" + querystring.stringify({
|
2016-12-09 19:05:49 +08:00
|
|
|
q: 'DROP TABLE IF EXISTS copy_test_table',
|
2016-12-09 18:10:08 +08:00
|
|
|
api_key: 1234
|
|
|
|
}),
|
|
|
|
headers: {host: 'vizzuality.cartodb.com'},
|
|
|
|
method: 'GET'
|
|
|
|
}, RESPONSE_OK, done);
|
|
|
|
});
|
|
|
|
|
|
|
|
// Test effects of COPY
|
|
|
|
// See https://github.com/Vizzuality/cartodb-management/issues/1502
|
|
|
|
it('COPY TABLE with GET and auth', function(done){
|
|
|
|
assert.response(server, {
|
|
|
|
url: "/api/v1/sql?" + querystring.stringify({
|
|
|
|
q: 'COPY copy_test_table FROM stdin;',
|
|
|
|
api_key: 1234
|
|
|
|
}),
|
|
|
|
headers: {host: 'vizzuality.cartodb.com'},
|
|
|
|
method: 'GET'
|
|
|
|
},{}, function(err, res) {
|
|
|
|
// We expect a problem, actually
|
|
|
|
assert.equal(res.statusCode, 400, res.statusCode + ': ' + res.body);
|
|
|
|
assert.deepEqual(res.headers['content-type'], 'application/json; charset=utf-8');
|
|
|
|
assert.deepEqual(res.headers['content-disposition'], 'inline');
|
|
|
|
assert.deepEqual(JSON.parse(res.body), {"error":["COPY from stdin failed: No source stream defined"]});
|
|
|
|
done();
|
|
|
|
});
|
|
|
|
});
|
|
|
|
|
|
|
|
it('COPY TABLE with GET and auth', function(done){
|
|
|
|
assert.response(server, {
|
|
|
|
url: "/api/v1/sql?" + querystring.stringify({
|
|
|
|
q: "COPY copy_test_table to '/tmp/x';",
|
|
|
|
api_key: 1234
|
|
|
|
}),
|
|
|
|
headers: {host: 'vizzuality.cartodb.com'},
|
|
|
|
method: 'GET'
|
|
|
|
},{}, function(err, res) {
|
|
|
|
// We expect a problem, actually
|
|
|
|
assert.equal(res.statusCode, 400, res.statusCode + ': ' + res.body);
|
|
|
|
assert.deepEqual(res.headers['content-type'], 'application/json; charset=utf-8');
|
|
|
|
assert.deepEqual(res.headers['content-disposition'], 'inline');
|
2019-01-02 20:09:42 +08:00
|
|
|
const error_exp = /must be superuser.* to COPY.* a file/;
|
|
|
|
const hint_exp = /Anyone can COPY to stdout or from stdin. psql's \\copy command also works for anyone./;
|
|
|
|
assert.ok(JSON.parse(res.body).error[0].match(error_exp));
|
|
|
|
assert.ok(JSON.parse(res.body).hint.match(hint_exp));
|
2016-12-09 18:10:08 +08:00
|
|
|
done();
|
|
|
|
});
|
|
|
|
});
|
|
|
|
|
|
|
|
|
|
|
|
});
|