👷🏻 📚 ✅ Complete test and add misc docs

Kwenta · Jul 10, 2024 · d73eb2b · d73eb2b
1 parent e2840cc
commit d73eb2b
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 7 deletions.
diff --git a/src/Account.sol b/src/Account.sol
@@ -64,6 +64,7 @@ contract Account is IAccount, IERC721Receiver {
         }
     }
 
+    // add argument for amount to deposit with minimum value
     function setupAccount() external {
         accountId = perpsMarketSNXV3.createAccount();
         perpsMarketSNXV3.grantPermission({

diff --git a/src/MarginPaymaster.sol b/src/MarginPaymaster.sol
@@ -79,7 +79,7 @@ contract MarginPaymaster is IPaymaster, Zap {
             uint128(actualGasCostInWei), // TODO: account for gas costs of postOp func
             address(weth),
             address(_USDC)
-        ) * 110) / 100; // 10% slippage
+        ) * 110) / 100; // allow for 10% slippage
         uint256 USDCToSwapForWETH = costOfGasInUSDC;
         address sender = abi.decode(context, (address));
         uint256 availableUSDCInWallet = getUSDCAvailableInWallet(sender);

diff --git a/test/MarginPaymaster.t.sol b/test/MarginPaymaster.t.sol
@@ -147,12 +147,12 @@ contract MarginPaymasterTest is Bootstrap {
         assertEq(usdc.balanceOf(address(this)), 995 * 1e6);
         assertEq(usdc.balanceOf(sender), 0);
         assertEq(usdc.balanceOf(marginPaymasterAddress), 0);
-        // uint256 colAmount = perpsMarketProxy.getCollateralAmount(
-        //     account.accountId(),
-        //     sUSDId
-        // );
-        // assertGt(colAmount, 4 ether);
-        // assertLt(colAmount, 5 ether);
+        uint256 colAmount = perpsMarketProxy.getCollateralAmount(
+            account.accountId(),
+            sUSDId
+        );
+        assertGt(colAmount, 4 ether);
+        assertLt(colAmount, 5 ether);
     }
 
     function testTransferToWalletAndApprove() public {