test: add AsyncLocalStorage tests using udp, tcp and tls sockets

Fixes: #40693
Signed-off-by: Darshan Sen <darshan.sen@postman.com>
PR-URL: #40741
Reviewed-By: Stephen Belanger <admin@stephenbelanger.com>
Reviewed-By: Gerhard Stöbich <deb2001-github@yahoo.de>
Reviewed-By: Benjamin Gruenbaum <benjamingr@gmail.com>

Author: RaisinTen

test/async-hooks/test-async-local-storage-dgram.js

@@ -0,0 +1,26 @@
+'use strict';
+
+require('../common');
+
+// Regression tests for https://github.com/nodejs/node/issues/40693
+
+constassert=require('assert');
+constdgram=require('dgram');
+const{ AsyncLocalStorage }=require('async_hooks');
+
+dgram.createSocket('udp4')
+.on('message',function(msg,rinfo){this.send(msg,rinfo.port);})
+.on('listening',function(){
+constasyncLocalStorage=newAsyncLocalStorage();
+conststore={val: 'abcd'};
+asyncLocalStorage.run(store,()=>{
+constclient=dgram.createSocket('udp4');
+client.on('message',(msg,rinfo)=>{
+assert.deepStrictEqual(asyncLocalStorage.getStore(),store);
+client.close();
+this.close();
+});
+client.send('Hello, world!',this.address().port);
+});
+})
+.bind(0);

test/async-hooks/test-async-local-storage-socket.js

@@ -0,0 +1,27 @@
+'use strict';
+
+require('../common');
+
+// Regression tests for https://github.com/nodejs/node/issues/40693
+
+constassert=require('assert');
+constnet=require('net');
+const{ AsyncLocalStorage }=require('async_hooks');
+
+net
+.createServer((socket)=>{
+socket.write('Hello, world!');
+socket.pipe(socket);
+})
+.listen(0,function(){
+constasyncLocalStorage=newAsyncLocalStorage();
+conststore={val: 'abcd'};
+asyncLocalStorage.run(store,()=>{
+constclient=net.connect({port: this.address().port});
+client.on('data',()=>{
+assert.deepStrictEqual(asyncLocalStorage.getStore(),store);
+client.end();
+this.close();
+});
+});
+});

test/async-hooks/test-async-local-storage-tlssocket.js

@@ -0,0 +1,36 @@
+'use strict';
+
+constcommon=require('../common');
+if(!common.hasCrypto)
+common.skip('missing crypto');
+
+// Regression tests for https://github.com/nodejs/node/issues/40693
+
+constassert=require('assert');
+constfixtures=require('../common/fixtures');
+consttls=require('tls');
+const{ AsyncLocalStorage }=require('async_hooks');
+
+constoptions={
+cert: fixtures.readKey('rsa_cert.crt'),
+key: fixtures.readKey('rsa_private.pem'),
+rejectUnauthorized: false
+};
+
+tls
+.createServer(options,(socket)=>{
+socket.write('Hello, world!');
+socket.pipe(socket);
+})
+.listen(0,function(){
+constasyncLocalStorage=newAsyncLocalStorage();
+conststore={val: 'abcd'};
+asyncLocalStorage.run(store,()=>{
+constclient=tls.connect({port: this.address().port, ...options});
+client.on('data',()=>{
+assert.deepStrictEqual(asyncLocalStorage.getStore(),store);
+client.end();
+this.close();
+});
+});
+});