Automatically intercept requests by Undici (v4.8.2+)

Previously versions of Undici don't support any kind of agent, which
makes interception much more difficult. 4.8.2 includes an agent and
a mechanism to configure it globally, making everything much easier.

Author: pimterry

overrides/js/prepend-node.js

@@ -80,6 +80,18 @@ wrapModule('superagent', function wrapSuperagent (loadedModule) {
     };
 });
 
+wrapModule('undici', function wrapUndici (loadedModule) {
+    const ProxyAgent = loadedModule.ProxyAgent;
+    const setGlobalDispatcher = loadedModule.setGlobalDispatcher;
+
+    // Old Undici release, which can't be intercepted:
+    if (!ProxyAgent || !setGlobalDispatcher) return;
+
+    setGlobalDispatcher(
+        new ProxyAgent(process.env.HTTP_PROXY)
+    );
+});
+
 wrapModule('stripe', function wrapStripe (loadedModule) {
     if (loadedModule.INTERCEPTED_BY_HTTPTOOLKIT) return;
 

package-lock.json

@@ -119,6 +119,7 @@
     "ts-loader": "^6.0.0",
     "ts-node": "^8.4.1",
     "typescript": "^4.4.3",
+    "undici": "^4.9.5",
     "unirest": "^0.6.0",
     "webpack": "^4.46.0",
     "webpack-cli": "^3.3.9",

package.json

@@ -1,6 +1,6 @@
 // Test built-ins against their specific protocol
-require('http').get('http://example.com/js/http');
-require('https').get('https://example.com/js/https');
+require('http').get('http://example.test/js/http');
+require('https').get('https://example.test/js/https');
 
 function sendRequestsTo(baseUrl) {
     require('request').get(baseUrl + '/request');
@@ -12,11 +12,12 @@ function sendRequestsTo(baseUrl) {
     require('unirest').get(baseUrl + '/unirest', () => {});
     require('reqwest')(baseUrl + '/reqwest');
     require('needle').get(baseUrl + '/needle', () => {});
+    require('undici').request(baseUrl + '/undici');
 }
 
 // Test all other libs against both protocols
-sendRequestsTo('http://example.com/js');
-sendRequestsTo('https://example.com/js');
+sendRequestsTo('http://example.test/js');
+sendRequestsTo('https://example.test/js');
 
 // Test libraries that need manual steps, and use their own URLs:
 require('stripe')('sk_test_hunter2').customers.list();

test/fixtures/terminal/js-test-script.js

@@ -79,7 +79,7 @@ describe('Existing terminal interceptor', function () {
         const { interceptor, server } = await interceptorSetup;
         const result = await interceptor.activate(server.port) as { port: number };
 
-        const mainRule = await server.get(/https?:\/\/example.com\/js\/.*/).thenReply(200);
+        const mainRule = await server.get(/https?:\/\/example.test\/js\/.*/).thenReply(200);
         const stripeRule = await server.get('https://api.stripe.com/v1/customers').thenJson(200, {});
 
         const scriptOutput = await execAsync(`
@@ -98,8 +98,8 @@ describe('Existing terminal interceptor', function () {
         ])).map(r => r.url.replace(':443', '').replace(':80', ''));
 
         // Built-in modules
-        expect(seenRequests).to.include('http://example.com/js/http');
-        expect(seenRequests).to.include('https://example.com/js/https');
+        expect(seenRequests).to.include('http://example.test/js/http');
+        expect(seenRequests).to.include('https://example.test/js/https');
 
         // http & https with lots of popular libraries
         ['http', 'https'].forEach((protocol) =>
@@ -112,9 +112,10 @@ describe('Existing terminal interceptor', function () {
                 'bent',
                 'unirest',
                 'reqwest',
-                'needle'
+                'needle',
+                'undici'
             ].forEach((library) =>
-                expect(seenRequests).to.include(`${protocol}://example.com/js/${library}`)
+                expect(seenRequests).to.include(`${protocol}://example.test/js/${library}`)
             )
         );
 

test/interceptors/existing-terminal.spec.ts

@@ -44,7 +44,7 @@ describe('Fresh terminal interceptor', function () {
 
             const { server, httpsConfig } = await interceptorSetup;
 
-            const mainRule = await server.get(/https?:\/\/example.com\/js\/.*/).thenReply(200);
+            const mainRule = await server.get(/https?:\/\/example.test\/js\/.*/).thenReply(200);
             const stripeRule = await server.get('https://api.stripe.com/v1/customers').thenJson(200, {});
 
             // Spawn node, as if it was run inside an intercepted terminal
@@ -64,8 +64,8 @@ describe('Fresh terminal interceptor', function () {
             ])).map(r => r.url.replace(':443', '').replace(':80', ''));
 
             // Built-in modules
-            expect(seenRequests).to.include('http://example.com/js/http');
-            expect(seenRequests).to.include('https://example.com/js/https');
+            expect(seenRequests).to.include('http://example.test/js/http');
+            expect(seenRequests).to.include('https://example.test/js/https');
 
             // http & https with lots of popular libraries
             ['http', 'https'].forEach((protocol) =>
@@ -78,9 +78,10 @@ describe('Fresh terminal interceptor', function () {
                     'bent',
                     'unirest',
                     'reqwest',
-                    'needle'
+                    'needle',
+                    'undici'
                 ].forEach((library) =>
-                    expect(seenRequests).to.include(`${protocol}://example.com/js/${library}`)
+                    expect(seenRequests).to.include(`${protocol}://example.test/js/${library}`)
                 )
             );