first commit

2023-05-19 00:42:48 +08:00
commit 53de9c6c51
243 changed files with 39485 additions and 0 deletions
@@ -0,0 +1,101 @@
+'use strict';
+
+require('../helper');
+
+var qs = require('querystring');
+
+var server = require('../../app/server')();
+var assert = require('../support/assert');
+
+describe('query-tables-api', function() {
+
+    beforeEach(function(done) {
+        var tableCacheEnabled = global.settings.tableCacheEnabled || false;
+        if(!tableCacheEnabled) {
+            this.skip("tableCache is disabled");
+        }
+        done();
+    });
+
+    function getCacheStatus(callback) {
+        assert.response(
+            server,
+            {
+                method: 'GET',
+                url: '/api/v1/cachestatus'
+            },
+            {
+                status: 200
+            },
+            function(err, res) {
+                callback(null, JSON.parse(res.body));
+            }
+        );
+    }
+
+    var request = {
+        url: '/api/v1/sql?' + qs.stringify({
+            q: 'SELECT * FROM untitle_table_4'
+        }),
+        headers: {
+            host: 'vizzuality.cartodb.com'
+        },
+        method: 'GET'
+    };
+
+    var RESPONSE_OK = {
+        status: 200
+    };
+
+    it('should create a key in affected tables cache', function(done) {
+        assert.response(server, request, RESPONSE_OK, function(err) {
+            assert.ok(!err, err);
+
+            getCacheStatus(function(err, cacheStatus) {
+                assert.ok(!err, err);
+                assert.equal(cacheStatus.explain.keys, 1);
+                assert.equal(cacheStatus.explain.hits, 0);
+
+                done();
+            });
+        });
+    });
+
+    it('should use cache to retrieve affected tables', function(done) {
+        assert.response(server, request, RESPONSE_OK, function(err) {
+            assert.ok(!err, err);
+
+            getCacheStatus(function(err, cacheStatus) {
+                assert.ok(!err, err);
+                assert.equal(cacheStatus.explain.keys, 1);
+                assert.equal(cacheStatus.explain.hits, 1);
+
+                done();
+            });
+        });
+    });
+
+    it('should skip cache to retrieve affected tables', function(done) {
+        var masterRequest = {
+            url: '/api/v1/sql?' + qs.stringify({
+                q: 'SELECT * FROM untitle_table_4',
+                api_key: '1234'
+            }),
+            headers: {
+                host: 'vizzuality.cartodb.com'
+            },
+            method: 'GET'
+        };
+        assert.response(server, masterRequest, RESPONSE_OK, function(err) {
+            assert.ok(!err, err);
+
+            getCacheStatus(function(err, cacheStatus) {
+                assert.ok(!err, err);
+                assert.equal(cacheStatus.explain.keys, 1);
+                assert.equal(cacheStatus.explain.hits, 0);
+
+                done();
+            });
+        });
+    });
+});