fix(artillery): align init helpers, state cache, and docs

Author: Ansonhkg

README.md

@@ -99,6 +99,50 @@ DIRECTORY_NAME=naga-local
 NETWORK=naga-local pnpm run test:e2e all
 ```
 
+# Artillery Load Testing
+
+Use the standalone Artillery project under `packages/artillery` to exercise Lit endpoints with realistic workloads.
+
+## Preparation
+
+```bash
+# from the repo root
+pnpm install
+
+# pick your target network: naga-dev | naga-staging | naga-test | naga-local
+export NETWORK=naga-staging
+export LOG_LEVEL=info           # optional: debug | debug2 | silent
+```
+
+If you want Artillery Cloud reports, set `ARTILLERY_KEY=<your-key>` in `.env` before running a scenario.
+
+## One-time initialisation
+
+Master account, auth data and PKP info are written to this file:
+`packages/artillery/artillery-state.json`.
+
+```bash
+pnpm nx run artillery:init
+```
+
+(optional) Check master balances before blasting a load test:
+
+```bash
+pnpm nx run artillery:balance-status
+```
+
+## Run a workload
+
+Each scenario is exposed as an Nx target. Use the `run:` prefixed name:
+
+```bash
+pnpm nx run artillery:run:pkp-sign          # PKP signing focus
+pnpm nx run artillery:run:encrypt-decrypt   # Encryption/decryption focus
+pnpm nx run artillery:run:execute           # Lit Action execution
+pnpm nx run artillery:run:mix               # Mixed workload
+pnpm nx run artillery:run:sign-session-key  # Session key signing
+```
+
 # Manual Publishing
 
 ```bash

packages/artillery/src/StateManager.ts

@@ -24,7 +24,7 @@ export const readFile = async (): Promise<State> => {
 
     // If content is empty object, write base state
     if (Object.keys(content).length === 0) {
-      await fs.writeFile(FILE_NAME, JSON.stringify(StateObject, null, 2));
+      await fs.writeFile(FILE_NAME, stringify(StateObject));
       return StateObject;
     }
 
@@ -42,7 +42,7 @@ export const readFile = async (): Promise<State> => {
 
 // create the file if it doesn't exist
 export const createFile = async () => {
-  await fs.writeFile(FILE_NAME, JSON.stringify(StateObject, null, 2));
+  await fs.writeFile(FILE_NAME, stringify(StateObject));
 };
 
 // Type-safe field paths - dynamically derived from State type
@@ -93,7 +93,7 @@ export const updateField = async <T extends StatePaths>(
     state[rootKey] !== null
   ) {
     (state[rootKey] as any)[nestedKey] = value;
-    await fs.writeFile(FILE_NAME, JSON.stringify(state, null, 2));
+    await fs.writeFile(FILE_NAME, stringify(state));
   } else {
     throw new Error(`Invalid path: ${path}`);
   }
@@ -123,7 +123,7 @@ export const getOrUpdate = async <T extends StatePaths>(
 
     // Otherwise, update with default value and return it
     (state as any)[path] = defaultValue;
-    await fs.writeFile(FILE_NAME, JSON.stringify(state, null, 2));
+    await fs.writeFile(FILE_NAME, stringify(state));
     return defaultValue;
   } else {
     // Nested property
@@ -144,10 +144,17 @@ export const getOrUpdate = async <T extends StatePaths>(
 
       // Otherwise, update with default value and return it
       (state[rootKey] as any)[nestedKey] = defaultValue;
-      await fs.writeFile(FILE_NAME, JSON.stringify(state, null, 2));
+      await fs.writeFile(FILE_NAME, stringify(state));
       return defaultValue;
     } else {
       throw new Error(`Invalid path: ${path}`);
     }
   }
 };
+
+const stringify = (value: unknown) =>
+  JSON.stringify(
+    value,
+    (_, val) => (typeof val === 'bigint' ? val.toString() : val),
+    2
+  );

packages/artillery/src/init.ts

@@ -6,8 +6,11 @@ import {
 } from '@lit-protocol/auth';
 import * as StateManager from './StateManager';
 import { createLitClient } from '@lit-protocol/lit-client';
-import { getOrCreatePkp } from '@lit-protocol/e2e/src/helper/pkp-utils';
-import * as NetworkManager from '@lit-protocol/e2e/src/helper/NetworkManager';
+import {
+  getOrCreatePkp,
+  getLitNetworkModule,
+  getViemPublicClient,
+} from '@lit-protocol/e2e';
 import * as AccountManager from '../src/AccountManager';
 
 const _network = process.env['NETWORK'];