added unit tests

Author: avcribl

test/parallel/test-stream-finished-async-local-storage.js

@@ -0,0 +1,23 @@
+// Flags: --expose-internals
+'use strict';
+
+const common = require('../common');
+const { Readable, finished } = require('stream');
+const { AsyncLocalStorage } = require('async_hooks');
+const { strictEqual } = require('assert');
+const AsyncContextFrame = require('internal/async_context_frame');
+const internalAsyncHooks = require('internal/async_hooks');
+
+// This test verifies that ALS context is preserved when using stream.finished()
+
+const als = new AsyncLocalStorage();
+const readable = new Readable();
+
+als.run('test-context-1', () => {
+  finished(readable, common.mustCall(() => {
+    strictEqual(AsyncContextFrame.enabled || internalAsyncHooks.getHookArrays()[0].length > 0, true, 'test-context-1', 'One of AsyncContextFrame or async hooks criteria should be met');
+    strictEqual(als.getStore(), 'test-context-1', 'ALS context should be preserved');
+  }));
+});
+
+readable.destroy();

test/parallel/test-stream-finished-bindAsyncResource-path.js

@@ -0,0 +1,20 @@
+// Flags: --expose-internals
+'use strict';
+
+const common = require('../common');
+const { Readable, finished } = require('stream');
+const { createHook } = require('async_hooks');
+const { strictEqual } = require('assert');
+const internalAsyncHooks = require('internal/async_hooks');
+
+// This test verifies that when there are active async hooks, stream.finished() uses
+// the bindAsyncResource path
+
+createHook({}).enable();
+const readable = new Readable();
+
+finished(readable, common.mustCall(() => {
+    strictEqual(internalAsyncHooks.getHookArrays()[0].length > 0, true, 'Should have active user async hook');
+}));
+
+readable.destroy();

test/parallel/test-stream-finished-default-path.js

@@ -0,0 +1,19 @@
+// Flags: --expose-internals --no-async-context-frame
+'use strict';
+
+const common = require('../common');
+const { Readable, finished } = require('stream');
+const { strictEqual } = require('assert');
+const AsyncContextFrame = require('internal/async_context_frame');
+const internalAsyncHooks = require('internal/async_hooks');
+
+// This test verifies that when there are no active async hooks, stream.finished() uses the default callback path
+
+const readable = new Readable();
+
+finished(readable, common.mustCall(() => {
+    strictEqual(internalAsyncHooks.getHookArrays()[0].length === 0, true, 'Should not have active user async hook');
+    strictEqual(AsyncContextFrame.current() || internalAsyncHooks.getHookArrays()[0].length > 0, false, 'Default callback path should be used');
+}));
+
+readable.destroy();