supleed2/COMP70017-PoDL-CW
1
0
Fork 0
mirror of https://github.com/supleed2/COMP70017-PoDL-CW.git synced 2024-09-19 20:56:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add tests for TicketNFT for full branch coverage

Browse source
This commit is contained in:
Aadi Desai 2023-02-21 15:57:44 +00:00
parent 30fe43eb28
commit 9592b337a0
No known key found for this signature in database
GPG key ID: CFFFE425830EF4D9
1 changed files with 62 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

65
test/TicketNFT.t.sol
View file

@ -89,14 +89,49 @@ contract TicketNFTTest is BaseTicketNFTTest {
assertEq(nft.balanceOf(alice), 0);
}
function testBalanceOf() public {
assertEq(nft.balanceOf(alice), 0);
_buyTicket(alice, "alice");
assertEq(nft.balanceOf(alice), 1);
}
function testHolderOf() public {
_buyTicket(alice, "alice");
assertEq(nft.holderOf(1), alice);
vm.expectRevert("holderOf: ticket doesn't exist");
nft.holderOf(2);
}
function testHolderNameOf() public {
_buyTicket(alice, "alice");
assertEq(nft.holderNameOf(1), "alice");
vm.expectRevert("holderNameOf: ticket doesn't exist");
nft.holderNameOf(2);
}
function testUpdateHolderName() public {
_buyTicket(alice, "alice1");
assertEq(nft.holderNameOf(1), "alice1");
vm.prank(alice);
nft.updateHolderName(1, "alice2");
assertEq(nft.holderNameOf(1), "alice2");
}
function testUpdateHolderNameInvalid() public {
_buyTicket(alice, "alice1");
assertEq(nft.holderNameOf(1), "alice1");
vm.prank(alice);
vm.expectRevert("updateHolderName: ticket doesn't exist");
nft.updateHolderName(2, "alice2");
}
function testUpdateHolderNameUnauthorized() public {
_buyTicket(alice, "alice");
assertEq(nft.holderNameOf(1), "alice");
vm.prank(bob);
vm.expectRevert("updateHolderName: msg.sender doesn't own ticket");
nft.updateHolderName(1, "bob");
assertEq(nft.holderNameOf(1), "alice");
}
function testTransfer() public {
@ -113,9 +148,6 @@ contract TicketNFTTest is BaseTicketNFTTest {
assertEq(nft.getApproved(1), address(0));
assertEq(nft.holderNameOf(1), "alice");
assertEq(nft.isExpiredOrUsed(1), false);
vm.prank(bob);
nft.updateHolderName(1, "bob");
assertEq(nft.holderNameOf(1), "bob");
}
function testTransferInvalid() public {
@ -146,6 +178,13 @@ contract TicketNFTTest is BaseTicketNFTTest {
assertEq(nft.getApproved(1), bob);
}
function testApprovalInvalid() public {
_buyTicket(alice, "alice");
vm.prank(alice);
vm.expectRevert("approve: ticket doesn't exist");
nft.approve(bob, 2);
}
function testApprovalUnauthorized() public {
_buyTicket(alice, "alice");
vm.prank(bob);
@ -154,6 +193,11 @@ contract TicketNFTTest is BaseTicketNFTTest {
assertEq(nft.getApproved(1), address(0));
}
function testGetApprovedInvalid() public {
vm.expectRevert("getApproved: ticket doesn't exist");
nft.getApproved(1);
}
function testTransferFrom() public {
_buyTicket(alice, "alice");
vm.prank(alice);
@ -333,6 +377,21 @@ contract PrimaryMarketTest is BaseTicketNFTTest {
assertEq(nft.holderNameOf(2), "alice's plus one");
assertEq(nft.isExpiredOrUsed(2), false);
}
function testPurchaseLimit() public {
// Set contract totalSupply to 999 using foundry cheats rather than minting repeatedly
vm.store(address(nft), bytes32(uint256(1)), bytes32(uint256(999)));
// for (uint256 i = 0; i < 999; i++) {
// _buyTicket(alice, "alice");
// }
_buyTicket(alice, "alice"); // Buy 1000th ticket
_topUpTokens(alice, 1);
vm.prank(alice);
token.approve(address(nft), ticketPrice);
vm.prank(alice);
vm.expectRevert("purchase: maximum tickets reached");
nft.purchase("alice");
}
}
contract SecondaryMarketTest is BaseTicketNFTTest {