Implement tests to check the limit error is the expected one

test/acceptance/max-waiting-workers.js

@@ -0,0 +1,129 @@
+'use strict';
+
+require('../support/test_helper');
+
+const assert = require('../support/assert');
+const TestClient = require('../support/test-client');
+const createMapConfig = ({
+    version = '1.8.0',
+    type = 'cartodb',
+    sql = TestClient.SQL.ONE_POINT,
+    cartocss = TestClient.CARTOCSS.POINTS,
+    cartocss_version = '2.3.0',
+    interactivity = 'cartodb_id'
+} = {}) => ({
+    version,
+    layers: [{
+        type,
+        options: {
+            source: {
+                id: 'a0'
+            },
+            cartocss,
+            cartocss_version,
+            interactivity
+        }
+    }],
+    analyses: [
+        {
+            id: 'a0',
+            type: 'source',
+            params: {
+                query: sql
+            }
+        }
+    ]
+});
+const coords = [
+    [0, 0, 0],
+    [1, 0, 0],
+    [1, 0, 1],
+    [1, 1, 0],
+    [1, 1, 1],
+    [2, 0, 0],
+    [2, 0, 1],
+    [2, 0, 2],
+    [2, 0, 3],
+    [2, 1, 0],
+    [2, 1, 1],
+    [2, 1, 2],
+    [2, 1, 3],
+    [2, 2, 0],
+    [2, 2, 1],
+    [2, 2, 2],
+    [2, 2, 3],
+    [2, 3, 0],
+    [2, 3, 1],
+    [2, 3, 2],
+    [2, 3, 3]
+];
+
+function getTiles ({ testClient, layergroupid, coords }) {
+    return Promise.all(coords.map((coord) => getTile({ testClient, layergroupid, coord })));
+}
+
+function getTile ({ testClient, layergroupid, coord }) {
+    return new Promise((resolve, reject) => {
+        const [ z, x, y ] = coord;
+        const params = {
+            layergroupid,
+            format: 'png',
+            response: {
+                status: [ 200, 429 ],
+                headers: {
+                    'Content-Type': /^(image\/png|application\/json; charset=utf-8)$/
+                }
+            }
+        };
+
+        testClient.getTile(z, x, y, params, (err, res, tile) => {
+            if (err) {
+                return reject(err);
+            }
+
+            return resolve({ res, tile });
+        });
+    });
+}
+
+describe('exceeding max waiting workers', function () {
+    const limitErrorMessage = 'You are over platform\'s limits: Max render capacity exceeded Contact CARTO support for more details.';
+    const originalPoolSize = global.environment.renderer.mapnik.poolSize;
+    const poolMaxWaitingClients = global.environment.renderer.mapnik.poolMaxWaitingClients;
+    const apikey = 1234;
+    const testClient = new TestClient(createMapConfig(), apikey);
+    let layergroupid;
+
+    before(function (done) {
+        global.environment.renderer.mapnik.poolSize = 1;
+        global.environment.renderer.mapnik.poolMaxWaitingClients = 1;
+
+        testClient.getLayergroup({ status: 200 }, (err, res) => {
+            if (err) {
+                return done(err);
+            }
+
+            layergroupid = res.layergroupid;
+            done();
+        });
+    });
+
+    after(function () {
+        global.environment.renderer.mapnik.poolSize = originalPoolSize;
+        global.environment.renderer.mapnik.poolMaxWaitingClients = poolMaxWaitingClients;
+    });
+
+    it('should get 429: You are over platform\'s limits', function (done) {
+        const testClient = new TestClient(createMapConfig(), apikey);
+
+        getTiles({ testClient, layergroupid, coords })
+            .then((results) => {
+                const errs = results
+                    .filter(({ res }) => res.headers['content-type'] === 'application/json; charset=utf-8')
+                    .filter(({ tile }) => tile.errors && tile.errors[0] === limitErrorMessage);
+
+                assert.ok(errs.length > 0);
+                testClient.drain(done);
+            });
+    });
+});

test/support/assert.js

@@ -142,16 +142,18 @@ function validateResponseBody(response, expected) {
 
 function validateResponseStatus(response, expected) {
     var status = expected.status || expected.statusCode;
+    const message = colorize('[red]{Invalid response status code.}\n' +
+                    '     Expected: [green]{' + status + '}\n' +
+                    '     Got: [red]{' + response.statusCode + '}\n' +
+                    '     Body: ' + response.body);
+
     // Assert response status
-    if (typeof status === 'number') {
-        if (response.statusCode !== status) {
-            return new Error(colorize(
-                '[red]{Invalid response status code.}\n' +
-                '     Expected: [green]{' + status + '}\n' +
-                '     Got: [red]{' + response.statusCode + '}\n' +
-                '     Body: ' + response.body)
-            );
-        }
+    if (typeof status === 'number' && response.statusCode !== status) {
+        return new Error(message);
+    }
+
+    if (Array.isArray(status) && !status.includes(response.statusCode)) {
+        return new Error(message);
     }
 }