test: add custom function test

Author: sylingd

.github/workflows/dev.yml

@@ -83,10 +83,10 @@ jobs:
     - name: Install Edge
       id: install-edge
       run: |
-        wget -q https://packages.microsoft.com/keys/microsoft.asc -O- | sudo apt-key add -
+        sudo wget -qO /etc/apt/trusted.gpg.d/microsoft.gpg https://packages.microsoft.com/keys/microsoft.asc
         sudo add-apt-repository "deb [arch=amd64] https://packages.microsoft.com/repos/edge stable main"
-        sudo apt update
-        sudo apt install -y microsoft-edge-stable=138.0.3351.109-1
+        sudo apt-get update
+        sudo apt-get install -y microsoft-edge-stable=138.0.3351.109-1
         microsoft-edge --version
         EDGE_PATH=$(which microsoft-edge)
         echo "edge-path=$EDGE_PATH" >> $GITHUB_OUTPUT

tests/e2e/browser.spec.mjs

@@ -132,3 +132,27 @@ test.each([['edge_v2'], ['chrome_v3'], ['firefox_v2'], ['firefox_v3']])(
       await remove();
     }),
 );
+
+test.each([['edge_v2'], ['firefox_v2']])(
+  '[%s] - Custom Function',
+  async browserKey =>
+    runTest(browserKey, async browser => {
+      const key = String(Math.random()).replace('.', '');
+
+      const { remove } = await saveRule(browser.popup, {
+        name: 'test custom function',
+        ruleType: 'modifySendHeader',
+        pattern: '^' + testServer,
+        matchType: 'regexp',
+        isFunction: true,
+        enable: true,
+        code: `val.push({ "name": "X-Custom-Header", "value": "${key}" })`,
+      });
+
+      const header = await getHeader(browser.browser);
+
+      expect(header['X_CUSTOM_HEADER']).toBe(key);
+
+      await remove();
+    }),
+);