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

Merge branch 'test/referrals' of github.com:lens-protocol/core-private into test/referrals

Browse Source
This commit is contained in:
donosonaumczuk
2023-06-23 19:10:06 +01:00
parent 0b96a74607 c727ba96c5
commit d031829c49
1 changed files with 93 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

99
test/LegacyCollectTest.t.sol
View File

@@ -7,13 +7,15 @@ import {MockDeprecatedCollectModule} from 'test/mocks/MockDeprecatedCollectModul
import {ICollectNFT} from 'contracts/interfaces/ICollectNFT.sol';
import {LegacyCollectLib} from 'contracts/libraries/LegacyCollectLib.sol';
import {ILegacyCollectModule} from 'contracts/interfaces/ILegacyCollectModule.sol';
import {ReferralSystemTest} from 'test/ReferralSystem.t.sol';
contract LegacyCollectTest is BaseTest {
contract LegacyCollectTest is BaseTest, ReferralSystemTest {
uint256 pubId;
Types.CollectParams defaultCollectParams;
MockDeprecatedCollectModule mockDeprecatedCollectModule;
TestAccount blockedProfile;
bool skipTest;
event Transfer(address indexed from, address indexed to, uint256 indexed tokenId);
event CollectedLegacy(
@@ -26,13 +28,11 @@ contract LegacyCollectTest is BaseTest {
uint256 timestamp
);
function setUp() public virtual override {
super.setUp();
function setUp() public virtual override(BaseTest, ReferralSystemTest) {
ReferralSystemTest.setUp();
blockedProfile = _loadAccountAs('BLOCKED_PROFILE');
mockDeprecatedCollectModule = new MockDeprecatedCollectModule();
// Create a V1 pub
vm.prank(defaultAccount.owner);
pubId = hub.post(_getDefaultPostParams());
@@ -197,6 +197,93 @@ contract LegacyCollectTest is BaseTest {
return hub.collect(collectParams);
}
function _referralSystem_PrepareOperation(
TestPublication memory target,
uint256[] memory referrerProfileIds,
uint256[] memory referrerPubIds
) internal virtual override {
if (referrerProfileIds.length == 0 && referrerPubIds.length == 0) {
defaultCollectParams.referrerProfileId = 0;
defaultCollectParams.referrerPubId = 0;
} else if (referrerProfileIds.length == 1 && referrerPubIds.length == 1) {
defaultCollectParams.referrerProfileId = referrerProfileIds[0];
defaultCollectParams.referrerPubId = referrerPubIds[0];
} else {
skipTest = true;
}
defaultCollectParams.publicationCollectedProfileId = target.profileId;
defaultCollectParams.publicationCollectedId = target.pubId;
_refreshCachedNonces();
}
function _referralSystem_ExpectRevertsIfNeeded(
TestPublication memory target,
uint256[] memory referrerProfileIds,
uint256[] memory referrerPubIds
) internal virtual override returns (bool) {
if (skipTest) {
return true;
}
Types.Publication memory targetPublication = hub.getPublication(target.profileId, target.pubId);
if (defaultCollectParams.referrerPubId == 0) {
// Cannot pass unverified referrer for LegacyCollect
vm.expectRevert(Errors.InvalidReferrer.selector);
return true;
}
if (defaultCollectParams.referrerProfileId != 0 && defaultCollectParams.referrerPubId != 0) {
if (!_isV1LegacyPub(targetPublication)) {
// Cannot collect V2 publications
vm.expectRevert(Errors.CollectNotAllowed.selector);
return true;
} else {
if (
hub.getPublicationType(
defaultCollectParams.referrerProfileId,
defaultCollectParams.referrerPubId
) != Types.PublicationType.Mirror
) {
vm.expectRevert(Errors.InvalidReferrer.selector);
return true;
}
Types.Publication memory referrerPublication = hub.getPublication(
defaultCollectParams.referrerProfileId,
defaultCollectParams.referrerPubId
);
// A mirror can only be a referrer of a legacy publication if it is pointing to it.
if (
referrerPublication.pointedProfileId != target.profileId ||
referrerPublication.pointedPubId != target.pubId
) {
vm.expectRevert(Errors.InvalidReferrer.selector);
return true;
}
}
}
return false;
}
function _referralSystem_ExecutePreparedOperation() internal virtual override {
console.log(
'LEGACY COLLECTING: (%s, %s)',
defaultCollectParams.publicationCollectedProfileId,
defaultCollectParams.publicationCollectedId
);
console.log(
' with referrer: (%s, %s)',
defaultCollectParams.referrerProfileId,
defaultCollectParams.referrerPubId
);
if (skipTest) {
console.log(' ^^^ SKIPPED ^^^');
return;
}
_collect(defaultAccount.ownerPk, defaultCollectParams);
}
function _refreshCachedNonces() internal virtual {
// Nothing to do there.
}