github/core
1
1
Fork 0
mirror of https://github.com/lens-protocol/core.git synced 2026-04-22 03:02:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

test: Added tests.

Browse Source
This commit is contained in:
Peter Michael
2022-09-21 16:22:37 -04:00
parent 92c49ea7b5
commit f3feae19a1
4 changed files with 100 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/foundry/base/BaseTest.t.sol
View File

@@ -32,7 +32,7 @@ contract BaseTest is TestSetup {
return _calculateDigest(structHash);
}
function _getCollectTypeDataHash(
function _getCollectTypedDataHash(
uint256 profileId,
uint256 pubId,
bytes memory data,

25
test/foundry/base/TestSetup.t.sol
View File

@@ -51,6 +51,8 @@ contract TestSetup is Test {
});
DataTypes.PostData mockPostData;
DataTypes.CommentData mockCommentData;
DataTypes.MirrorData mockMirrorData;
constructor() {
// Start deployments.
@@ -117,6 +119,29 @@ contract TestSetup is Test {
referenceModule: address(0),
referenceModuleInitData: ''
});
// Precompute basic comment data.
mockCommentData = DataTypes.CommentData({
profileId: firstProfileId,
contentURI: mockURI,
profileIdPointed: firstProfileId,
pubIdPointed: 1,
referenceModuleData: '',
collectModule: address(freeCollectModule),
collectModuleInitData: abi.encode(false),
referenceModule: address(0),
referenceModuleInitData: ''
});
// Precompute basic mirror data.
mockMirrorData = DataTypes.MirrorData({
profileId: firstProfileId,
profileIdPointed: firstProfileId,
pubIdPointed: 1,
referenceModuleData: '',
referenceModule: address(0),
referenceModuleInitData: ''
});
}
function setUp() public virtual {

61
test/foundry/interactions/CollectTest.t.sol
View File

@@ -87,7 +87,7 @@ contract CollectTest is BaseTest {
function testCollectWithSigInvalidSignerFails() public {
uint256 nonce = 0;
uint256 deadline = type(uint256).max;
bytes32 digest = _getCollectTypeDataHash(firstProfileId, 1, '', nonce, deadline);
bytes32 digest = _getCollectTypedDataHash(firstProfileId, 1, '', nonce, deadline);
vm.expectRevert(Errors.CallerInvalid.selector);
hub.collectWithSig(
DataTypes.CollectWithSigData({
@@ -104,7 +104,7 @@ contract CollectTest is BaseTest {
function testCollectWithSig() public {
uint256 nonce = 0;
uint256 deadline = type(uint256).max;
bytes32 digest = _getCollectTypeDataHash(firstProfileId, 1, '', nonce, deadline);
bytes32 digest = _getCollectTypedDataHash(firstProfileId, 1, '', nonce, deadline);
uint256 nftId = hub.collectWithSig(
DataTypes.CollectWithSigData({
collector: signer,
@@ -121,7 +121,9 @@ contract CollectTest is BaseTest {
}
function testCollectWithSigMirror() public {
assertEq(hub.getCollectNFT(1, 1), address(0));
uint256 nonce = 0;
uint256 deadline = type(uint256).max;
bytes32 digest = _getCollectTypedDataHash(firstProfileId, 2, '', nonce, deadline);
vm.prank(profileOwner);
hub.mirror(
@@ -135,7 +137,15 @@ contract CollectTest is BaseTest {
})
);
uint256 nftId = hub.collect(me, firstProfileId, 2, '');
uint256 nftId = hub.collectWithSig(
DataTypes.CollectWithSigData({
collector: signer,
profileId: firstProfileId,
pubId: 2,
data: '',
sig: _getSigStruct(signerKey, digest, deadline)
})
);
// Ensure the mirror doesn't have an associated collect NFT.
assertEq(hub.getCollectNFT(firstProfileId, 2), address(0));
@@ -143,7 +153,7 @@ contract CollectTest is BaseTest {
// Ensure the original publication does have an associated collect NFT.
CollectNFT nft = CollectNFT(hub.getCollectNFT(1, 1));
assertEq(nftId, 1);
assertEq(nft.ownerOf(1), me);
assertEq(nft.ownerOf(1), signer);
}
function testExecutorCollectWithSig() public {
@@ -152,7 +162,7 @@ contract CollectTest is BaseTest {
uint256 nonce = 0;
uint256 deadline = type(uint256).max;
bytes32 digest = _getCollectTypeDataHash(firstProfileId, 1, '', nonce, deadline);
bytes32 digest = _getCollectTypedDataHash(firstProfileId, 1, '', nonce, deadline);
uint256 nftId = hub.collectWithSig(
DataTypes.CollectWithSigData({
collector: signer,
@@ -167,4 +177,43 @@ contract CollectTest is BaseTest {
assertEq(nftId, 1);
assertEq(nft.ownerOf(1), signer);
}
function testExecutorCollectWithSigMirror() public {
vm.prank(signer);
hub.setDelegatedExecutorApproval(otherSigner, true);
uint256 nonce = 0;
uint256 deadline = type(uint256).max;
bytes32 digest = _getCollectTypedDataHash(firstProfileId, 2, '', nonce, deadline);
vm.prank(profileOwner);
hub.mirror(
DataTypes.MirrorData({
profileId: firstProfileId,
profileIdPointed: firstProfileId,
pubIdPointed: 1,
referenceModuleData: '',
referenceModule: address(0),
referenceModuleInitData: ''
})
);
uint256 nftId = hub.collectWithSig(
DataTypes.CollectWithSigData({
collector: signer,
profileId: firstProfileId,
pubId: 2,
data: '',
sig: _getSigStruct(otherSignerKey, digest, deadline)
})
);
// Ensure the mirror doesn't have an associated collect NFT.
assertEq(hub.getCollectNFT(firstProfileId, 2), address(0));
// Ensure the original publication does have an associated collect NFT.
CollectNFT nft = CollectNFT(hub.getCollectNFT(1, 1));
assertEq(nftId, 1);
assertEq(nft.ownerOf(1), signer);
}
}

21
test/foundry/interactions/PublishingTest.t.sol
View File

@@ -5,9 +5,26 @@ import '../base/BaseTest.t.sol';
contract PublishingTest is BaseTest {
// negatives
function testPostCallerInvalidFails() public {
vm.prank(otherUser);
function testPostNotExecutorFails() public {
vm.expectRevert(Errors.CallerInvalid.selector);
hub.post(mockPostData);
}
function testCommentNotExecutorFails() public {
vm.prank(profileOwner);
hub.post(mockPostData);
vm.expectRevert(Errors.CallerInvalid.selector);
hub.comment(mockCommentData);
}
function testMirrorNotExecutorFails() public {
vm.prank(profileOwner);
hub.post(mockPostData);
vm.expectRevert(Errors.CallerInvalid.selector);
hub.mirror(mockMirrorData);
}
// positives
}