Merge pull request #844 from LIT-Protocol/feature/node-4612-add-ledger-deposit-for-e2e-tests

Ansonhkg · web-flow · commit 4970ac4fec8b · 2025-07-25T23:41:32.000+01:00
Feature/node 4612 add ledger deposit for e2e tests
diff --git a/apps/lit-auth-service/bun.lock b/apps/lit-auth-service/bun.lock
diff --git a/apps/lit-login-service/bun.lock b/apps/lit-login-service/bun.lock
diff --git a/bun.lock b/bun.lock
diff --git a/e2e/src/helper/tests/encrypt-decrypt-flow.ts b/e2e/src/helper/tests/encrypt-decrypt-flow.ts
@@ -9,15 +9,9 @@ export const createEncryptDecryptFlowTest = (
     const { createAccBuilder } = await import(
       '@lit-protocol/access-control-conditions'
     );
-    const { generatePrivateKey, privateKeyToAccount } = await import(
-      'viem/accounts'
-    );
 
     const authContext = getAuthContext();
 
-    // Create a test account for Bob (recipient)
-    const bobAccount = privateKeyToAccount(generatePrivateKey());
-
     // Determine which address to use for Alice based on auth context type
     let aliceAddress: string;
     if (authContext === ctx.aliceEoaAuthContext) {
@@ -29,7 +23,7 @@ export const createEncryptDecryptFlowTest = (
     // Set up access control conditions requiring Bob's wallet ownership
     const builder = createAccBuilder();
     const accs = builder
-      .requireWalletOwnership(bobAccount.address)
+      .requireWalletOwnership(ctx.bobViemAccount.address)
       .on('ethereum')
       .build();
 
@@ -50,7 +44,7 @@ export const createEncryptDecryptFlowTest = (
     const jsonData = {
       message: 'Test JSON data',
       sender: aliceAddress,
-      recipient: bobAccount.address,
+      recipient: ctx.bobViemAccount.address,
       timestamp: Date.now(),
     };
 
@@ -105,7 +99,7 @@ export const createEncryptDecryptFlowTest = (
     // Create Bob's auth context for decryption
     const bobAuthContext = await ctx.authManager.createEoaAuthContext({
       config: {
-        account: bobAccount,
+        account: ctx.bobViemAccount,
       },
       authConfig: {
         domain: 'localhost',
diff --git a/e2e/src/init.ts b/e2e/src/init.ts
@@ -25,6 +25,7 @@ type LogLevel = z.infer<typeof LogLevelSchema>;
 // Configurations
 const LIVE_NETWORK_FUNDING_AMOUNT = '0.01';
 const LOCAL_NETWORK_FUNDING_AMOUNT = '1';
+const LIVE_NETWORK_LEDGER_DEPOSIT_AMOUNT = '2';
 
 export const init = async (
   network?: SupportedNetwork,
@@ -41,6 +42,7 @@ export const init = async (
   bobViemAccountPkp: any;
   aliceEoaAuthContext: any;
   alicePkpAuthContext: any;
+  masterDepositForUser: (userAddress: string) => Promise<void>;
   // alicePkpViemAccountPermissionsManager: any,
 }> => {
   /**
@@ -132,6 +134,18 @@ export const init = async (
    */
   const litClient = await createLitClient({ network: _networkModule });
 
+  /**
+   * ====================================
+   * (Master) Initialise Payment Manager
+   * ====================================
+  */
+  const masterPaymentManager = await litClient.getPaymentManager({
+    account: masterAccount,
+  });
+
+  const masterPaymentBalance = await masterPaymentManager.getBalance({ userAddress: masterAccount.address })
+  console.log('✅ Master Payment Balance:', masterPaymentBalance);
+
   /**
    * ====================================
    * Initialise the AuthManager
@@ -211,16 +225,39 @@ export const init = async (
     litClient: litClient,
   });
 
-  // const alicePkpViemAccount = await litClient.getPkpViemAccount({
-  //   pkpPublicKey: aliceViemAccountPkp.publicKey,
-  //   authContext: alicePkpAuthContext,
-  //   chainConfig: _networkModule.getChainConfig(),
-  // });
+  const alicePkpViemAccount = await litClient.getPkpViemAccount({
+    pkpPublicKey: aliceViemAccountPkp.publicKey,
+    authContext: alicePkpAuthContext,
+    chainConfig: _networkModule.getChainConfig(),
+  });
 
-  // await fundAccount(alicePkpViemAccount, localMasterAccount, _networkModule, {
-  //   ifLessThan: LOCAL_NETWORK_FUNDING_AMOUNT,
-  //   thenFundWith: LOCAL_NETWORK_FUNDING_AMOUNT,
-  // });
+  await fundAccount(alicePkpViemAccount, masterAccount, _networkModule, {
+    ifLessThan: LOCAL_NETWORK_FUNDING_AMOUNT,
+    thenFundWith: LOCAL_NETWORK_FUNDING_AMOUNT,
+  });
+
+  /**
+   * ====================================
+   * Depositing to Lit Ledger for differen 
+   * ====================================
+   */
+
+  async function masterDepositForUser(userAddress: string) {
+    await masterPaymentManager.depositForUser({ userAddress: userAddress, amountInEth: LIVE_NETWORK_LEDGER_DEPOSIT_AMOUNT });
+    console.log(`✅ New ${userAddress} Ledger Balance:`, await masterPaymentManager.getBalance({ userAddress: userAddress }));
+  }
+
+  // Deposit to the Alice EOA Ledger
+  await masterDepositForUser(aliceViemAccount.address);
+
+  // Deposit to the PKP Ledger
+  await masterDepositForUser(alicePkpViemAccount.address);
+
+  // Deposit to the Bob EOA Ledger
+  await masterDepositForUser(bobViemAccount.address);
+
+  // Deposit to the Bob PKP Ledger
+  await masterDepositForUser(bobViemAccountPkp.ethAddress);
 
   // const alicePkpViemAccountPermissionsManager = await litClient.getPKPPermissionsManager({
   //   pkpIdentifier: {
@@ -247,6 +284,7 @@ export const init = async (
     bobViemAccountPkp,
     aliceEoaAuthContext,
     alicePkpAuthContext,
+    masterDepositForUser
     // alicePkpViemAccountPermissionsManager
   };
 };
diff --git a/package.json b/package.json
@@ -34,7 +34,7 @@
   "private": true,
   "dependencies": {
     "@dotenvx/dotenvx": "^1.6.4",
-    "@lit-protocol/contracts": "^0.2.3",
+    "@lit-protocol/contracts": "^0.4.0",
     "@lit-protocol/nacl": "7.1.1",
     "@lit-protocol/uint8arrays": "7.1.1",
     "@metamask/eth-sig-util": "5.0.2",
diff --git a/packages/auth-services/package.json b/packages/auth-services/package.json
@@ -20,7 +20,7 @@
     "@elysiajs/cors": "^1.2.0",
     "@elysiajs/static": "^1.3.0",
     "@elysiajs/swagger": "^1.2.0",
-    "@lit-protocol/contracts": "^0.2.3",
+    "@lit-protocol/contracts": "^0.4.0",
     "@simplewebauthn/server": "6.2.1",
     "@simplewebauthn/typescript-types": "^8.3.4",
     "@t3-oss/env-core": "^0.12.0",
diff --git a/packages/networks/package.json b/packages/networks/package.json
@@ -18,7 +18,7 @@
     "directory": "../../dist/packages/networks"
   },
   "dependencies": {
-    "@lit-protocol/contracts": "^0.2.3",
+    "@lit-protocol/contracts": "^0.4.0",
     "@lit-protocol/nacl": "7.1.1",
     "@noble/curves": "^1.8.1",
     "@wagmi/core": "^2.17.1",
diff --git a/packages/networks/src/networks/vNaga/LitChainClient/apis/highLevelApis/PKPPermissionsManager/handlers/removePermittedAuthMethodByIdentifier.ts b/packages/networks/src/networks/vNaga/LitChainClient/apis/highLevelApis/PKPPermissionsManager/handlers/removePermittedAuthMethodByIdentifier.ts
@@ -43,9 +43,9 @@ export async function removePermittedAuthMethodByIdentifier(
     networkCtx
   );
 
-  console.log('❌ AUTH METHOD TYPE:', authMethodType);
-  console.log('❌ AUTH METHOD ID:', authMethodId);
-  console.log('❌ PKP TOKEN ID:', pkpTokenId);
+  // console.log('❌ AUTH METHOD TYPE:', authMethodType);
+  // console.log('❌ AUTH METHOD ID:', authMethodId);
+  // console.log('❌ PKP TOKEN ID:', pkpTokenId);
 
   return removePermittedAuthMethod(
     {
diff --git a/packages/networks/src/networks/vNaga/LitChainClient/apis/highLevelApis/PKPPermissionsManager/handlers/removePermittedAuthMethodScopeByIdentifier.ts b/packages/networks/src/networks/vNaga/LitChainClient/apis/highLevelApis/PKPPermissionsManager/handlers/removePermittedAuthMethodScopeByIdentifier.ts
@@ -44,10 +44,10 @@ export async function removePermittedAuthMethodScopeByIdentifier(
     networkCtx
   );
 
-  console.log('🔥 AUTH METHOD TYPE:', authMethodType);
-  console.log('🔥 AUTH METHOD ID:', authMethodId);
-  console.log('🔥 SCOPE ID:', scopeId);
-  console.log('🔥 PKP TOKEN ID:', pkpTokenId);
+  // console.log('🔥 AUTH METHOD TYPE:', authMethodType);
+  // console.log('🔥 AUTH METHOD ID:', authMethodId);
+  // console.log('🔥 SCOPE ID:', scopeId);
+  // console.log('🔥 PKP TOKEN ID:', pkpTokenId);
 
   return removePermittedAuthMethodScope(
     {
