first commit

2023-05-19 00:42:48 +08:00
commit 53de9c6c51
243 changed files with 39485 additions and 0 deletions
@@ -0,0 +1,150 @@
+'use strict';
+
+require('../helper');
+var assert = require('assert');
+var appServer = require('../../app/server');
+var redisUtils = require('./redis_utils');
+const step = require('step');
+const PSQL = require('cartodb-psql');
+const _ = require('underscore');
+
+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, override, callback) {
+    if (!callback) {
+        callback = override;
+        override = {};
+    }
+
+    assert.response(
+        this.server,
+        {
+            url: this.getUrl(override),
+            headers: {
+                host: this.getHost(override),
+                'Content-Type': this.getContentType(override),
+                authorization: this.getAuthorization(override)
+            },
+            method: 'POST',
+            data: this.getParser(override)({
+                q: query,
+                format: this.getFormat(override),
+                filename: this.getFilename(override)
+            })
+        },
+        this.getExpectedResponse(override),
+        function (err, res) {
+            if (err) {
+                return callback(err);
+            }
+            var result = JSON.parse(res.body);
+
+            if (res.statusCode > 299) {
+                return callback(null, result);
+            }
+
+            return callback(null, result.rows || [], result);
+        }
+    );
+};
+
+TestClient.prototype.getHost = function(override) {
+    return override.host || this.config.host || 'vizzuality.cartodb.com';
+};
+
+TestClient.prototype.getAuthorization = function (override) {
+    const auth = override.authorization || this.config.authorization;
+
+    if (auth) {
+        return `Basic ${new Buffer(auth).toString('base64')}`;
+    }
+};
+
+TestClient.prototype.getContentType = function(override) {
+    return override['Content-Type'] || this.config['Content-Type'] || 'application/json';
+};
+
+TestClient.prototype.getParser = function (override) {
+    return override.parser || this.config.parser || JSON.stringify
+}
+
+TestClient.prototype.getUrl = function(override) {
+    if (override.anonymous) {
+        return '/api/v1/sql?';
+    }
+
+    return '/api/v2/sql?api_key=' + (override.apiKey || this.config.apiKey || '1234');
+};
+
+TestClient.prototype.getExpectedResponse = function (override) {
+    return override.response || this.config.response || RESPONSE.OK;
+};
+
+TestClient.prototype.getFormat = function (override) {
+    return override.format || this.config.format || undefined;
+};
+
+TestClient.prototype.getFilename = function (override) {
+    return override.filename || this.config.filename || undefined;
+};
+
+TestClient.prototype.setUserRenderTimeoutLimit = function (user, userTimeoutLimit, callback) {
+    const userTimeoutLimitsKey = `limits:timeout:${user}`;
+    const params = [
+        userTimeoutLimitsKey,
+        'render', userTimeoutLimit,
+        'render_public', userTimeoutLimit
+    ];
+
+    redisUtils.configureUserMetadata('hmset', params, callback);
+};
+
+TestClient.prototype.setUserDatabaseTimeoutLimit = function (user, timeoutLimit, callback) {
+    const dbname = _.template(global.settings.db_base_name, { user_id: 1 });
+    const dbuser = _.template(global.settings.db_user, { user_id: 1 })
+    const pass = _.template(global.settings.db_user_pass, { user_id: 1 })
+    const publicuser = global.settings.db_pubuser;
+
+    const psql = new PSQL({
+        user: 'postgres',
+        dbname: dbname,
+        host: global.settings.db_host,
+        port: global.settings.db_port
+    });
+
+    // we need to guarantee all new connections have the new settings
+    psql.end();
+
+    step(
+        function configureTimeouts () {
+            const timeoutSQLs = [
+                `ALTER ROLE "${publicuser}" SET STATEMENT_TIMEOUT TO ${timeoutLimit}`,
+                `ALTER ROLE "${dbuser}" SET STATEMENT_TIMEOUT TO ${timeoutLimit}`,
+                `ALTER DATABASE "${dbname}" SET STATEMENT_TIMEOUT TO ${timeoutLimit}`
+            ];
+
+            const group = this.group();
+
+            timeoutSQLs.forEach(sql => psql.query(sql, group()));
+        },
+        callback
+    );
+};