Add tests for EmailAccountRecovery for jwt

packages/contracts/test/EmailAccountRecovery_jwt/EmailAccountRecovery_jwt_completeRecovery.t.sol

@@ -0,0 +1,237 @@
+// SPDX-License-Identifier: MIT
+pragma solidity ^0.8.12;
+
+import "forge-std/Test.sol";
+import "forge-std/console.sol";
+import {EmailAuth, EmailAuthMsg} from "../../src/EmailAuth.sol";
+import {RecoveryController} from "../helpers/RecoveryController.sol";
+import {StructHelper} from "../helpers/StructHelper.sol";
+import {SimpleWallet} from "../helpers/SimpleWallet.sol";
+import {OwnableUpgradeable} from "@openzeppelin/contracts-upgradeable/access/OwnableUpgradeable.sol";
+
+contract EmailAccountRecoveryTest_jwt_completeRecovery is StructHelper {
+    constructor() {}
+
+    function setUp() public override {
+        super.setUp();
+    }
+
+    function requestGuardian() public {
+        setUp();
+        require(
+            jwtRecoveryController.guardians(jwtGuardian) ==
+                RecoveryController.GuardianStatus.NONE
+        );
+
+        vm.startPrank(deployer);
+        jwtRecoveryController.requestGuardian(jwtGuardian);
+        vm.stopPrank();
+
+        require(
+            jwtRecoveryController.guardians(jwtGuardian) ==
+                RecoveryController.GuardianStatus.REQUESTED
+        );
+    }
+
+    function handleAcceptance() public {
+        requestGuardian();
+
+        console.log("jwtGuardian", jwtGuardian);
+
+        require(
+            jwtRecoveryController.guardians(jwtGuardian) ==
+                RecoveryController.GuardianStatus.REQUESTED
+        );
+
+        uint templateIdx = 0;
+
+        EmailAuthMsg memory emailAuthMsg = buildEmailAuthMsg();
+        uint templateId = jwtRecoveryController.computeAcceptanceTemplateId(
+            templateIdx
+        );
+        emailAuthMsg.templateId = templateId;
+        bytes[] memory commandParamsForAcceptance = new bytes[](1);
+        commandParamsForAcceptance[0] = abi.encode(address(jwtSimpleWallet));
+        emailAuthMsg.commandParams = commandParamsForAcceptance;
+
+        vm.mockCall(
+            address(jwtRecoveryController.emailAuthImplementationAddr()),
+            abi.encodeWithSelector(EmailAuth.authEmail.selector, emailAuthMsg),
+            abi.encode(0x0)
+        );
+
+        // acceptGuardian is internal, we call handleAcceptance, which calls acceptGuardian internally.
+        vm.startPrank(someRelayer);
+        jwtRecoveryController.handleAcceptance(emailAuthMsg, templateIdx);
+        vm.stopPrank();
+
+        require(
+            jwtRecoveryController.guardians(jwtGuardian) ==
+                RecoveryController.GuardianStatus.ACCEPTED
+        );
+    }
+
+    function handleRecovery() public {
+        handleAcceptance();
+
+        assertEq(jwtRecoveryController.isRecovering(address(jwtSimpleWallet)), false);
+        assertEq(
+            jwtRecoveryController.currentTimelockOfAccount(address(jwtSimpleWallet)),
+            0
+        );
+        assertEq(jwtSimpleWallet.owner(), deployer);
+        assertEq(
+            jwtRecoveryController.newSignerCandidateOfAccount(
+                address(jwtSimpleWallet)
+            ),
+            address(0x0)
+        );
+
+        uint templateIdx = 0;
+
+        EmailAuthMsg memory emailAuthMsg = buildEmailAuthMsg();
+        uint templateId = jwtRecoveryController.computeRecoveryTemplateId(
+            templateIdx
+        );
+        emailAuthMsg.templateId = templateId;
+        bytes[] memory commandParamsForRecovery = new bytes[](2);
+        commandParamsForRecovery[0] = abi.encode(jwtSimpleWallet);
+        commandParamsForRecovery[1] = abi.encode(newSigner);
+        emailAuthMsg.commandParams = commandParamsForRecovery;
+
+        vm.mockCall(
+            address(jwtRecoveryController.emailAuthImplementationAddr()),
+            abi.encodeWithSelector(EmailAuth.authEmail.selector, emailAuthMsg),
+            abi.encode(0x0)
+        );
+
+        vm.startPrank(someRelayer);
+        jwtRecoveryController.handleRecovery(emailAuthMsg, templateIdx);
+        vm.stopPrank();
+
+        assertEq(jwtRecoveryController.isRecovering(address(jwtSimpleWallet)), true);
+        assertEq(jwtSimpleWallet.owner(), deployer);
+        assertEq(
+            jwtRecoveryController.newSignerCandidateOfAccount(
+                address(jwtSimpleWallet)
+            ),
+            newSigner
+        );
+        assertEq(
+            jwtRecoveryController.currentTimelockOfAccount(address(jwtSimpleWallet)),
+            block.timestamp +
+                jwtRecoveryController.timelockPeriodOfAccount(
+                    address(jwtSimpleWallet)
+                )
+        );
+    }
+
+    function testCompleteRecovery() public {
+        skipIfZkSync();
+
+        handleRecovery();
+
+        assertEq(jwtRecoveryController.isRecovering(address(jwtSimpleWallet)), true);
+        assertEq(
+            jwtRecoveryController.currentTimelockOfAccount(address(jwtSimpleWallet)),
+            block.timestamp +
+                jwtRecoveryController.timelockPeriodOfAccount(
+                    address(jwtSimpleWallet)
+                )
+        );
+        assertEq(jwtSimpleWallet.owner(), deployer);
+        assertEq(
+            jwtRecoveryController.newSignerCandidateOfAccount(
+                address(jwtSimpleWallet)
+            ),
+            newSigner
+        );
+
+        vm.startPrank(someRelayer);
+        vm.warp(4 days);
+        jwtRecoveryController.completeRecovery(
+            address(jwtSimpleWallet),
+            new bytes(0)
+        );
+        vm.stopPrank();
+
+        assertEq(jwtRecoveryController.isRecovering(address(jwtSimpleWallet)), false);
+        assertEq(
+            jwtRecoveryController.currentTimelockOfAccount(address(jwtSimpleWallet)),
+            0
+        );
+        assertEq(jwtSimpleWallet.owner(), newSigner);
+        assertEq(
+            jwtRecoveryController.newSignerCandidateOfAccount(
+                address(jwtSimpleWallet)
+            ),
+            address(0x0)
+        );
+    }
+
+    function testExpectRevertCompleteRecoveryRecoveryNotInProgress() public {
+        skipIfZkSync();
+
+        handleAcceptance();
+
+        assertEq(jwtRecoveryController.isRecovering(address(jwtSimpleWallet)), false);
+        assertEq(
+            jwtRecoveryController.currentTimelockOfAccount(address(jwtSimpleWallet)),
+            0
+        );
+        assertEq(jwtSimpleWallet.owner(), deployer);
+        assertEq(
+            jwtRecoveryController.newSignerCandidateOfAccount(
+                address(jwtSimpleWallet)
+            ),
+            address(0x0)
+        );
+
+        vm.startPrank(someRelayer);
+        vm.warp(4 days);
+        vm.expectRevert(bytes("recovery not in progress"));
+        bytes memory recoveryCalldata;
+        jwtRecoveryController.completeRecovery(
+            address(jwtSimpleWallet),
+            recoveryCalldata
+        );
+
+        vm.stopPrank();
+    }
+
+    function testExpectRevertCompleteRecovery() public {
+        skipIfZkSync();
+
+        vm.warp(block.timestamp + 3 days);
+
+        handleRecovery();
+
+        assertEq(jwtRecoveryController.isRecovering(address(jwtSimpleWallet)), true);
+        assertEq(
+            jwtRecoveryController.currentTimelockOfAccount(address(jwtSimpleWallet)),
+            block.timestamp +
+                jwtRecoveryController.timelockPeriodOfAccount(
+                    address(jwtSimpleWallet)
+                )
+        );
+        assertEq(jwtSimpleWallet.owner(), deployer);
+        assertEq(
+            jwtRecoveryController.newSignerCandidateOfAccount(
+                address(jwtSimpleWallet)
+            ),
+            newSigner
+        );
+
+        vm.warp(0);
+
+        vm.startPrank(someRelayer);
+        vm.expectRevert(bytes("timelock not expired"));
+        bytes memory recoveryCalldata;
+        jwtRecoveryController.completeRecovery(
+            address(jwtSimpleWallet),
+            recoveryCalldata
+        );
+
+        vm.stopPrank();
+    }
+}

packages/contracts/test/EmailAccountRecovery_jwt/EmailAccountRecovery_jwt_handleAcceptance.t.sol

@@ -99,7 +99,7 @@ contract EmailAccountRecoveryTest_jwt_handleAcceptance is StructHelper {
         );
         emailAuthMsg.templateId = templateId;
         bytes[] memory commandParamsForAcceptance = new bytes[](1);
-        commandParamsForAcceptance[0] = abi.encode(address(simpleWallet));
+        commandParamsForAcceptance[0] = abi.encode(address(jwtSimpleWallet));
         emailAuthMsg.commandParams = commandParamsForAcceptance;
         emailAuthMsg.proof.accountSalt = 0x0;
 
@@ -133,7 +133,7 @@ contract EmailAccountRecoveryTest_jwt_handleAcceptance is StructHelper {
         );
         emailAuthMsg.templateId = templateId;
         bytes[] memory commandParamsForAcceptance = new bytes[](1);
-        commandParamsForAcceptance[0] = abi.encode(address(simpleWallet));
+        commandParamsForAcceptance[0] = abi.encode(address(jwtSimpleWallet));
         emailAuthMsg.commandParams = commandParamsForAcceptance;
 
         vm.mockCall(
@@ -166,8 +166,8 @@ contract EmailAccountRecoveryTest_jwt_handleAcceptance is StructHelper {
         );
         emailAuthMsg.templateId = templateId;
         bytes[] memory commandParamsForAcceptance = new bytes[](2);
-        commandParamsForAcceptance[0] = abi.encode(address(simpleWallet));
-        commandParamsForAcceptance[1] = abi.encode(address(simpleWallet));
+        commandParamsForAcceptance[0] = abi.encode(address(jwtSimpleWallet));
+        commandParamsForAcceptance[1] = abi.encode(address(jwtSimpleWallet));
         emailAuthMsg.commandParams = commandParamsForAcceptance;
 
         vm.mockCall(