shipit/test/SendIt.t.sol

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

import "forge-std/Test.sol";
import {SendIt} from "../src/SendIt.sol";
import {NFT} from "../src/NFT.sol";

contract SendItTest is Test {
    using stdStorage for StdStorage;
    SendIt public sendit;
    NFT public nft;

    function setUp() public {
        sendit = new SendIt();
        nft = new NFT();
    }

    function testBulkTransferSucceeds() public {
        uint256 amt = 20;
        uint256 fee = sendit.usageFee();
        uint256 val = fee * amt;
        uint256[] memory tokenIndexes = new uint256[](amt);
        address[] memory recipients = new address[](amt);
        for (uint256 i; i < amt; i++) {
            tokenIndexes[i] = i + 1;
            recipients[i] = address(1);
        }
        vm.deal(address(5), 1 ether);
        vm.startPrank(address(5));
        nft.mint(address(5), amt);
        nft.setApprovalForAll(address(sendit), true);
        sendit.contractBulkTransfer{value: val}(
            address(nft),
            tokenIndexes,
            recipients,
            false
        );
        vm.stopPrank();
    }

    function testSingleTransfer() public {
        nft.mint(address(5), 20);
        vm.startPrank(address(5));
        for (uint256 i; i < 20; i++) {
            nft.safeTransferFrom(address(5), address(1), i + 1);
        }
        vm.stopPrank();
    }

    function testUpdateVault() public {
        vm.prank(address(1));
        sendit.updateVault(address(3));
        assertEq(sendit.addressVault(address(1)), address(3));
    }

    function testTokenOwnerCanSend() public {
        uint256 amt = 1;
        uint256 fee = sendit.usageFee();
        uint256 val = fee * amt;
        uint256[] memory tokenIndexes = new uint256[](amt);
        address[] memory recipients = new address[](amt);
        for (uint256 i; i < amt; i++) {
            tokenIndexes[i] = i + 1;
            recipients[i] = address(1);
        }
        vm.deal(address(5), 1 ether);
        vm.startPrank(address(5));
        nft.mint(address(5), amt);
        // vm.stopPrank();
        nft.setApprovalForAll(address(sendit), true);
        sendit.contractBulkTransfer{value: val}(
            address(nft),
            tokenIndexes,
            recipients,
            false
        );
    }

    // only token owner can bulk transfer
    // can only proceed after approval set
    // only owner can updateFee
    // only owner can withdraw
}