1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
'use strict';
const common = require('../common');
const assert = require('assert');
const cluster = require('cluster');
const totalWorkers = 2;
// Cluster setup
if (cluster.isWorker) {
const http = require('http');
http.Server(() => {
}).listen(common.PORT, '127.0.0.1');
} else if (process.argv[2] === 'cluster') {
// Send PID to testcase process
let forkNum = 0;
cluster.on('fork', common.mustCall(function forkEvent(worker) {
// Send PID
process.send({
cmd: 'worker',
workerPID: worker.process.pid
});
// Stop listening when done
if (++forkNum === totalWorkers) {
cluster.removeListener('fork', forkEvent);
}
}));
// Throw accidental error when all workers are listening
let listeningNum = 0;
cluster.on('listening', common.mustCall(function listeningEvent() {
// When all workers are listening
if (++listeningNum === totalWorkers) {
// Stop listening
cluster.removeListener('listening', listeningEvent);
// Throw accidental error
process.nextTick(() => {
throw new Error('accidental error');
});
}
}));
// Startup a basic cluster
cluster.fork();
cluster.fork();
} else {
// This is the testcase
const fork = require('child_process').fork;
let masterExited = false;
let workersExited = false;
// List all workers
const workers = [];
// Spawn a cluster process
const master = fork(process.argv[1], ['cluster'], {silent: true});
// Handle messages from the cluster
master.on('message', common.mustCall((data) => {
// Add worker pid to list and progress tracker
if (data.cmd === 'worker') {
workers.push(data.workerPID);
}
}, totalWorkers));
// When cluster is dead
master.on('exit', common.mustCall((code) => {
// Check that the cluster died accidentally (non-zero exit code)
masterExited = !!code;
const pollWorkers = function() {
// When master is dead all workers should be dead too
let alive = false;
workers.forEach((pid) => alive = common.isAlive(pid));
if (alive) {
setTimeout(pollWorkers, 50);
} else {
workersExited = true;
}
};
// Loop indefinitely until worker exit
pollWorkers();
}));
process.once('exit', () => {
assert(masterExited,
'The master did not die after an error was thrown');
assert(workersExited,
'The workers did not die after an error in the master');
});
}
|