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