integrated memory-usage.js into test.js

added linear regression test to detect memory leaks
calvium · Aug 8, 2016 · 5f602c4 · 5f602c4
1 parent 2ef1c73
commit 5f602c4
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 29 deletions.
diff --git a/test/memory-usage.js b/test/memory-usage.js
diff --git a/test/test.js b/test/test.js
@@ -1,3 +1,4 @@
+'use strict';
 var assert = require('assert');
 var Request = require('tedious').Request;
 var ConnectionPool = require('../lib/connection-pool');
@@ -375,3 +376,47 @@ describe('ConnectionPool', function () {
         }, 4);
     });
 });
+
+describe('Load Test', function() {
+    var statistics = require('simple-statistics');
+
+    it('Memory Leak Detection', function(done) {
+        this.timeout(60000);
+        if (!global.gc)
+            throw new Error('must run nodejs with --expose-gc');
+        var count = 0;
+        var mem = [];
+        var groupCount = 20;
+        var poolSize = 1000;
+        var max = poolSize * groupCount;
+
+        var pool = new ConnectionPool({ max: poolSize, min: poolSize, retryDelay: 1}, {
+            userName: 'testLogin',
+            password: 'wrongPassword',
+            server: 'localhost'
+        });
+
+        pool.on('error', function() {
+            if ((++count % poolSize) !== 0)
+                return;
+
+            global.gc();
+
+            var heapUsedKB = Math.round(process.memoryUsage().heapUsed / 1024);
+            mem.push([count, heapUsedKB]);
+
+            console.log(count + ': ' + heapUsedKB + 'KB');
+
+            if (count === max) {
+                var data = statistics.linearRegression(mem);
+                //console.log(data.m);
+                if (data.m >= 0.025)
+                    done(new Error('Memory leak not detected.'));
+                else
+                    done();
+
+                pool.drain();
+            }
+        });
+    });
+});