00-Depositor.t.sol

// SPDX-License-Identifier: UNLICENSED
pragma solidity ^0.8.13;

import {Test, console} from "forge-std/Test.sol";
import {Vault, SmartWalletReceiver} from "../src/01-CallerV1.sol";

contract CallerV1Test is Test {
    address user;
    Vault public vault;

    function setUp() public {
        vault = new Vault();

        user = address(new SmartWalletReceiver());
        vm.deal(user, 10 ether);
        vm.prank(user);
        vault.deposit{value: 10 ether}();
    }

    function test_Depositor() public {
        address attacker = makeAddr("ATTACKER");
        vm.deal(attacker, 1 ether);

        vm.startPrank(user);

        // START OF SOLUTION
        // (You can create any additional contract if needed)

        vault.withdraw();

        // END OF SOLUTION

        vm.stopPrank();

        uint256 userBalanceETH = address(user).balance;
        assertEq(userBalanceETH, 10 ether, "User should get their ETH back");

        uint256 userBalanceInVault = vault.balances(user);
        assertEq(userBalanceInVault, 0, "User shouldn't have balance in the vault");

        uint256 vaultBalanceETH = address(vault).balance;
        assertEq(vaultBalanceETH, 0, "Vault shouldn't have any ETH");
    }
}