fix: test that included guardian contract

matter-labs · Jan 30, 2025 · 3603d8c · 3603d8c
1 parent cb4f402
commit 3603d8c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/test/SessionKeyTest.ts b/test/SessionKeyTest.ts
@@ -310,9 +310,12 @@ describe("SessionKeyModule tests", function () {
     assert(beaconContract != null, "No Beacon deployed");
     const factoryContract = await fixtures.getAaFactory();
     assert(factoryContract != null, "No AA Factory deployed");
+    const guardianRecoveryContract = await fixtures.getGuardianRecoveryValidator();
+    assert(guardianRecoveryContract != null, "No Guardian Recovery deployed");
     const authServerPaymaster = await fixtures.deployExampleAuthServerPaymaster(
       await factoryContract.getAddress(),
       await sessionModuleContract.getAddress(),
+      await guardianRecoveryContract.getAddress(),
     );
     assert(authServerPaymaster != null, "No Auth Server Paymaster deployed");
 

diff --git a/test/utils.ts b/test/utils.ts
@@ -150,6 +150,7 @@ export class ContractFixtures {
   async deployExampleAuthServerPaymaster(
     aaFactoryAddress: string,
     sessionKeyValidatorAddress: string,
+    guardianRecoveryValidatorAddress: string,
   ): Promise<ExampleAuthServerPaymaster> {
     const contract = await create2(
       "ExampleAuthServerPaymaster",
@@ -158,6 +159,7 @@ export class ContractFixtures {
       [
         aaFactoryAddress,
         sessionKeyValidatorAddress,
+        guardianRecoveryValidatorAddress,
       ],
     );
     const paymasterAddress = ExampleAuthServerPaymaster__factory.connect(await contract.getAddress(), this.wallet);