fix: tests

madjid80 · Oct 15, 2024 · b238db5 · b238db5
1 parent 5b188f5
commit b238db5
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 5 deletions.
diff --git a/src/module/smart-sessions/usage.ts b/src/module/smart-sessions/usage.ts
@@ -650,7 +650,7 @@ export const getEnableSessionDetails = async ({
     signature: '0x' as Hex,
     enableSessionData: {
       enableSession: {
-        chainDigestIndex: 0,
+        chainDigestIndex: sessionIndex || 0,
         hashesAndChainIds: chainDigests,
         sessionToEnable,
         permissionEnableSig: '0x' as Hex,

diff --git a/test/unit/module/smartSessions/smartSessionPolicies.test.ts b/test/unit/module/smartSessions/smartSessionPolicies.test.ts
@@ -38,7 +38,6 @@ describe('Smart Sessions Polices', () => {
       UNIVERSAL_ACTION_POLICY_ADDRESS,
     )
     expect(installUniversalPolicy.initData).toBeDefined()
-    expect(installUniversalPolicy.deInitData).toEqual('0x')
   })
 
   // -----------------------
@@ -48,7 +47,6 @@ describe('Smart Sessions Polices', () => {
     const installSudoActionPolicy = getSudoPolicy()
     expect(installSudoActionPolicy.address).toBeDefined()
     expect(installSudoActionPolicy.initData).toEqual('0x')
-    expect(installSudoActionPolicy.deInitData).toEqual('0x')
   })
 
   // -----------------------
@@ -61,6 +59,5 @@ describe('Smart Sessions Polices', () => {
 
     expect(installSpendingLimitPolicy.address).toBeDefined()
     expect(installSpendingLimitPolicy.initData).toBeDefined()
-    expect(installSpendingLimitPolicy.deInitData).toEqual('0x')
   })
 })
diff --git a/test/unit/module/smartSessions/smartSessionsValidator.test.ts b/test/unit/module/smartSessions/smartSessionsValidator.test.ts
@@ -20,7 +20,7 @@ import {
   getPermissionId,
 } from 'src/module/smart-sessions'
 import { privateKeyToAccount, generatePrivateKey } from 'viem/accounts'
-import { mainnet } from 'viem/chains'
+import { mainnet, sepolia } from 'viem/chains'
 import { getPublicClient } from 'test/utils/userOps/clients'
 import {
   encodeValidationData,
@@ -42,6 +42,7 @@ describe('Smart Sessions Polices', () => {
           allowedERC7739Content: [],
           erc1271Policies: [],
         },
+        chainId: BigInt(sepolia.id),
       },
     ]
 
@@ -104,6 +105,7 @@ describe('Smart Sessions Polices', () => {
         allowedERC7739Content: [],
         erc1271Policies: [],
       },
+      chainId: BigInt(sepolia.id),
     }
 
     const permissionId = getPermissionId({
@@ -159,6 +161,9 @@ describe('Smart Sessions Polices', () => {
       account,
     })
 
+    decodedSig.enableSessionData!.enableSession.sessionToEnable.chainId =
+      BigInt(sepolia.id)
+
     expect(decodedSig.mode).toEqual(SmartSessionMode.ENABLE)
     expect(decodedSig.permissionId).toEqual(permissionId)
     expect(decodedSig.signature).toEqual(signature)
@@ -212,6 +217,7 @@ describe('Smart Sessions Polices', () => {
         allowedERC7739Content: [],
         erc1271Policies: [],
       },
+      chainId: BigInt(sepolia.id),
     }
     const permissionId = getPermissionId({
       session,
@@ -266,6 +272,9 @@ describe('Smart Sessions Polices', () => {
       account,
     })
 
+    decodedSig.enableSessionData!.enableSession.sessionToEnable.chainId =
+      BigInt(sepolia.id)
+
     expect(decodedSig.mode).toEqual(SmartSessionMode.ENABLE)
     expect(decodedSig.permissionId).toEqual(permissionId)
     expect(decodedSig.signature).toEqual(signature)
@@ -338,6 +347,7 @@ describe('Smart Sessions Polices', () => {
         allowedERC7739Content: [],
         erc1271Policies: [],
       },
+      chainId: BigInt(sepolia.id),
     }
 
     const sessionDigest = keccak256(toHex('sessionDigest'))