Integration tests for requirements/limits

CartoDB · Oct 28, 2016 · 173a8d9 · 173a8d9
1 parent 880fa16
commit 173a8d9
Showing 1 changed file with 66 additions and 0 deletions.
diff --git a/test/integration/analysis.js b/test/integration/analysis.js
@@ -102,6 +102,72 @@ describe('workflow', function() {
             });
         });
 
+        it('should compute node requirements and limits for source', function(done) {
+            Analysis.create(testConfig, sourceAnalysisDefinition, function(err, analysis) {
+                assert.ok(!err, err);
+                assert.equal(analysis.getRoot().estimatedRequirements.numberOfRows, 6);
+                assert.equal(analysis.getRoot().limits.maximumNumberOfRows, 1000000);
+                done();
+            });
+        });
+
+        it('should abort analysis over the limits for source', function(done) {
+            var limitedConfig = testConfig.create({
+                limits: {
+                    analyses: {
+                        source: {
+                            maximumNumberOfRows: 5
+                        }
+                    }
+                }
+            });
+            Analysis.create(limitedConfig, sourceAnalysisDefinition, function(err, analysis) {
+                assert.ok(err);
+                done();
+            });
+        });
+
+        it('should compute node requirements and limits for trade areas', function(done) {
+            var enqueueFn = BatchClient.prototype.enqueue;
+
+            BatchClient.prototype.enqueue = function(query, callback) {
+                return callback(null, {status: 'ok'});
+            };
+
+            Analysis.create(testConfig, tradeAreaAnalysisDefinition, function(err, analysis) {
+                BatchClient.prototype.enqueue = enqueueFn;
+
+                assert.ok(!err, err);
+                assert.equal(analysis.getRoot().estimatedRequirements.numberOfRows, 6);
+                assert.equal(analysis.getRoot().limits.maximumNumberOfRows, 1000000);
+                done();
+            });
+        });
+
+        it('should abort analysis over the limits for trade areas', function(done) {
+            var limitedConfig = testConfig.create({
+                limits: {
+                    analyses: {
+                        'trade-area': {
+                            maximumNumberOfRows: 5
+                        }
+                    }
+                }
+            });
+
+            var enqueueFn = BatchClient.prototype.enqueue;
+            BatchClient.prototype.enqueue = function(query, callback) {
+                return callback(null, {status: 'ok'});
+            };
+
+            Analysis.create(limitedConfig, tradeAreaAnalysisDefinition, function(err, analysis) {
+                BatchClient.prototype.enqueue = enqueueFn;
+
+                assert.ok(err);
+                done();
+            });
+        });
+
     });
 
 });