From 923d0ff04ed43c289c4f3c23c72eadc46f2b6335 Mon Sep 17 00:00:00 2001
From: lza_menace <lza_menace@protonmail.com>
Date: Thu, 17 Nov 2022 16:19:42 -0800
Subject: [PATCH] start more tests

---
 test/SendIt.t.sol | 26 +++++++++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/test/SendIt.t.sol b/test/SendIt.t.sol
index 4192d13..dad3e70 100644
--- a/test/SendIt.t.sol
+++ b/test/SendIt.t.sol
@@ -48,8 +48,32 @@ contract SendItTest is Test {
     }
 
     function testUpdateVault() public {
+        vm.prank(address(1));
         sendit.updateVault(address(3));
-        assertEq(sendit.addressVault(address(2)), address(0));
+        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