From 1abdf8893c221fbd9d4e31c91273aa0294da170c Mon Sep 17 00:00:00 2001
From: Aadi Desai <21363892+supleed2@users.noreply.github.com>
Date: Sun, 19 Feb 2023 00:24:52 +0000
Subject: [PATCH] Add PurchaseTokenTests

---
 test/PurchaseToken.t.sol | 117 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 117 insertions(+)
 create mode 100644 test/PurchaseToken.t.sol

diff --git a/test/PurchaseToken.t.sol b/test/PurchaseToken.t.sol
new file mode 100644
index 0000000..363c955
--- /dev/null
+++ b/test/PurchaseToken.t.sol
@@ -0,0 +1,117 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity ^0.8.10;
+
+import "forge-std/Test.sol";
+import "../src/contracts/PurchaseToken.sol";
+
+contract PurchaseTokenTest is Test {
+    event Transfer(address indexed _from, address indexed _to, uint256 _value);
+    event Approval(
+        address indexed _owner,
+        address indexed _spender,
+        uint256 _value
+    );
+
+    PurchaseToken public token;
+
+    address public alice = makeAddr("alice");
+    address public bob = makeAddr("bob");
+    address public charlie = makeAddr("charlie");
+
+    function setUp() public {
+        token = new PurchaseToken();
+    }
+
+    function testName() public {
+        assertEq(token.name(), "PurchaseToken");
+    }
+
+    function testSymbol() public {
+        assertEq(token.symbol(), "PT");
+    }
+
+    function testDecimals() public {
+        assertEq(token.decimals(), 18);
+    }
+
+    function testMint() public {
+        vm.deal(alice, 1 ether);
+        assertEq(token.balanceOf(alice), 0);
+        assertEq(token.totalSupply(), 0);
+        vm.prank(alice);
+        vm.expectEmit(true, true, true, false);
+        emit Transfer(address(0), alice, 1e19);
+        token.mint{value: 1e17}();
+        assertEq(token.balanceOf(alice), 1e19);
+        assertEq(token.totalSupply(), 1e19);
+    }
+
+    function testMintInsufficientFunds() public {
+        vm.deal(alice, 0.01 ether);
+        assertEq(token.balanceOf(alice), 0);
+        assertEq(token.totalSupply(), 0);
+        vm.prank(alice);
+        vm.expectRevert();
+        token.mint{value: 1e17}();
+    }
+
+    function testTransfer() public {
+        vm.deal(alice, 1 ether);
+        vm.prank(alice);
+        token.mint{value: 1e17}();
+        vm.prank(alice);
+        vm.expectEmit(true, true, false, true);
+        emit Transfer(alice, bob, 1e18);
+        token.transfer(bob, 1e18);
+        assertEq(token.balanceOf(alice), 9e18);
+        assertEq(token.balanceOf(bob), 1e18);
+    }
+
+    function testTransferInsufficientFunds() public {
+        vm.deal(alice, 1 ether);
+        vm.prank(alice);
+        token.mint{value: 1e17}();
+        vm.prank(alice);
+        vm.expectRevert("ERC20: transfer amount exceeds balance");
+        token.transfer(bob, 2e19);
+    }
+
+    function testApprove() public {
+        vm.prank(alice);
+        vm.expectEmit(true, true, false, true);
+        emit Approval(alice, bob, 1e18);
+        token.approve(bob, 1e18);
+        assertEq(token.allowance(alice, bob), 1e18);
+    }
+
+    function testTransferFrom() public {
+        vm.deal(alice, 1 ether);
+        vm.prank(alice);
+        vm.expectEmit(true, true, true, false);
+        emit Transfer(address(0), alice, 1e19);
+        token.mint{value: 1e17}();
+        vm.prank(alice);
+        token.approve(bob, 3e18);
+        vm.prank(bob);
+        vm.expectEmit(true, true, false, true);
+        emit Transfer(alice, charlie, 1e18);
+        token.transferFrom(alice, charlie, 1e18);
+        assertEq(token.balanceOf(alice), 9e18);
+        assertEq(token.allowance(alice, bob), 2e18);
+        assertEq(token.balanceOf(bob), 0);
+        assertEq(token.balanceOf(charlie), 1e18);
+    }
+
+    function testTransferFromInsufficientAllowance() public {
+        vm.deal(alice, 1 ether);
+        vm.prank(alice);
+        vm.expectEmit(true, true, true, false);
+        emit Transfer(address(0), alice, 1e19);
+        token.mint{value: 1e17}();
+        vm.prank(alice);
+        token.approve(bob, 3e18);
+        vm.prank(bob);
+        vm.expectRevert("ERC20: insufficient allowance");
+        token.transferFrom(alice, charlie, 4e18);
+    }
+}