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

@@ -1,18 +1,21 @@
-import { createAuthManager, storagePlugins } from "@lit-protocol/auth";
-import { createLitClient } from "@lit-protocol/lit-client";
-import { z } from "zod";
-import * as StateManager from "../StateManager";
-import * as NetworkManager from "../../../src/helper/NetworkManager";
-import * as AccountManager from "../AccountManager";
+import { createAuthManager, storagePlugins } from '@lit-protocol/auth';
+import { createLitClient } from '@lit-protocol/lit-client';
+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({
-  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"),
+  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'),
 });
 
 // Global variables to cache expensive operations
@@ -57,11 +60,15 @@ const createAuthContextFromState = async () => {
 
     // Validate that master account authData and PKP exist
     if (!state.masterAccount.authData) {
-      throw new Error('❌ Master account authData not found in state. Run init.ts first.');
+      throw new Error(
+        '❌ Master account authData not found in state. Run init.ts first.'
+      );
     }
 
     if (!state.masterAccount.pkp) {
-      throw new Error('❌ Master account PKP not found in state. Run init.ts first.');
+      throw new Error(
+        '❌ Master account PKP not found in state. Run init.ts first.'
+      );
     }
 
     // Get the master account from environment (same as init.ts)
@@ -126,12 +133,78 @@ export async function runPkpSignTest() {
 
     // For Artillery, just return - no need to call next()
     return;
+  } catch (error) {
+    const endTime = Date.now();
+    const duration = endTime - startTime;
+
+    console.error(
+      `❌ pkpSign failed in ${duration}ms:`,
+      error instanceof Error ? error.message : String(error)
+    );
+
+    // Throw the error to let Artillery handle it
+    throw error;
+  }
+}
 
+// 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(`❌ pkpSign failed in ${duration}ms:`, error instanceof Error ? error.message : String(error));
+    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;
@@ -140,7 +213,6 @@ export async function runPkpSignTest() {
 
 // test '/web/sign_session_key' endpoint
 export async function runSignSessionKeyTest() {
-
   // ❗️ IT'S IMPORTANT TO SET THIS TO FALSE FOR TESTING
   const DELEGATION_AUTH_SIG_CACHE = false;
 
@@ -167,18 +239,20 @@ export async function runSignSessionKeyTest() {
       litClient: litClient,
       cache: {
         delegationAuthSig: DELEGATION_AUTH_SIG_CACHE,
-      }
+      },
     });
 
     // console.log('✅ Master Account PKP Auth Context:', masterAccountPkpAuthContext);
-
   } catch (error) {
     const endTime = Date.now();
     const duration = endTime - startTime;
 
-    console.error(`❌ signSessionKey failed in ${duration}ms:`, error instanceof Error ? error.message : String(error));
+    console.error(
+      `❌ signSessionKey failed in ${duration}ms:`,
+      error instanceof Error ? error.message : String(error)
+    );
 
     // Throw the error to let Artillery handle it
     throw error;
   }
-}
+}

package.json

@@ -27,6 +27,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,