feat(wrapped-keys): naga-test works when running against wrapped-keys.spec.ts directly, but not e2e.spec.ts

Author: Ansonhkg

packages/auth/src/lib/AuthManager/authAdapters/getPkpAuthContextFromPreGeneratedAdapter.ts

@@ -292,6 +292,8 @@ function extractAuthConfigFromDelegationAuthSig(delegationAuthSig: AuthSig): {
 } {
   const siweMessage = delegationAuthSig.signedMessage;
 
+  // const parsedSiweMessage = new SiweMessage(siweMessage);
+
   // Extract domain
   const domainMatch = siweMessage.match(/^([^\s]+) wants you to sign in/m);
   const domain = domainMatch ? domainMatch[1] : undefined;

packages/e2e/src/helper/createEnvVars.ts

@@ -123,6 +123,7 @@ export function createEnvVars(): EnvVars {
   });
 
   console.log('✅ Env Vars:', clone);
+  console.log("When RPC URL is undefined, module's default RPC will be used.")
 
   return result;
 }

packages/e2e/src/helper/createTestAccount.ts

@@ -1,5 +1,5 @@
 import { ViemAccountAuthenticator } from '@lit-protocol/auth';
-import { PKPData } from '@lit-protocol/schemas';
+import { AuthData, PKPData } from '@lit-protocol/schemas';
 import { generatePrivateKey, privateKeyToAccount } from 'viem/accounts';
 import { AuthContext } from '../types';
 import { TestEnv } from './createTestEnv';
@@ -40,6 +40,7 @@ export type CreateTestAccountResult = {
   paymentManager?: Awaited<
     ReturnType<TestEnv['litClient']['getPaymentManager']>
   >;
+  authData?: AuthData;
 };
 
 export async function createTestAccount(
@@ -55,11 +56,15 @@ export async function createTestAccount(
     pkpAuthContext: undefined,
     pkpViemAccount: undefined,
     paymentManager: undefined,
+    authData: undefined,
   };
 
   const personAccountAuthData = await ViemAccountAuthenticator.authenticate(
     person.account
   );
+
+  person.authData = personAccountAuthData;
+
   console.log(`Address`, person.account.address);
   console.log(`opts:`, opts);
 

packages/e2e/src/helper/createTestEnv.ts

@@ -1,17 +1,21 @@
+import {
+  createAuthManager,
+  generateSessionKeyPair,
+  storagePlugins,
+} from '@lit-protocol/auth';
+import { createLitClient } from '@lit-protocol/lit-client';
 import {
   LitNetworkModule,
+  naga,
   nagaDev,
   nagaLocal,
+  nagaProto,
   nagaStaging,
   nagaTest,
-  nagaProto,
-  naga,
   PaymentManager,
 } from '@lit-protocol/networks';
-import { EnvVars } from './createEnvVars';
-import { createLitClient } from '@lit-protocol/lit-client';
-import { createAuthManager, storagePlugins } from '@lit-protocol/auth';
 import { privateKeyToAccount } from 'viem/accounts';
+import { EnvVars } from './createEnvVars';
 
 export const CONFIG = {
   LOCAL: {
@@ -24,7 +28,7 @@ export const CONFIG = {
   },
   LIVE: {
     nativeFundingAmount: '0.1',
-    ledgerDepositAmount: '0.1',
+    ledgerDepositAmount: '10',
     sponsorshipLimits: {
       totalMaxPriceInWei: '50000000000000000',
       userMaxPrice: 50000000000000000n,
@@ -59,6 +63,9 @@ export type TestEnv = {
       userMaxPrice: bigint;
     };
   };
+
+  // --- Wrapped Keys related ---
+  sessionKeyPair: ReturnType<typeof generateSessionKeyPair>;
 };
 
 export const createTestEnv = async (envVars: EnvVars): Promise<TestEnv> => {
@@ -149,12 +156,16 @@ export const createTestEnv = async (envVars: EnvVars): Promise<TestEnv> => {
     account: masterAccount,
   });
 
+  // --- This info are used by wrapped keys tests ---
+  const sessionKeyPair = generateSessionKeyPair();
+
   return {
     masterAccount,
     masterPaymentManager,
     networkModule,
     litClient,
     authManager,
     config,
+    sessionKeyPair,
   };
 };

packages/e2e/src/helper/fundAccount.ts

@@ -126,6 +126,8 @@ export const fundAccount = async (
     console.log(`- Using default network RPC URL:`, defaultRpcUrl);
   }
 
+  const rpcUrl = customRpcUrl ?? defaultRpcUrl;
+
   // check account balance
   const publicClient = createPublicClient({
     chain: networkModule.getChainConfig(),

packages/e2e/src/helper/network.ts

@@ -16,6 +16,8 @@ export const NETWORKS = {
   'naga-test': { importName: 'nagaTest', type: 'live' },
   'naga-local': { importName: 'nagaLocal', type: 'local' },
   'naga-staging': { importName: 'nagaStaging', type: 'live' },
+  'naga-proto': { importName: 'nagaProto', type: 'live' },
+  naga: { importName: 'naga', type: 'live' },
 } as const;
 
 export type NetworkName = keyof typeof NETWORKS;

packages/e2e/src/init.ts

@@ -42,7 +42,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';
+const LIVE_NETWORK_LEDGER_DEPOSIT_AMOUNT = '0.1';
 const MAINNET_NETWORK_FUNDING_AMOUNT = '0.01';
 const MAINNET_LEDGER_DEPOSIT_AMOUNT = '0.01';