Add execute-js load test

Author: hwrdtm

e2e/artillery/configs/execute.yml

@@ -0,0 +1,24 @@
+config:
+  target: "dummy"
+  phases:
+    # Over 60s, ramp up to creating 50 vusers per second
+    - duration: 60
+      arrivalRate: 5
+      rampTo: 50
+      name: "Ramp Up"
+    # Over 300s, create 50 vusers per second
+    - duration: 300
+      arrivalRate: 50
+      name: "Sustained Encrypt & Decrypt"
+    # Over 60s, ramp down to creating 5 vusers per second
+    - duration: 60
+      arrivalRate: 10
+      name: "Ramp Down"
+  processor: "../src/processors/multi-endpoints.ts"
+
+scenarios:
+  - name: "Execute JS Stress Test"
+    weight: 100
+    flow:
+      - function: "runExecuteJSTest"
+      - think: 0.1 

e2e/artillery/src/processors/multi-endpoints.ts

@@ -18,6 +18,28 @@ const PkpSignResultSchema = z.object({
   sigType: z.string().min(1, 'Signature type cannot be empty'),
 });
 
+// Execute JS Result Schema
+const ExecuteJsResultSchema = z.object({
+  success: z.boolean(),
+  signatures: z.record(
+    z.string(),
+    z.object({
+      signature: z.string().regex(/^0x[a-fA-F0-9]+$/, 'Invalid hex signature'),
+      verifyingKey: z
+        .string()
+        .regex(/^0x[a-fA-F0-9]+$/, 'Invalid hex verifying key'),
+      signedData: z
+        .string()
+        .regex(/^0x[a-fA-F0-9]+$/, 'Invalid hex signed data'),
+      recoveryId: z.number().int().min(0).max(3, 'Recovery ID must be 0-3'),
+      publicKey: z.string().regex(/^0x[a-fA-F0-9]+$/, 'Invalid hex public key'),
+      sigType: z.string().min(1, 'Signature type cannot be empty'),
+    })
+  ),
+  response: z.string(),
+  logs: z.string(),
+});
+
 // Global variables to cache expensive operations
 let litClient: LitClientType;
 let authManager: any = null;
@@ -212,6 +234,73 @@ export async function runEncryptDecryptTest() {
   }
 }
 
+// test '/web/execute/v2' endpoint
+export async function runExecuteJSTest() {
+  const startTime = Date.now();
+
+  try {
+    // 1. Initialise shared resources (only happens once)
+    await initialiseSharedResources();
+
+    // 2. Read state
+    const state = await StateManager.readFile();
+
+    // Create auth context
+    const authContext = await createAuthContextFromState();
+
+    // Perform executeJs operation
+    const litActionCode = `
+    (async () => {
+      const { sigName, toSign, publicKey } = jsParams;
+      const { keccak256, arrayify } = ethers.utils;
+      
+      const toSignBytes = new TextEncoder().encode(toSign);
+      const toSignBytes32 = keccak256(toSignBytes);
+      const toSignBytes32Array = arrayify(toSignBytes32);
+      
+      const sigShare = await Lit.Actions.signEcdsa({
+        toSign: toSignBytes32Array,
+        publicKey,
+        sigName,
+      });  
+    })();`;
+
+    const result = await litClient.executeJs({
+      code: litActionCode,
+      authContext,
+      jsParams: {
+        message: 'Test message from e2e executeJs',
+        sigName: 'e2e-test-sig',
+        toSign: 'Test message from e2e executeJs',
+        publicKey: state.masterAccount.pkp.publicKey,
+      },
+    });
+
+    // Validate the result using Zod schema
+    const validatedResult = ExecuteJsResultSchema.parse(result);
+
+    const endTime = Date.now();
+    const duration = endTime - startTime;
+
+    console.log(`✅ executeJs successful in ${duration}ms`);
+    console.log('✅ executeJs result:', validatedResult);
+
+    // For Artillery, just return - no need to call next()
+    return;
+  } catch (error) {
+    const endTime = Date.now();
+    const duration = endTime - startTime;
+
+    console.error(
+      `❌ executeJs failed in ${duration}ms:`,
+      error instanceof Error ? error.message : String(error)
+    );
+
+    // Throw the error to let Artillery handle it
+    throw error;
+  }
+}
+
 // test '/web/sign_session_key' endpoint
 export async function runSignSessionKeyTest() {
   // ❗️ IT'S IMPORTANT TO SET THIS TO FALSE FOR TESTING

package.json

@@ -29,6 +29,7 @@
     "artillery:balance-status": "LOG_LEVEL=silent bun run ./e2e/artillery/src/balance-status.ts",
     "artillery:pkp-sign": "DEBUG_HTTP=true LOG_LEVEL=silent dotenvx run --env-file=.env -- sh -c 'artillery run ./e2e/artillery/configs/pkp-sign.yml ${ARTILLERY_KEY:+--record --key $ARTILLERY_KEY}'",
     "artillery:encrypt-decrypt": "DEBUG_HTTP=true LOG_LEVEL=silent dotenvx run --env-file=.env -- sh -c 'artillery run ./e2e/artillery/configs/encrypt-decrypt.yml ${ARTILLERY_KEY:+--record --key $ARTILLERY_KEY}'",
+    "artillery:execute-js": "DEBUG_HTTP=true LOG_LEVEL=silent dotenvx run --env-file=.env -- sh -c 'artillery run ./e2e/artillery/configs/execute.yml ${ARTILLERY_KEY:+--record --key $ARTILLERY_KEY}'",
     "artillery:sign-session-key": "DEBUG_HTTP=true LOG_LEVEL=silent dotenvx run --env-file=.env -- sh -c 'artillery run ./e2e/artillery/configs/sign-session-key.yml ${ARTILLERY_KEY:+--record --key $ARTILLERY_KEY}'"
   },
   "private": true,