Fixed InsufficientBalance VRF Mock

test/unit/RaffleTest.t.sol

@@ -1,7 +1,7 @@
 // SPDX-License-Identifier: MIT
 pragma solidity 0.8.19;
 
-import {Test, console2} from "forge-std/Test.sol";
+import {Test, console2, console} from "forge-std/Test.sol";
 import {DeployRaffle} from "script/DeployRaffle.s.sol";
 import {Raffle} from "contracts/Raffle.sol";
 import {HelperConfig, CodeConstants} from "script/HelperConfig.s.sol";
@@ -32,11 +32,14 @@ contract RaffleTest is Test, CodeConstants {
     address public PLAYER = makeAddr("player");
     uint256 public constant STARTING_USER_BALANCE = 100 ether;
     uint256 public constant STARTING_PLAYER_BALANCE = 10 ether;
+    uint256 public constant LINK_BALANCE = 100 ether;
 
-    function setUp() public {
+    function setUp() external {
         DeployRaffle deployer = new DeployRaffle();
 
         (raffle, helperConfig) = deployer.run();
+        vm.deal(PLAYER, STARTING_USER_BALANCE);
+
         HelperConfig.NetworkConfig memory config = helperConfig.getConfig();
         vrfCoordinatorV2_5 = config.vrfCoordinatorV2_5;
         subscriptionId = config.subscriptionId;
@@ -44,8 +47,26 @@ contract RaffleTest is Test, CodeConstants {
         interval = config.automationUpdateInterval;
         entranceFee = config.raffleEntranceFee;
         callbackGasLimit = config.callbackGasLimit;
+        link = LinkToken(config.link);
+    }
 
-        vm.deal(PLAYER, STARTING_USER_BALANCE);
+    modifier raffleEntered() {
+        vm.prank(PLAYER);
+        raffle.enterRaffle{value: entranceFee}();
+        vm.warp(block.timestamp + interval + 1);
+        vm.roll(block.number + 1);
+        _;
+    }
+
+    modifier onlyOnDeployedContracts() {
+        if (block.chainid == 31337) {
+            return;
+        }
+        try vm.activeFork() returns (uint256) {
+            return;
+        } catch {
+            _;
+        }
     }
 
     function testRaffleInitializesInOpenState() public view {
@@ -233,24 +254,10 @@ contract RaffleTest is Test, CodeConstants {
     /*//////////////////////////////////////////////////////////////
                            FULFILLRANDOMWORDS
     //////////////////////////////////////////////////////////////*/
-    modifier raffleEntered() {
-        vm.prank(PLAYER);
-        raffle.enterRaffle{value: entranceFee}();
-        vm.warp(block.timestamp + interval + 1);
-        vm.roll(block.number + 1);
-        _;
-    }
-
-    modifier skipFork() {
-        if (block.chainid != 31337) {
-            return;
-        }
-        _;
-    }
 
     function testFulfillRandomWordsCanOnlyBeCalledAfterPerformUpkeep(
         uint256 randomRequestId
-    ) public raffleEntered skipFork {
+    ) public raffleEntered onlyOnDeployedContracts {
         // Arrange
         // Act / Assert
         vm.expectRevert(VRFCoordinatorV2_5Mock.InvalidRequest.selector);
@@ -264,7 +271,11 @@ contract RaffleTest is Test, CodeConstants {
         //..
     }
 
-    function testFulfillRandomWordsPicksAWinnerResetsAndSendsMoney() public raffleEntered skipFork {
+    function testFulfillRandomWordsPicksAWinnerResetsAndSendsMoney()
+        public
+        raffleEntered
+        onlyOnDeployedContracts
+    {
         address expectedWinner = address(1);
 
         // Arrange
@@ -288,6 +299,7 @@ contract RaffleTest is Test, CodeConstants {
         console2.logBytes32(entries[1].topics[1]);
         bytes32 requestId = entries[1].topics[1]; // get the requestId from the logs
 
+        /*
         FundSubscription fundSubscription = new FundSubscription();
         fundSubscription.fundSubscription(
             helperConfig.getConfig().vrfCoordinatorV2_5,
@@ -295,7 +307,7 @@ contract RaffleTest is Test, CodeConstants {
             helperConfig.getConfig().link,
             helperConfig.getConfig().myAccount
         );
-
+*/
         VRFCoordinatorV2_5Mock(vrfCoordinatorV2_5).fulfillRandomWords(
             uint256(requestId),
             address(raffle)
@@ -313,4 +325,45 @@ contract RaffleTest is Test, CodeConstants {
         assert(winnerBalance == startingBalance + prize);
         assert(endingTimeStamp > startingTimeStamp);
     }
+
+    function testHoelee() public raffleEntered onlyOnDeployedContracts {
+        address expectedWinner = address(1);
+
+        // Arrange
+        uint256 additionalEntrances = 3;
+        uint256 startingIndex = 1; // We have starting index be 1 so we can start with address(1) and not address(0)
+
+        for (uint256 i = startingIndex; i < startingIndex + additionalEntrances; i++) {
+            address player = address(uint160(i));
+            hoax(player, 1 ether); // deal 1 eth to the player
+            raffle.enterRaffle{value: entranceFee}();
+        }
+
+        uint256 startingTimeStamp = raffle.getLastTimeStamp();
+        uint256 startingBalance = expectedWinner.balance;
+
+        // Act
+        vm.recordLogs();
+        raffle.performUpkeep(""); // emits requestId
+        Vm.Log[] memory entries = vm.getRecordedLogs();
+        console2.logBytes32(entries[1].topics[1]);
+        bytes32 requestId = entries[1].topics[1]; // get the requestId from the logs
+
+        VRFCoordinatorV2_5Mock(vrfCoordinatorV2_5).fulfillRandomWords(
+            uint256(requestId),
+            address(raffle)
+        );
+
+        // Assert
+        address recentWinner = raffle.getRecentWinner();
+        Raffle.RaffleState raffleState = raffle.getRaffleState();
+        uint256 winnerBalance = recentWinner.balance;
+        uint256 endingTimeStamp = raffle.getLastTimeStamp();
+        uint256 prize = entranceFee * (additionalEntrances + 1);
+
+        assert(recentWinner == expectedWinner);
+        assert(uint256(raffleState) == 0);
+        assert(winnerBalance == startingBalance + prize);
+        assert(endingTimeStamp > startingTimeStamp);
+    }
 }