Add test for isExtConnCheck

Author: ruihildt

src/helpers/socksProxy/socksProxy.test.ts

@@ -1,4 +1,5 @@
-import { isLocalOrReservedIP } from '@/helpers/socksProxy/socksProxy';
+import { isExtConnCheck, isLocalOrReservedIP } from '@/helpers/socksProxy/socksProxy';
+import { RequestDetails } from './socksProxy.types';
 
 describe('isLocalOrReservedIP', () => {
   it('should return true for localhost', () => {
@@ -34,3 +35,87 @@ describe('isLocalOrReservedIP', () => {
     expect(isLocalOrReservedIP('')).toBeFalsy();
   });
 });
+
+describe('isExtConnCheck', () => {
+  const baseDetails: RequestDetails = {
+    requestId: '5979',
+    url: '',
+    method: 'GET',
+    type: 'xmlhttprequest',
+    fromCache: false,
+    incognito: false,
+    thirdParty: false,
+    originUrl: '',
+    documentUrl: '',
+    frameId: 0,
+    parentFrameId: -1,
+    frameAncestors: [],
+    timeStamp: 1740215207080,
+    tabId: 4,
+    cookieStoreId: 'firefox-default',
+    urlClassification: {
+      firstParty: [],
+      thirdParty: [],
+    },
+  };
+
+  it('should return true for extension IPv4 connection check', () => {
+    const details: RequestDetails = {
+      ...baseDetails,
+      url: 'https://ipv4.am.i.mullvad.net/json',
+      originUrl: 'moz-extension://8ad8e256-a9a0-4017-b302-1345ac426553/dist/options/index.html',
+      documentUrl: 'moz-extension://8ad8e256-a9a0-4017-b302-1345ac426553/dist/options/index.html',
+    };
+    expect(isExtConnCheck(details)).toBeTruthy();
+  });
+
+  it('should return true for extension IPv6 connection check', () => {
+    const details: RequestDetails = {
+      ...baseDetails,
+      url: 'https://ipv6.am.i.mullvad.net/json',
+      originUrl: 'moz-extension://8ad8e256-a9a0-4017-b302-1345ac426553/dist/options/index.html',
+      documentUrl: 'moz-extension://8ad8e256-a9a0-4017-b302-1345ac426553/dist/options/index.html',
+    };
+    expect(isExtConnCheck(details)).toBeTruthy();
+  });
+
+  it('should return true for extension DNS check', () => {
+    const details: RequestDetails = {
+      ...baseDetails,
+      url: 'https://c30d3da2-fc4f-4732-86cc-f233e1692eac.dnsleak.am.i.mullvad.net/',
+      originUrl: 'moz-extension://8ad8e256-a9a0-4017-b302-1345ac426553/dist/options/index.html',
+      documentUrl: 'moz-extension://8ad8e256-a9a0-4017-b302-1345ac426553/dist/options/index.html',
+    };
+    expect(isExtConnCheck(details)).toBeTruthy();
+  });
+
+  it('should return false for DNS check with the wrong base domain', () => {
+    const details: RequestDetails = {
+      ...baseDetails,
+      url: 'https://example.com#am.i.mullvad.net/',
+      originUrl: 'moz-extension://8ad8e256-a9a0-4017-b302-1345ac426553/dist/options/index.html',
+      documentUrl: 'moz-extension://8ad8e256-a9a0-4017-b302-1345ac426553/dist/options/index.html',
+    };
+    expect(isExtConnCheck(details)).toBeFalsy();
+  });
+
+  it('should return false for non-extension requests', () => {
+    const details: RequestDetails = {
+      ...baseDetails,
+      url: 'https://ipv4.am.i.mullvad.net/json',
+      documentUrl: 'https://example.com',
+      originUrl: 'https://example.com',
+    };
+    expect(isExtConnCheck(details)).toBeFalsy();
+  });
+
+  it('should return false for extension requests to other URLs', () => {
+    const details: RequestDetails = {
+      ...baseDetails,
+      url: 'https://example.com',
+      originUrl: 'moz-extension://8ad8e256-a9a0-4017-b302-1345ac426553/dist/options/index.html',
+      documentUrl: 'moz-extension://8ad8e256-a9a0-4017-b302-1345ac426553/dist/options/index.html',
+    };
+    expect(isExtConnCheck(details)).toBeFalsy();
+  });
+});

src/helpers/socksProxy/socksProxy.ts

@@ -131,7 +131,7 @@ const getCurrentHost = (details: RequestDetails) => {
   return new URL(details.url).hostname;
 };
 
-const isExtConnCheck = (details: RequestDetails): boolean => {
+export const isExtConnCheck = (details: RequestDetails): boolean => {
   const isExtensionRequest = Boolean(details.documentUrl?.startsWith('moz-extension://'));
   const isConnCheck =
     details.url === 'https://ipv4.am.i.mullvad.net/json' ||