fmt

Author: Ansonhkg

.github/workflows/e2e.yml

@@ -2,6 +2,10 @@ name: E2E Tests
 
 on:
   push:
+    branches:
+      - main
+      - master
+      - develop
   pull_request:
   workflow_dispatch:
     inputs:
@@ -50,6 +54,30 @@ jobs:
       - name: Build project
         run: bun run build
 
+      - name: Validate environment variables
+        run: |
+          echo "Checking environment variables..."
+          if [ -z "$LIVE_MASTER_ACCOUNT" ]; then
+            echo "❌ LIVE_MASTER_ACCOUNT is not set"
+            exit 1
+          fi
+          if [ -z "$LOCAL_MASTER_ACCOUNT" ]; then
+            echo "❌ LOCAL_MASTER_ACCOUNT is not set"  
+            exit 1
+          fi
+          echo "✅ Required environment variables are set"
+          
+          # Validate private key format (should be 0x followed by 64 hex chars)
+          if [[ ! "$LIVE_MASTER_ACCOUNT" =~ ^0x[a-fA-F0-9]{64}$ ]]; then
+            echo "❌ LIVE_MASTER_ACCOUNT is not in correct format (should be 0x followed by 64 hex characters)"
+            exit 1
+          fi
+          if [[ ! "$LOCAL_MASTER_ACCOUNT" =~ ^0x[a-fA-F0-9]{64}$ ]]; then
+            echo "❌ LOCAL_MASTER_ACCOUNT is not in correct format (should be 0x followed by 64 hex characters)"
+            exit 1
+          fi
+          echo "✅ Private key formats are valid"
+
       - name: Run health check for naga-dev
         run: NETWORK=naga-dev bun run test:e2e pkpSign
         timeout-minutes: 10

e2e/artillery/configs/pkp-sign.yml

@@ -1,26 +1,26 @@
 config:
-  target: "dummy"
+  target: 'dummy'
   phases:
     # Over 60s, ramp up to creating 50 vusers per second
     - duration: 60
       arrivalRate: 5
       # rampTo: 50
       rampTo: 10
-      name: "Ramp Up"
+      name: 'Ramp Up'
     # Over 300s, create 50 vusers per second
     - duration: 300
       # arrivalRate: 50
       arrivalRate: 10
-      name: "Sustained PKP Signing"
+      name: 'Sustained PKP Signing'
     # Over 60s, ramp down to creating 5 vusers per second
     - duration: 60
       arrivalRate: 5
-      name: "Ramp Down"
-  processor: "../src/processors/multi-endpoints.ts"
+      name: 'Ramp Down'
+  processor: '../src/processors/multi-endpoints.ts'
 
 scenarios:
-  - name: "PKP Sign Stress Test"
+  - name: 'PKP Sign Stress Test'
     weight: 100
     flow:
-      - function: "runPkpSignTest"
-      - think: 0.1 
+      - function: 'runPkpSignTest'
+      - think: 0.1

e2e/artillery/configs/sign-session-key.yml

@@ -1,27 +1,26 @@
 config:
-  target: "dummy"
+  target: 'dummy'
   phases:
     # Over 60s, ramp up to creating 50 vusers per second
     - duration: 60
       arrivalRate: 5
       # rampTo: 50
       rampTo: 10
-      name: "Ramp Up"
+      name: 'Ramp Up'
     # Over 300s, create 50 vusers per second
     - duration: 300
       # arrivalRate: 50
       arrivalRate: 10
-      name: "Sustained Sign Session Key"
+      name: 'Sustained Sign Session Key'
     # Over 60s, ramp down to creating 5 vusers per second
     - duration: 60
       arrivalRate: 5
-      name: "Ramp Down"
-  processor: "../src/processors/multi-endpoints.ts"
+      name: 'Ramp Down'
+  processor: '../src/processors/multi-endpoints.ts'
 
 scenarios:
-  - name: "Sign Session Key Stress Test"
+  - name: 'Sign Session Key Stress Test'
     weight: 100
     flow:
-
-      - function: "runSignSessionKeyTest"
-      - think: 0.1 
+      - function: 'runSignSessionKeyTest'
+      - think: 0.1

e2e/artillery/src/AccountManager.ts

@@ -1,9 +1,9 @@
-import * as NetworkManager from "../../../e2e/src/helper/NetworkManager";
-import { privateKeyToAccount } from "viem/accounts";
-import { createPublicClient, formatEther, http } from "viem";
-import { createLitClient } from "@lit-protocol/lit-client";
-import * as StateManager from "./StateManager";
-import { printAligned } from "../../../e2e/src/helper/utils";
+import * as NetworkManager from '../../../e2e/src/helper/NetworkManager';
+import { privateKeyToAccount } from 'viem/accounts';
+import { createPublicClient, formatEther, http } from 'viem';
+import { createLitClient } from '@lit-protocol/lit-client';
+import * as StateManager from './StateManager';
+import { printAligned } from '../../../e2e/src/helper/utils';
 
 export const getMasterAccount = async () => {
   const privateKey = process.env['LIVE_MASTER_ACCOUNT'];
@@ -13,13 +13,13 @@ export const getMasterAccount = async () => {
   }
 
   return privateKeyToAccount(privateKey as `0x${string}`);
-}
+};
 
 export const getAccountDetails = async ({
   account,
   publicClient,
   litClient,
-  accountLabel = "Account",
+  accountLabel = 'Account',
 }: {
   account: any;
   publicClient: any;
@@ -29,40 +29,44 @@ export const getAccountDetails = async ({
   console.log(`\n========== ${accountLabel} Details ==========`);
 
   // Get all the data first
-  const ethBalance = formatEther(await publicClient.getBalance({
-    address: account.address,
-  }));
+  const ethBalance = formatEther(
+    await publicClient.getBalance({
+      address: account.address,
+    })
+  );
 
   const paymentManager = await litClient.getPaymentManager({
     account: account,
   });
 
   const paymentBalance = await paymentManager.getBalance({
-    userAddress: account.address
+    userAddress: account.address,
   });
 
   // Determine status
-  let statusLabel = "";
-  let statusValue = "";
+  let statusLabel = '';
+  let statusValue = '';
 
   if (Number(paymentBalance.availableBalance) < 0) {
-    statusLabel = "🚨 Status:";
+    statusLabel = '🚨 Status:';
     statusValue = `Negative balance (debt): ${paymentBalance.availableBalance}`;
   }
 
   // Print all information with consistent alignment
   printAligned([
-    { label: "🔑 Address:", value: account.address },
-    { label: "💰 ETH Balance:", value: `${ethBalance} ETH` },
-    { label: "💳 Ledger Total Balance:", value: paymentBalance.totalBalance },
-    { label: "💳 Ledger Available Balance:", value: paymentBalance.availableBalance },
+    { label: '🔑 Address:', value: account.address },
+    { label: '💰 ETH Balance:', value: `${ethBalance} ETH` },
+    { label: '💳 Ledger Total Balance:', value: paymentBalance.totalBalance },
+    {
+      label: '💳 Ledger Available Balance:',
+      value: paymentBalance.availableBalance,
+    },
     { label: statusLabel, value: statusValue },
   ]);
 
   return {
     ethBalance,
     ledgerBalance: paymentBalance.availableBalance,
     paymentManager,
-  }
-}
-
+  };
+};

e2e/artillery/src/StateManager.ts

@@ -8,7 +8,7 @@ const StateObject = {
   masterAccount: {
     // privateKey: undefined as string | `0x${string}` | undefined,
     // address: undefined as string | `0x${string}` | undefined,
-    authData: undefined as any | undefined,  // Changed from string to any since authData is an object
+    authData: undefined as any | undefined, // Changed from string to any since authData is an object
     pkp: undefined as any | undefined,
   },
 };
@@ -34,7 +34,7 @@ export const readFile = async (): Promise<State> => {
   try {
     return await _readFile();
   } catch (error) {
-    console.log("🚨 Failed to read file, creating new file...");
+    console.log('🚨 Failed to read file, creating new file...');
     await createFile();
     return await _readFile();
   }
@@ -48,22 +48,23 @@ export const createFile = async () => {
 // Type-safe field paths - dynamically derived from State type
 type StatePaths = {
   [K in keyof State]: K extends string
-  ? State[K] extends object
-  ? {
-    [P in keyof State[K]]: P extends string ? `${K}.${P}` : never
-  }[keyof State[K]] | K  // Include both nested paths AND top-level key
-  : K
-  : never
+    ? State[K] extends object
+      ?
+          | {
+              [P in keyof State[K]]: P extends string ? `${K}.${P}` : never;
+            }[keyof State[K]]
+          | K // Include both nested paths AND top-level key
+      : K
+    : never;
 }[keyof State];
 
 // Helper type to get nested property type
-type GetNestedType<T, P extends string> =
-  P extends `${infer K}.${infer Rest}`
+type GetNestedType<T, P extends string> = P extends `${infer K}.${infer Rest}`
   ? K extends keyof T
-  ? Rest extends keyof T[K]
-  ? T[K][Rest]
-  : never
-  : never
+    ? Rest extends keyof T[K]
+      ? T[K][Rest]
+      : never
+    : never
   : P extends keyof T
   ? T[P]
   : never;
@@ -86,7 +87,11 @@ export const updateField = async <T extends StatePaths>(
   const pathParts = path.split('.') as [keyof State, string];
   const [rootKey, nestedKey] = pathParts;
 
-  if (rootKey in state && typeof state[rootKey] === 'object' && state[rootKey] !== null) {
+  if (
+    rootKey in state &&
+    typeof state[rootKey] === 'object' &&
+    state[rootKey] !== null
+  ) {
     (state[rootKey] as any)[nestedKey] = value;
     await fs.writeFile(FILE_NAME, JSON.stringify(state, null, 2));
   } else {
@@ -125,7 +130,11 @@ export const getOrUpdate = async <T extends StatePaths>(
     const pathParts = path.split('.') as [keyof State, string];
     const [rootKey, nestedKey] = pathParts;
 
-    if (rootKey in state && typeof state[rootKey] === 'object' && state[rootKey] !== null) {
+    if (
+      rootKey in state &&
+      typeof state[rootKey] === 'object' &&
+      state[rootKey] !== null
+    ) {
       const currentValue = (state[rootKey] as any)[nestedKey];
 
       // If value exists and is not null/undefined/empty string, return it

e2e/artillery/src/balance-status.ts

@@ -1,12 +1,13 @@
-import { createLitClient } from "@lit-protocol/lit-client";
-import * as NetworkManager from "../../src/helper/NetworkManager";
+import { createLitClient } from '@lit-protocol/lit-client';
+import * as NetworkManager from '../../src/helper/NetworkManager';
 import * as AccountManager from '../src/AccountManager';
 
 (async () => {
-
   // 1. Setup network and chain client
   const networkModule = await NetworkManager.getLitNetworkModule();
-  const publicClient = await NetworkManager.getViemPublicClient({ networkModule });
+  const publicClient = await NetworkManager.getViemPublicClient({
+    networkModule,
+  });
   const litClient = await createLitClient({ network: networkModule });
 
   // 2. Get the master account
@@ -15,11 +16,10 @@ import * as AccountManager from '../src/AccountManager';
   // every 5 seconds, check the balance of the master account
   setInterval(async () => {
     await AccountManager.getAccountDetails({
-      accountLabel: "Master Account",
+      accountLabel: 'Master Account',
       account: masterAccount,
       publicClient,
       litClient,
     });
   }, 3000);
-
-})();
+})();