Latest Code

node_modules/strong-cluster-connect-store/test/cluster-store.js

@@ -0,0 +1,155 @@
+var cluster = require('cluster');
+var http = require('http');
+var expect = require('chai').expect;
+var express = require('express');
+var bodyParser = require('body-parser');
+var cookieParser = require('cookie-parser');
+var session = require('express-session');
+var request = require('request');
+var async = require('async');
+var ClusterStore = require('..')(session);
+
+var workerUrl;
+
+// verify we can call setup without connect in master and workers
+require('..').setup();
+
+if (cluster.isWorker) {
+  startConnectServer();
+  return;
+}
+
+describe('clustered connect server', function() {
+  before(setupWorkers);
+  after(stopWorkers);
+
+  var KEY = 'a-key';
+  var PAYLOAD = 'a-value';
+
+  // NOTE We assume that the cluster does a perfect round-robin
+  // distribution of requests among the workers
+
+  it('shares sessions between workers', function(done) {
+    async.series(
+      [
+        save,
+        load
+      ],
+      function(err, results) {
+        if (err) {
+          return done(err);
+        }
+        expect(results.pop().value).to.equal(PAYLOAD);
+        done();
+      }
+    );
+  });
+
+  it('destroys a session shared between workers', function(done) {
+    async.series(
+      [
+        save,
+        destroy,
+        load
+      ],
+      function(err, results) {
+        if (err) {
+          return done(err);
+        }
+        expect(results.pop().value).to.equal(undefined);
+        done();
+      }
+    );
+  });
+
+  function save(next) {
+    sendCommand({ cmd: 'set', key: KEY, value: PAYLOAD }, next);
+  }
+
+  function destroy(next) {
+    sendCommand({ cmd: 'del', key: KEY }, next);
+  }
+
+  function load(next) {
+    sendCommand({ cmd: 'get', key: KEY }, next);
+  }
+});
+
+function sendCommand(command, cb) {
+  request(
+    {
+      url: workerUrl,
+      method: 'POST',
+      json: command
+    },
+    function(err, res, body) {
+      if (err) {
+        return cb(err);
+      }
+      cb(null, body);
+    }
+  );
+}
+
+var WORKER_COUNT = 2;
+
+function getNumberOfWorkers() {
+  return Object.keys(cluster.workers).length;
+}
+
+function setupWorkers(done) {
+  if (getNumberOfWorkers() > 0) {
+    var msg = 'Cannot setup workers: there are already other workers running.';
+    return done(new Error(msg));
+  }
+
+  cluster.setupMaster({ exec: __filename });
+  ClusterStore.setup();
+
+  var workersListening = 0;
+  cluster.on('listening', function(w, addr) {
+    if (!workerUrl) workerUrl = 'http://localhost:' + addr.port;
+
+    workersListening++;
+    if (workersListening == WORKER_COUNT) {
+      done();
+    }
+  });
+
+  for (var i = 0; i < WORKER_COUNT; i++) {
+    cluster.fork();
+  }
+}
+
+function stopWorkers(done) {
+  cluster.disconnect(done);
+}
+
+function startConnectServer() {
+  var PORT = 0; // Let the OS pick any available port
+  var app = express()
+    .use(cookieParser())
+    .use(session({ store: new ClusterStore(), secret: 'a-secret', key: 'sid' }))
+    .use(bodyParser.json())
+    .use(requestHandler);
+
+  var server = http.createServer(app).listen(PORT);
+
+  function requestHandler(req, res) {
+    var result = {};
+    switch (req.body.cmd) {
+      case 'set':
+        req.session[req.body.key] = req.body.value;
+        break;
+      case 'get':
+        result.value = req.session[req.body.key];
+        break;
+      case 'del':
+        req.session.destroy();
+        break;
+    }
+
+    res.setHeader('Content-Type', 'text/json');
+    res.end(JSON.stringify(result));
+  }
+}