From c64a86befb3aff2b89ce818391ebddbae4d22427 Mon Sep 17 00:00:00 2001
From: lucas-manuel <lucasmanuel.tech@gmail.com>
Date: Fri, 10 Nov 2023 11:25:47 -0500
Subject: [PATCH] fix: update to use specified caller, uniform tests

---
 test/SparkLendFreezerMom.t.sol | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/test/SparkLendFreezerMom.t.sol b/test/SparkLendFreezerMom.t.sol
index 49875dc..2f917ba 100644
--- a/test/SparkLendFreezerMom.t.sol
+++ b/test/SparkLendFreezerMom.t.sol
@@ -45,7 +45,7 @@ contract ConstructorTests is SparkLendFreezerMomUnitTestBase {
 
 contract SetOwnerTests is SparkLendFreezerMomUnitTestBase {
 
-    function test_setOwner_no_auth() public {
+    function test_setOwner_noAuth() public {
         vm.expectRevert("SparkLendFreezerMom/only-owner");
         freezer.setOwner(address(1));
     }
@@ -64,7 +64,7 @@ contract SetOwnerTests is SparkLendFreezerMomUnitTestBase {
 
 contract SetAuthorityTests is SparkLendFreezerMomUnitTestBase {
 
-    function test_setAuthority_no_auth() public {
+    function test_setAuthority_noAuth() public {
         vm.expectRevert("SparkLendFreezerMom/only-owner");
         freezer.setAuthority(makeAddr("newAuthority"));
     }
@@ -89,8 +89,10 @@ contract FreezeAllMarketsTests is SparkLendFreezerMomUnitTestBase {
     }
 
     function test_freezeAllMarkets() public {
+        address caller = makeAddr("caller");
+
         authority.__setCanCall(
-            address(this),
+            caller,
             address(freezer),
             freezer.freezeAllMarkets.selector,
             true
@@ -105,6 +107,7 @@ contract FreezeAllMarketsTests is SparkLendFreezerMomUnitTestBase {
         bytes4 poolSig   = PoolMock.getReservesList.selector;
         bytes4 configSig = ConfiguratorMock.setReserveFreeze.selector;
 
+        vm.prank(caller);
         vm.expectCall(pool,         abi.encodePacked(poolSig));
         vm.expectCall(configurator, abi.encodePacked(configSig, abi.encode(asset1, true)));
         vm.expectCall(configurator, abi.encodePacked(configSig, abi.encode(asset2, true)));
@@ -123,8 +126,10 @@ contract FreezeMarketTests is SparkLendFreezerMomUnitTestBase {
     }
 
     function test_freezeMarket() public {
+        address caller = makeAddr("caller");
+
         authority.__setCanCall(
-            address(this),
+            caller,
             address(freezer),
             freezer.freezeMarket.selector,
             true
@@ -132,6 +137,7 @@ contract FreezeMarketTests is SparkLendFreezerMomUnitTestBase {
 
         bytes4 configSig = ConfiguratorMock.setReserveFreeze.selector;
 
+        vm.prank(caller);
         vm.expectCall(configurator, abi.encodePacked(configSig, abi.encode(reserve, true)));
         freezer.freezeMarket(reserve);
     }