From 3603d8c0b4c494ad12328da36c8149364e7ae419 Mon Sep 17 00:00:00 2001
From: aon <21188659+aon@users.noreply.github.com>
Date: Thu, 30 Jan 2025 20:45:05 -0300
Subject: [PATCH] fix: test that included guardian contract

---
 test/SessionKeyTest.ts | 3 +++
 test/utils.ts          | 2 ++
 2 files changed, 5 insertions(+)

diff --git a/test/SessionKeyTest.ts b/test/SessionKeyTest.ts
index b8331055..e729b1a4 100644
--- 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
index c669bcee..4914c4cf 100644
--- 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);