Expose connect options and the client socket via a _mitm property.

Fixes #14.

Author: papandreou

index.js

@@ -79,6 +79,11 @@ Mitm.prototype.connect = function connect(orig, Socket, opts, done) {
   var server = client.server = new Socket({handle: sockets[1]})
   this.emit("connection", server, opts)
 
+  // Make the client socket and the connect options available via a "private"
+  // property of the server socket so they can be paired up with requests in
+  // a reliable way:
+  server._mitm = {client: client, opts: opts};
+
   // Ensure connect is emitted in next ticks, otherwise it would be impossible
   // to listen to it after calling Net.connect or listening to it after the
   // ClientRequest emits "socket".

test/index_test.js

@@ -429,6 +429,25 @@ describe("Mitm", function() {
         })
       })
 
+      it("must provide the client socket and options via _mitm", function(done) {
+        var clientSocket;
+        var connectOpts;
+        this.mitm.on("connect", function (socket, opts) {
+          clientSocket = socket
+          connectOpts = opts
+          clientSocket.must.be.an.instanceof(Object)
+          connectOpts.host.must.equal("foo")
+        }).on("request", function(req, res) {
+          req.connection._mitm.must.be.an.instanceof(Object)
+          req.connection._mitm.opts.must.be(connectOpts)
+          req.connection._mitm.client.must.be(clientSocket)
+          done()
+        })
+
+        var client = request({host: "foo"})
+        client.end()
+      })
+
       it("must emit request on Mitm after multiple requests", function(done) {
         request({host: "foo"}).end()
         request({host: "foo"}).end()