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: Foundry Publishing Post tests

Browse Source
This commit is contained in:
vicnaum
2022-10-26 15:08:11 +02:00
parent a216524674
commit 4cfaa55e73
3 changed files with 111 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

104
test/foundry/PublishingTest.t.sol
View File

@@ -22,12 +22,26 @@ contract PublishingTest_Post is BaseTest, SignatureHelpers, PublishingHelpers, S
}
// negatives
function testPostNotExecutorFails() public {
function testCannotPostIfNotExecutor() public {
vm.expectRevert(Errors.ExecutorInvalid.selector);
_post(mockPostData);
}
function testPostWithSigInvalidSignerFails() public {
function testCannotPostNotWhitelistedCollectModule() public {
mockPostData.collectModule = address(0xC0FFEE);
vm.prank(profileOwner);
vm.expectRevert(Errors.CollectModuleNotWhitelisted.selector);
_post(mockPostData);
}
function testCannotPostNotWhitelistedReferenceModule() public {
mockPostData.referenceModule = address(0xC0FFEE);
vm.prank(profileOwner);
vm.expectRevert(Errors.ReferenceModuleNotWhitelisted.selector);
_post(mockPostData);
}
function testCannotPostWithSigInvalidSigner() public {
bytes32 digest = _getPostTypedDataHash(mockPostData, nonce, deadline);
vm.expectRevert(Errors.SignatureInvalid.selector);
@@ -40,7 +54,62 @@ contract PublishingTest_Post is BaseTest, SignatureHelpers, PublishingHelpers, S
);
}
function testPostWithSigNotExecutorFails() public {
function testCannotPostWithSigInvalidNonce() public {
nonce = _getSigNonce(otherSigner) + 1;
bytes32 digest = _getPostTypedDataHash(mockPostData, nonce, deadline);
vm.expectRevert(Errors.SignatureInvalid.selector);
_postWithSig(
_buildPostWithSigData({
delegatedSigner: address(0),
postData: mockPostData,
sig: _getSigStruct(otherSignerKey, digest, deadline)
})
);
}
function testCannotPostIfNonceWasIncrementedWithAnotherAction() public {
assertEq(_getSigNonce(profileOwner), nonce);
bytes32 digest = _getPostTypedDataHash(mockPostData, nonce, deadline);
uint256 pubId = _postWithSig(
_buildPostWithSigData({
delegatedSigner: address(0),
postData: mockPostData,
sig: _getSigStruct(profileOwnerKey, digest, deadline)
})
);
assertEq(pubId, 1);
assert(_getSigNonce(profileOwner) != nonce);
digest = _getPostTypedDataHash(mockPostData, nonce, deadline);
vm.expectRevert(Errors.SignatureInvalid.selector);
_postWithSig(
_buildPostWithSigData({
delegatedSigner: address(0),
postData: mockPostData,
sig: _getSigStruct(profileOwnerKey, digest, deadline)
})
);
}
function testCannotPostWithSigExpiredDeadline() public {
deadline = 10;
vm.warp(20);
bytes32 digest = _getPostTypedDataHash(mockPostData, nonce, deadline);
vm.expectRevert(Errors.SignatureExpired.selector);
_postWithSig(
_buildPostWithSigData({
delegatedSigner: address(0),
postData: mockPostData,
sig: _getSigStruct(otherSignerKey, digest, deadline)
})
);
}
function testCannotPostWithSigNotExecutor() public {
bytes32 digest = _getPostTypedDataHash(mockPostData, nonce, deadline);
vm.expectRevert(Errors.ExecutorInvalid.selector);
@@ -63,6 +132,33 @@ contract PublishingTest_Post is BaseTest, SignatureHelpers, PublishingHelpers, S
_verifyPublication(pub, _expectedPubFromInitData(mockPostData));
}
function testPostWithAWhitelistedReferenceModule() public {
mockPostData.referenceModule = address(mockReferenceModule);
mockPostData.referenceModuleInitData = abi.encode(1);
vm.prank(profileOwner);
uint256 pubId = _post(mockPostData);
assertEq(pubId, 1);
DataTypes.PublicationStruct memory pub = _getPub(firstProfileId, pubId);
_verifyPublication(pub, _expectedPubFromInitData(mockPostData));
}
function testPostWithSig() public {
bytes32 digest = _getPostTypedDataHash(mockPostData, nonce, deadline);
uint256 pubId = _postWithSig(
_buildPostWithSigData({
delegatedSigner: address(0),
postData: mockPostData,
sig: _getSigStruct(profileOwnerKey, digest, deadline)
})
);
assertEq(pubId, 1);
DataTypes.PublicationStruct memory pub = _getPub(firstProfileId, pubId);
_verifyPublication(pub, _expectedPubFromInitData(mockPostData));
}
function testExecutorPost() public {
vm.prank(profileOwner);
_setDelegatedExecutorApproval(otherSigner, true);
@@ -79,8 +175,6 @@ contract PublishingTest_Post is BaseTest, SignatureHelpers, PublishingHelpers, S
vm.prank(profileOwner);
_setDelegatedExecutorApproval(otherSigner, true);
uint256 nonce = 0;
uint256 deadline = type(uint256).max;
bytes32 digest = _getPostTypedDataHash(mockPostData, nonce, deadline);
uint256 pubId = _postWithSig(

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

@@ -369,4 +369,8 @@ contract BaseTest is TestSetup {
{
return hub.getPub(profileId, pubId);
}
function _getSigNonce(address signer) internal view returns (uint256) {
return hub.sigNonces(signer);
}
}

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

@@ -14,6 +14,7 @@ import '../../../contracts/libraries/Errors.sol';
import '../../../contracts/libraries/GeneralLib.sol';
import '../../../contracts/libraries/ProfileTokenURILogic.sol';
import '../../../contracts/mocks/MockCollectModule.sol';
import '../../../contracts/mocks/MockReferenceModule.sol';
contract TestSetup is Test {
uint256 constant firstProfileId = 1;
@@ -37,6 +38,7 @@ contract TestSetup is Test {
TransparentUpgradeableProxy hubAsProxy;
LensHub hub;
MockCollectModule mockCollectModule;
MockReferenceModule mockReferenceModule;
DataTypes.CreateProfileData mockCreateProfileData;
@@ -71,6 +73,9 @@ contract TestSetup is Test {
// Deploy the MockCollectModule.
mockCollectModule = new MockCollectModule();
// Deploy the MockReferenceModule.
mockReferenceModule = new MockReferenceModule();
// End deployments.
vm.stopPrank();
@@ -83,6 +88,9 @@ contract TestSetup is Test {
// Whitelist the FreeCollectModule.
hub.whitelistCollectModule(address(mockCollectModule), true);
// Whitelist the MockReferenceModule.
hub.whitelistReferenceModule(address(mockReferenceModule), true);
// Whitelist the test contract as a profile creator
hub.whitelistProfileCreator(me, true);