Add encrypt-decrypt load test

Author: hwrdtm

e2e/artillery/configs/encrypt-decrypt.yml

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

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

@@ -4,6 +4,7 @@ import { z } from 'zod';
 import * as StateManager from '../StateManager';
 import * as NetworkManager from '../../../src/helper/NetworkManager';
 import * as AccountManager from '../AccountManager';
+import { createAccBuilder } from '@lit-protocol/access-control-conditions';
 
 // PKP Sign Result Schema
 const PkpSignResultSchema = z.object({
@@ -147,6 +148,70 @@ export async function runPkpSignTest() {
   }
 }
 
+// test '/web/encryption/sign/v2' endpoint
+export async function runEncryptDecryptTest() {
+  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();
+
+    // Set up access control conditions requiring wallet ownership
+    const addressToUse = authContext.account.address;
+    const builder = createAccBuilder();
+    const accs = builder
+      .requireWalletOwnership(addressToUse)
+      .on('ethereum')
+      .build();
+
+    // Encrypt data with the access control conditions
+    const dataToEncrypt = 'Hello from PKP encrypt-decrypt test!';
+    const encryptedData = await litClient.encrypt({
+      dataToEncrypt,
+      unifiedAccessControlConditions: accs,
+      chain: 'ethereum',
+    });
+
+    // Decrypt the data using the appropriate auth context
+    const decryptedData = await litClient.decrypt({
+      data: encryptedData,
+      unifiedAccessControlConditions: accs,
+      chain: 'ethereum',
+      authContext,
+    });
+
+    // Assert that the decrypted data is the same as the original data
+    if (decryptedData.convertedData !== dataToEncrypt) {
+      throw new Error('❌ Decrypted data does not match the original data');
+    }
+
+    const endTime = Date.now();
+    const duration = endTime - startTime;
+
+    console.log(`✅ encrypt & decrypt successful in ${duration}ms`);
+
+    // For Artillery, just return - no need to call next()
+    return;
+  } catch (error) {
+    const endTime = Date.now();
+    const duration = endTime - startTime;
+
+    console.error(
+      `❌ encrypt & decrypt 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

@@ -28,6 +28,7 @@
     "artillery:init": "bun run ./e2e/artillery/src/init.ts",
     "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: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,