mirror of https://github.com/lukechilds/node.git
Browse Source
It's important for people who monkey-patch `Socket.prototype.connect` that it's called by `net.connect` since it's not possible to monkey-patch `net.connect` directly (as the `connect` function is called directly by other parts of `lib/net.js` instead of calling the `exports.connect` function). Among the actors who monkey-patch `Socket.prototype.connect` are most APM vendors, the async-listener module and the continuation-local-storage module. Related: - https://github.com/nodejs/node/pull/12342 - https://github.com/nodejs/node/pull/12852 PR-URL: https://github.com/nodejs/node/pull/12861 Reviewed-By: Sam Roberts <vieuxtech@gmail.com> Reviewed-By: Luca Maraschi <luca.maraschi@gmail.com> Reviewed-By: Colin Ihrig <cjihrig@gmail.com> Reviewed-By: Anna Henningsen <anna@addaleax.net> Reviewed-By: James M Snell <jasnell@gmail.com> Reviewed-By: Jan Krems <jan.krems@gmail.com>v6
Thomas Watson
8 years ago
committed by
Sam Roberts
2 changed files with 57 additions and 14 deletions
@ -0,0 +1,39 @@ |
|||
'use strict'; |
|||
const common = require('../common'); |
|||
|
|||
// This test checks that calling `net.connect` internally calls
|
|||
// `Socket.prototype.connect`.
|
|||
//
|
|||
// This is important for people who monkey-patch `Socket.prototype.connect`
|
|||
// since it's not possible to monkey-patch `net.connect` directly (as the core
|
|||
// `connect` function is called internally in Node instead of calling the
|
|||
// `exports.connect` function).
|
|||
//
|
|||
// Monkey-patching of `Socket.prototype.connect` is done by - among others -
|
|||
// most APM vendors, the async-listener module and the
|
|||
// continuation-local-storage module.
|
|||
//
|
|||
// Related:
|
|||
// - https://github.com/nodejs/node/pull/12342
|
|||
// - https://github.com/nodejs/node/pull/12852
|
|||
|
|||
const net = require('net'); |
|||
const Socket = net.Socket; |
|||
|
|||
// Monkey patch Socket.prototype.connect to check that it's called.
|
|||
const orig = Socket.prototype.connect; |
|||
Socket.prototype.connect = common.mustCall(function() { |
|||
return orig.apply(this, arguments); |
|||
}); |
|||
|
|||
const server = net.createServer(); |
|||
|
|||
server.listen(common.mustCall(function() { |
|||
const port = server.address().port; |
|||
const client = net.connect({port}, common.mustCall(function() { |
|||
client.end(); |
|||
})); |
|||
client.on('end', common.mustCall(function() { |
|||
server.close(); |
|||
})); |
|||
})); |
Loading…
Reference in new issue