Skip to content

Commit

Permalink
stream: fix eventNames() to not return not defined events
Browse files Browse the repository at this point in the history
PR-URL: #51331
Reviewed-By: Benjamin Gruenbaum <benjamingr@gmail.com>
Reviewed-By: Stephen Belanger <admin@stephenbelanger.com>
Reviewed-By: Luigi Pinca <luigipinca@gmail.com>
  • Loading branch information
IlyasShabi authored and richardlau committed Mar 25, 2024
1 parent 306c1d3 commit 7931c3b
Show file tree
Hide file tree
Showing 2 changed files with 53 additions and 0 deletions.
11 changes: 11 additions & 0 deletions lib/internal/streams/legacy.js
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
const {
ArrayIsArray,
ObjectSetPrototypeOf,
ReflectOwnKeys,
} = primordials;

const EE = require('events');
Expand Down Expand Up @@ -93,6 +94,16 @@ Stream.prototype.pipe = function(dest, options) {
return dest;
};

Stream.prototype.eventNames = function eventNames() {
const names = [];
for (const key of ReflectOwnKeys(this._events)) {
if (typeof this._events[key] === 'function' || (ArrayIsArray(this._events[key]) && this._events[key].length > 0)) {
names.push(key);
}
}
return names;
};

function prependListener(emitter, event, fn) {
// Sadly this is not cacheable as some libraries bundle their own
// event emitter implementation with them.
Expand Down
42 changes: 42 additions & 0 deletions test/parallel/test-stream-event-names.js
Original file line number Diff line number Diff line change
@@ -0,0 +1,42 @@
'use strict';

require('../common');
const assert = require('assert');
const { Readable, Writable, Duplex } = require('stream');

{
const stream = new Readable();
assert.strictEqual(stream.eventNames().length, 0);
}

{
const stream = new Readable();
stream.on('foo', () => {});
stream.on('data', () => {});
stream.on('error', () => {});
assert.deepStrictEqual(stream.eventNames(), ['error', 'data', 'foo']);
}

{
const stream = new Writable();
assert.strictEqual(stream.eventNames().length, 0);
}

{
const stream = new Writable();
stream.on('foo', () => {});
stream.on('drain', () => {});
stream.on('prefinish', () => {});
assert.deepStrictEqual(stream.eventNames(), ['prefinish', 'drain', 'foo']);
}
{
const stream = new Duplex();
assert.strictEqual(stream.eventNames().length, 0);
}

{
const stream = new Duplex();
stream.on('foo', () => {});
stream.on('finish', () => {});
assert.deepStrictEqual(stream.eventNames(), ['finish', 'foo']);
}

0 comments on commit 7931c3b

Please sign in to comment.