Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update ERC7765Example.t.sol #1

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Update ERC7765Example.t.sol #1

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
97 changes: 36 additions & 61 deletions test/examples/ERC7765Example.t.sol
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -9,9 +9,9 @@ contract ERC7765ExampleTest is Test {
MultiplePrivilegeManagement private example;
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Main

address private owner;
address private user1;
uint256 private tokenId = 1;
uint256 private privilegeId = 1;
uint256 private unknownPrivilegeId = 2;
uint256 private constant TOKEN_ID = 1;
uint256 private constant PRIVILEGE_ID = 1;
uint256 private constant UNKNOWN_PRIVILEGE_ID = 2;

function setUp() public {
owner = address(this);
Expand All @@ -20,35 +20,28 @@ contract ERC7765ExampleTest is Test {
}

function testReleasePrivilege() public {

example.releasePrivilege(
"Test Privilege",
"Privilege for testing",
block.timestamp + 1000,
1,
100
);

(
uint256 id,
string memory name,,
// string memory description,
uint256 expiration,,,
// uint256 tokenIdRangeStart,
// uint256 tokenIdRangeEnd,
// uint256 exerciseCount
) = example.privileges(privilegeId);
) = example.privileges(PRIVILEGE_ID);

assertEq(id, privilegeId);
assertEq(id, PRIVILEGE_ID);
assertEq(name, "Test Privilege");
assertTrue(expiration > block.timestamp);
}

function testExercisePrivilegeInvalidId() public {
// First, mint a token to the user
example.safeMint(user1, tokenId);

// Release a privilege
example.safeMint(user1, TOKEN_ID);

example.releasePrivilege(
"Test Privilege",
"Privilege for testing",
Expand All @@ -59,16 +52,13 @@ contract ERC7765ExampleTest is Test {

vm.expectRevert("The privilege does not exist");

// Act as user and exercise privilege
vm.prank(user1);
example.exercisePrivilege(user1, tokenId, unknownPrivilegeId, "");
example.exercisePrivilege(user1, TOKEN_ID, UNKNOWN_PRIVILEGE_ID, "");
}

function testExercisePrivilegeInvalidTo() public {
// First, mint a token to the user
example.safeMint(user1, tokenId);

// Release a privilege
example.safeMint(user1, TOKEN_ID);

example.releasePrivilege(
"Test Privilege",
"Privilege for testing",
Expand All @@ -78,17 +68,14 @@ contract ERC7765ExampleTest is Test {
);

vm.expectRevert("The privilege does not exist");

// Act as user and exercise privilege

vm.prank(owner);
example.exercisePrivilege(owner, tokenId, unknownPrivilegeId, "");
example.exercisePrivilege(owner, TOKEN_ID, UNKNOWN_PRIVILEGE_ID, "");
}

function testExercisePrivilegeInvalidTokenId() public {
// First, mint a token to the user
example.safeMint(user1, tokenId);

// Release a privilege
example.safeMint(user1, TOKEN_ID);

example.releasePrivilege(
"Test Privilege",
"Privilege for testing",
Expand All @@ -99,16 +86,13 @@ contract ERC7765ExampleTest is Test {

vm.expectRevert("The _tokenId is out of range");

// Act as user and exercise privilege
vm.prank(user1);
example.exercisePrivilege(user1, tokenId, privilegeId, "");
example.exercisePrivilege(user1, TOKEN_ID, PRIVILEGE_ID, "");
}

function testExercisePrivilegeInvalidExpiration() public {
// First, mint a token to the user
example.safeMint(user1, tokenId);

// Release a privilege
example.safeMint(user1, TOKEN_ID);

example.releasePrivilege(
"Test Privilege",
"Privilege for testing",
Expand All @@ -119,16 +103,13 @@ contract ERC7765ExampleTest is Test {

vm.expectRevert("The privilege has expired");

// Act as user and exercise privilege
vm.prank(user1);
example.exercisePrivilege(user1, tokenId, privilegeId, "");
example.exercisePrivilege(user1, TOKEN_ID, PRIVILEGE_ID, "");
}

function testExercisePrivilegeDuplicate() public {
// First, mint a token to the user
example.safeMint(user1, tokenId);

// Release a privilege
example.safeMint(user1, TOKEN_ID);

example.releasePrivilege(
"Test Privilege",
"Privilege for testing",
Expand All @@ -137,21 +118,18 @@ contract ERC7765ExampleTest is Test {
100
);

// Act as user and exercise privilege
vm.prank(user1);
example.exercisePrivilege(user1, tokenId, privilegeId, "");
example.exercisePrivilege(user1, TOKEN_ID, PRIVILEGE_ID, "");

vm.expectRevert("You had exercised this privilege");

vm.prank(user1);
example.exercisePrivilege(user1, tokenId, privilegeId, "");
example.exercisePrivilege(user1, TOKEN_ID, PRIVILEGE_ID, "");
}

function testExercisePrivilege() public {
// First, mint a token to the user
example.safeMint(user1, tokenId);

// Release a privilege
example.safeMint(user1, TOKEN_ID);

example.releasePrivilege(
"Test Privilege",
"Privilege for testing",
Expand All @@ -160,21 +138,18 @@ contract ERC7765ExampleTest is Test {
100
);

// Act as user and exercise privilege
vm.prank(user1);
example.exercisePrivilege(user1, tokenId, privilegeId, "");
example.exercisePrivilege(user1, TOKEN_ID, PRIVILEGE_ID, "");

// Check if the privilege is exercised
address exercisedBy = example.privilegeExercisedInfo(tokenId, privilegeId);
address exercisedBy = example.privilegeExercisedInfo(TOKEN_ID, PRIVILEGE_ID);
assertEq(exercisedBy, user1);

// Check if the privilege has been marked as exercised
bool exercisable = example.isExercisable(user1, tokenId, privilegeId);
bool exercisable = example.isExercisable(user1, TOKEN_ID, PRIVILEGE_ID);
assertFalse(exercisable);
}

function testIsExercised() public {
example.safeMint(user1, tokenId);
example.safeMint(user1, TOKEN_ID);
example.releasePrivilege(
"Test Privilege",
"Privilege for testing",
Expand All @@ -184,14 +159,14 @@ contract ERC7765ExampleTest is Test {
);

vm.prank(user1);
example.exercisePrivilege(user1, tokenId, privilegeId, "");
example.exercisePrivilege(user1, TOKEN_ID, PRIVILEGE_ID, "");

bool exercised = example.isExercised(user1, tokenId, privilegeId);
bool exercised = example.isExercised(user1, TOKEN_ID, PRIVILEGE_ID);
assertTrue(exercised);
}

function testGetPrivilegeIds() public {
example.safeMint(user1, tokenId);
example.safeMint(user1, TOKEN_ID);
example.releasePrivilege(
"Test Privilege",
"Privilege for testing",
Expand All @@ -201,11 +176,11 @@ contract ERC7765ExampleTest is Test {
);

vm.prank(user1);
example.exercisePrivilege(user1, tokenId, privilegeId, "");
example.exercisePrivilege(user1, TOKEN_ID, PRIVILEGE_ID, "");

uint256[] memory privileges = example.getPrivilegeIds(tokenId);
uint256[] memory privileges = example.getPrivilegeIds(TOKEN_ID);
assertEq(privileges.length, 1);
assertEq(privileges[0], privilegeId);
assertEq(privileges[0], PRIVILEGE_ID);
}

function testPrivilegeURI() public {
Expand All @@ -217,7 +192,7 @@ contract ERC7765ExampleTest is Test {
100
);

string memory uri = example.privilegeURI(privilegeId);
string memory uri = example.privilegeURI(PRIVILEGE_ID);
assertTrue(bytes(uri).length > 0);
}
}