@@ -58,13 +58,17 @@ contract SpokePoolOverloadedDeprecatedMethodsTest is Test {
5858 mockWETH.deposit { value: depositAmount }();
5959 mockWETH.approve (address (spokePool), depositAmount);
6060 vm.stopPrank ();
61+
62+ // Assert that the spokePool balance is zero at the start
63+ assertEq (mockWETH.balanceOf (address (spokePool)), 0 , "SpokePool balance should be zero at setup " );
6164 }
6265
6366 function testDeprecatedDeposit () public {
6467 // Here, we are calling the deprecated deposit method, as defined in the deprecated interface. This should, in
6568 // theory, collide with the function selector depositDeprecated_5947912356, thereby calling the legacy deposit
6669 // method on the spoke pool, while using the old old deposit function signature.
6770 vm.startPrank (depositor);
71+
6872 DeprecatedSpokePoolInterface (address (spokePool)).deposit (
6973 depositor, // recipient
7074 address (mockWETH), // originToken
@@ -76,6 +80,8 @@ contract SpokePoolOverloadedDeprecatedMethodsTest is Test {
7680 0 // maxCount
7781 );
7882
83+ assertEq (mockWETH.balanceOf (address (spokePool)), depositAmount, "SpokePool balance should increase " );
84+
7985 // Test depositing native ETH directly
8086 DeprecatedSpokePoolInterface (address (spokePool)).deposit { value: depositAmount }(
8187 depositor, // recipient
@@ -88,10 +94,13 @@ contract SpokePoolOverloadedDeprecatedMethodsTest is Test {
8894 0 // maxCount
8995 );
9096 vm.stopPrank ();
97+
98+ assertEq (mockWETH.balanceOf (address (spokePool)), depositAmount * 2 , "SpokePool balance should increase " );
9199 }
92100
93101 function testBytes32Deposit () public {
94102 vm.prank (depositor);
103+
95104 // Show the bytes32 variant of the new deposit method works.
96105 spokePool.deposit (
97106 address (depositor).toBytes32 (), // depositor
@@ -107,6 +116,8 @@ contract SpokePoolOverloadedDeprecatedMethodsTest is Test {
107116 0 , // exclusivityParameter
108117 bytes ("" ) // message
109118 );
119+
120+ assertEq (mockWETH.balanceOf (address (spokePool)), depositAmount, "SpokePool balance should increase " );
110121 }
111122
112123 function testAddressDeposit () public {
@@ -126,5 +137,7 @@ contract SpokePoolOverloadedDeprecatedMethodsTest is Test {
126137 0 , // exclusivityParameter
127138 bytes ("" ) // message
128139 );
140+
141+ assertEq (mockWETH.balanceOf (address (spokePool)), depositAmount, "SpokePool balance should increase " );
129142 }
130143}
0 commit comments