test & example added

Author: shresthaoshan

examples/request_finished.js

@@ -0,0 +1,46 @@
+const { Server } = require('proxy-chain');
+const http = require('http');
+const request = require('request');
+
+(async () => {
+    // Create a target server
+    const targetServer = http.createServer((req, res) => {
+        res.writeHead(200, { 'Content-Type': 'text/plain' });
+        res.end('Hello World!');
+    });
+    await new Promise((resolve) => targetServer.listen(0, resolve));
+    const targetPort = targetServer.address().port;
+
+    // Create a proxy server
+    const server = new Server({
+        port: 0,
+        verbose: true,
+    });
+
+    server.on('requestFinished', ({ id, connectionId, request }) => {
+        console.log(`Request finished: { id: ${id}, connectionId: ${connectionId}, method: ${request.method}, url: ${request.url} }`);
+    });
+
+    await server.listen();
+    const proxyPort = server.port;
+
+    console.log(`Proxy server listening on port ${proxyPort}`);
+    console.log(`Target server listening on port ${targetPort}`);
+
+    // Make a request through the proxy
+    await new Promise((resolve, reject) => {
+        request({
+            url: `http://127.0.0.1:${targetPort}`,
+            proxy: `http://127.0.0.1:${proxyPort}`,
+        }, (error, response, body) => {
+            if (error) return reject(error);
+            console.log(`Response body: ${body}`);
+            resolve();
+        });
+    });
+
+    // Close servers
+    await server.close(true);
+    await new Promise((resolve) => targetServer.close(resolve));
+    console.log('Servers closed.');
+})(); 

test/server.js

@@ -180,6 +180,7 @@ const createTestSuite = ({
         const mainProxyServerConnectionIds = [];
         const mainProxyServerConnectionsClosed = [];
         const mainProxyServerConnectionId2Stats = {};
+        const mainProxyServerRequestsFinished = [];
 
         let upstreamProxyHostname = '127.0.0.1';
 
@@ -462,6 +463,10 @@ const createTestSuite = ({
                         mainProxyServerConnectionId2Stats[connectionId] = stats;
                     });
 
+                    mainProxyServer.on('requestFinished', ({ id, connectionId }) => {
+                        mainProxyServerRequestsFinished.push({ id, connectionId });
+                    });
+
                     return mainProxyServer.listen();
                 }
             })
@@ -894,6 +899,19 @@ const createTestSuite = ({
             });
         }
 
+        if (useMainProxy) {
+            _it('should emit requestFinished event', () => {
+                const opts = getRequestOpts('/hello-world');
+                opts.method = 'GET';
+                return requestPromised(opts)
+                    .then((response) => {
+                        expect(response.body).to.eql('Hello world!');
+                        expect(response.statusCode).to.eql(200);
+                        expect(mainProxyServerRequestsFinished.length).to.be.above(0);
+                    });
+            });
+        }
+
         if (!useSsl && mainProxyAuth && mainProxyAuth.username && mainProxyAuth.password) {
             it('handles GET request using puppeteer with invalid credentials', async () => {
                 const phantomUrl = `${useSsl ? 'https' : 'http'}://${LOCALHOST_TEST}:${targetServerPort}/hello-world`;
@@ -1259,6 +1277,7 @@ const createTestSuite = ({
                         expect(mainProxyServer.getConnectionIds()).to.be.deep.eql([]);
                     }
                     expect(mainProxyServerConnectionIds).to.be.deep.eql([]);
+                    mainProxyServerRequestsFinished.splice(0, mainProxyServerRequestsFinished.length);
 
                     const closedSomeConnectionsTwice = mainProxyServerConnectionsClosed
                         .reduce((duplicateConnections, id, index) => {