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

test: PublicActProxy initial scenario tests

Browse Source
This commit is contained in:
vicnaum
2023-09-25 23:36:42 +02:00
parent 4b4fc6bc6b
commit 6156b00961
1 changed files with 209 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

209
test/misc/PublicActProxyTest.t.sol Normal file
View File

@@ -0,0 +1,209 @@
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.13;
import 'test/base/BaseTest.t.sol';
import {PublicActProxy} from 'contracts/misc/PublicActProxy.sol';
import {Types} from 'contracts/libraries/constants/Types.sol';
import {MockCollectModule} from 'test/mocks/MockCollectModule.sol';
import {CollectPublicationAction} from 'contracts/modules/act/collect/CollectPublicationAction.sol';
import {CollectNFT} from 'contracts/modules/act/collect/CollectNFT.sol';
import {SimpleFeeCollectModule} from 'contracts/modules/act/collect/SimpleFeeCollectModule.sol';
import {BaseFeeCollectModuleInitData} from 'contracts/modules/interfaces/IBaseFeeCollectModule.sol';
import {MockCurrency} from 'test/mocks/MockCurrency.sol';
contract PublicActProxyTest is BaseTest {
using stdJson for string;
PublicActProxy publicActProxy;
uint256 defaultPubId;
Types.PublicationActionParams collectActionParams;
address payer;
uint256 payerPk;
address nftRecipient = makeAddr('NFT_RECIPIENT');
CollectPublicationAction collectPublicationAction;
TestAccount publicProfile;
function setUp() public override {
super.setUp();
(payer, payerPk) = makeAddrAndKey('PAYER');
(, address collectPublicationActionAddr) = loadOrDeploy_CollectPublicationAction();
collectPublicationAction = CollectPublicationAction(collectPublicationActionAddr);
if (fork && keyExists(json, string(abi.encodePacked('.', forkEnv, '.PublicActProxy')))) {
publicActProxy = PublicActProxy(
json.readAddress(string(abi.encodePacked('.', forkEnv, '.PublicActProxy')))
);
} else {
console.log('PublicActProxy key does not exist');
publicActProxy = new PublicActProxy(address(hub), address(collectPublicationAction));
}
publicProfile = _loadAccountAs('PUBLIC_PROFILE');
vm.prank(publicProfile.owner);
hub.changeDelegatedExecutorsConfig(
publicProfile.profileId,
_toAddressArray(address(publicActProxy)),
_toBoolArray(true)
);
}
function testCanPublicFreeAct() public {
address mockCollectModule = address(new MockCollectModule());
collectPublicationAction.registerCollectModule(mockCollectModule);
Types.PostParams memory postParams = _getDefaultPostParams();
postParams.actionModules[0] = address(collectPublicationAction);
postParams.actionModulesInitDatas[0] = abi.encode(mockCollectModule, abi.encode(true));
vm.prank(defaultAccount.owner);
defaultPubId = hub.post(postParams);
collectActionParams = Types.PublicationActionParams({
publicationActedProfileId: defaultAccount.profileId,
publicationActedId: defaultPubId,
actorProfileId: publicProfile.profileId,
referrerProfileIds: _emptyUint256Array(),
referrerPubIds: _emptyUint256Array(),
actionModuleAddress: address(collectPublicationAction),
actionModuleData: abi.encode(nftRecipient, abi.encode(true))
});
vm.prank(payer);
publicActProxy.publicFreeAct(collectActionParams);
CollectNFT collectNFT = CollectNFT(
CollectPublicationAction(collectPublicationAction)
.getCollectData(defaultAccount.profileId, defaultPubId)
.collectNFT
);
assertTrue(collectNFT.balanceOf(nftRecipient) > 0, 'NFT recipient balance is 0');
}
function testCanPublicCollect() public {
vm.prank(deployer);
address simpleFeeCollectModule = address(
new SimpleFeeCollectModule(address(hub), address(collectPublicationAction))
);
collectPublicationAction.registerCollectModule(simpleFeeCollectModule);
MockCurrency currency = new MockCurrency();
currency.mint(payer, 10 ether);
BaseFeeCollectModuleInitData memory exampleInitData;
exampleInitData.amount = 1 ether;
exampleInitData.collectLimit = 0;
exampleInitData.currency = address(currency);
exampleInitData.referralFee = 0;
exampleInitData.followerOnly = false;
exampleInitData.endTimestamp = 0;
exampleInitData.recipient = defaultAccount.owner;
Types.PostParams memory postParams = _getDefaultPostParams();
postParams.actionModules[0] = address(collectPublicationAction);
postParams.actionModulesInitDatas[0] = abi.encode(simpleFeeCollectModule, abi.encode(exampleInitData));
vm.prank(defaultAccount.owner);
defaultPubId = hub.post(postParams);
collectActionParams = Types.PublicationActionParams({
publicationActedProfileId: defaultAccount.profileId,
publicationActedId: defaultPubId,
actorProfileId: publicProfile.profileId,
referrerProfileIds: _emptyUint256Array(),
referrerPubIds: _emptyUint256Array(),
actionModuleAddress: address(collectPublicationAction),
actionModuleData: abi.encode(nftRecipient, abi.encode(currency, exampleInitData.amount))
});
vm.startPrank(payer);
currency.approve(address(publicActProxy), exampleInitData.amount);
publicActProxy.publicCollect(collectActionParams);
vm.stopPrank();
CollectNFT collectNFT = CollectNFT(
CollectPublicationAction(collectPublicationAction)
.getCollectData(defaultAccount.profileId, defaultPubId)
.collectNFT
);
assertTrue(collectNFT.balanceOf(nftRecipient) > 0, 'NFT recipient balance is 0');
}
function testCanPublicCollectWithSig() public {
vm.prank(deployer);
address simpleFeeCollectModule = address(
new SimpleFeeCollectModule(address(hub), address(collectPublicationAction))
);
collectPublicationAction.registerCollectModule(simpleFeeCollectModule);
MockCurrency currency = new MockCurrency();
currency.mint(payer, 10 ether);
BaseFeeCollectModuleInitData memory exampleInitData;
exampleInitData.amount = 1 ether;
exampleInitData.collectLimit = 0;
exampleInitData.currency = address(currency);
exampleInitData.referralFee = 0;
exampleInitData.followerOnly = false;
exampleInitData.endTimestamp = 0;
exampleInitData.recipient = defaultAccount.owner;
Types.PostParams memory postParams = _getDefaultPostParams();
postParams.actionModules[0] = address(collectPublicationAction);
postParams.actionModulesInitDatas[0] = abi.encode(simpleFeeCollectModule, abi.encode(exampleInitData));
vm.prank(defaultAccount.owner);
defaultPubId = hub.post(postParams);
collectActionParams = Types.PublicationActionParams({
publicationActedProfileId: defaultAccount.profileId,
publicationActedId: defaultPubId,
actorProfileId: publicProfile.profileId,
referrerProfileIds: _emptyUint256Array(),
referrerPubIds: _emptyUint256Array(),
actionModuleAddress: address(collectPublicationAction),
actionModuleData: abi.encode(nftRecipient, abi.encode(currency, exampleInitData.amount))
});
vm.prank(payer);
currency.approve(address(publicActProxy), exampleInitData.amount);
domainSeparator = keccak256(
abi.encode(
Typehash.EIP712_DOMAIN,
keccak256('PublicActProxy'),
MetaTxLib.EIP712_DOMAIN_VERSION_HASH,
block.chainid,
address(publicActProxy)
)
);
publicActProxy.publicCollectWithSig({
publicationActionParams: collectActionParams,
signature: _getSigStruct({
pKey: payerPk,
digest: _getActTypedDataHash(collectActionParams, publicActProxy.nonces(payer), type(uint256).max),
deadline: type(uint256).max
})
});
CollectNFT collectNFT = CollectNFT(
CollectPublicationAction(collectPublicationAction)
.getCollectData(defaultAccount.profileId, defaultPubId)
.collectNFT
);
assertTrue(collectNFT.balanceOf(nftRecipient) > 0, 'NFT recipient balance is 0');
}
}