feat: add ability to reject connection with ECONNREFUSED

index.js

@@ -108,11 +108,22 @@ Mitm.prototype.connect = function connect(orig, Socket, opts, done) {
 
   this.emit("connection", server, 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".
-  setTimeout(client.emit.bind(client, "connect"))
-  setTimeout(server.emit.bind(server, "connect"))
+  if (client.rejected) {
+    const connectionRefusedError = new Error()
+    connectionRefusedError.address = "127.0.0.1"
+    connectionRefusedError.code = "ECONNREFUSED"
+    connectionRefusedError.errno = -4078
+    connectionRefusedError.port = 9090
+    connectionRefusedError.syscall = "connect"
+    setTimeout(client.emit.bind(client, "error", connectionRefusedError))
+    setTimeout(client.emit.bind(client, "close", true))
+  } else {
+    // 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".
+    setTimeout(client.emit.bind(client, "connect"))
+    setTimeout(server.emit.bind(server, "connect"))
+  }
 
   return client
 }

lib/socket.js

@@ -10,3 +10,7 @@ Socket.prototype = Object.create(Net.Socket.prototype, {
 Socket.prototype.bypass = function() {
   this.bypassed = true
 }
+
+Socket.prototype.reject = function() {
+  this.rejected = true
+}