// Copyright Joyent, Inc. and other Node contributors. // // Permission is hereby granted, free of charge, to any person obtaining a // copy of this software and associated documentation files (the // "Software"), to deal in the Software without restriction, including // without limitation the rights to use, copy, modify, merge, publish, // distribute, sublicense, and/or sell copies of the Software, and to permit // persons to whom the Software is furnished to do so, subject to the // following conditions: // // The above copyright notice and this permission notice shall be included // in all copies or substantial portions of the Software. // // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS // OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN // NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, // DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR // OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE // USE OR OTHER DEALINGS IN THE SOFTWARE. var common = require('../common'); var assert = require('assert'); var cluster = require('cluster'); var net = require('net'); if (cluster.isWorker) { net.createServer(function(socket) { socket.end('echo'); }).listen(common.PORT, '127.0.0.1'); net.createServer(function(socket) { socket.end('echo'); }).listen(common.PORT + 1, '127.0.0.1'); } else if (cluster.isMaster) { // test a single TCP server var testConnection = function(port, cb) { var socket = net.connect(port, '127.0.0.1', function() { // buffer result var result = ''; socket.on('data', function(chunk) { result += chunk; }); // check result socket.on('end', function() { cb(result === 'echo'); }); }); }; // test both servers created in the cluster var testCluster = function(cb) { var servers = 2; var done = 0; for (var i = 0, l = servers; i < l; i++) { testConnection(common.PORT + i, function(success) { assert.ok(success); done += 1; if (done === servers) { cb(); } }); } }; // start two workers and execute callback when both is listening var startCluster = function(cb) { var workers = 8; var online = 0; for (var i = 0, l = workers; i < l; i++) { var worker = cluster.fork(); worker.on('listening', function() { online += 1; if (online === workers) { cb(); } }); } }; var results = { start: 0, test: 0, disconnect: 0 }; var test = function(again) { //1. start cluster startCluster(function() { results.start += 1; //2. test cluster testCluster(function() { results.test += 1; //3. disconnect cluster cluster.disconnect(function() { results.disconnect += 1; // run test again to confirm cleanup if (again) { test(); } }); }); }); }; test(true); process.once('exit', function() { assert.equal(results.start, 2); assert.equal(results.test, 2); assert.equal(results.disconnect, 2); }); }