1234567891011121314151617import {IKeyringPair} from '@polkadot/types/types';18import {usingPlaygrounds, expect, itSub} from './util';19import {ICollectionPermissions, NON_EXISTENT_COLLECTION_ID} from './util/playgrounds/types';2021describe('Integration Test ext. Allow list tests', () => {22 let alice: IKeyringPair;23 let bob: IKeyringPair;24 let charlie: IKeyringPair;2526 before(async () => {27 await usingPlaygrounds(async (helper, privateKey) => {28 const donor = await privateKey({url: import.meta.url});29 [alice, bob, charlie] = await helper.arrange.createAccounts([30n, 10n, 10n], donor);30 });31 });3233 describe('Positive', () => {34 itSub('Owner can add address to allow list', async ({helper}) => {35 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});36 37 await helper.nft.addToAllowList(alice, collectionId, {Substrate: bob.address});38 const allowList = await helper.nft.getAllowList(collectionId);39 expect(allowList).to.deep.contain({Substrate: bob.address});40 });4142 itSub('Admin can add address to allow list', async ({helper}) => {43 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});44 await helper.nft.addAdmin(alice, collectionId, {Substrate: bob.address});4546 47 await helper.nft.addToAllowList(bob, collectionId, {Substrate: charlie.address});48 const allowList = await helper.nft.getAllowList(collectionId);49 expect(allowList).to.deep.contain({Substrate: charlie.address});50 });5152 itSub('If address is already added to allow list, nothing happens', async ({helper}) => {53 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});54 await helper.nft.addToAllowList(alice, collectionId, {Substrate: bob.address});55 await helper.nft.addToAllowList(alice, collectionId, {Substrate: bob.address});56 const allowList = await helper.nft.getAllowList(collectionId);57 expect(allowList).to.deep.contain({Substrate: bob.address});58 });59 });6061 describe('Negative', () => {62 itSub('Nobody can add address to allow list of non-existing collection', async ({helper}) => {63 const collectionId = NON_EXISTENT_COLLECTION_ID;64 await expect(helper.nft.addToAllowList(bob, collectionId, {Substrate: charlie.address}))65 .to.be.rejectedWith(/common\.CollectionNotFound/);66 });6768 itSub('Nobody can add address to allow list of destroyed collection', async ({helper}) => {69 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});70 await helper.collection.burn(alice, collectionId);71 await expect(helper.nft.addToAllowList(alice, collectionId, {Substrate: bob.address}))72 .to.be.rejectedWith(/common\.CollectionNotFound/);73 });7475 itSub('Non-privileged user cannot add address to allow list', async ({helper}) => {76 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});77 await expect(helper.nft.addToAllowList(bob, collectionId, {Substrate: charlie.address}))78 .to.be.rejectedWith(/common\.NoPermission/);79 });80 });81});8283describe('Integration Test ext. Remove from Allow List', () => {84 let alice: IKeyringPair;85 let bob: IKeyringPair;86 let charlie: IKeyringPair;8788 before(async () => {89 await usingPlaygrounds(async (helper, privateKey) => {90 const donor = await privateKey({url: import.meta.url});91 [alice, bob, charlie] = await helper.arrange.createAccounts([30n, 10n, 10n], donor);92 });93 });9495 describe('Positive', () => {96 itSub('Owner can remove address from allow list', async ({helper}) => {97 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});98 await helper.nft.addToAllowList(alice, collectionId, {Substrate: bob.address});99100 await helper.collection.removeFromAllowList(alice, collectionId, {Substrate: bob.address});101102 const allowList = await helper.nft.getAllowList(collectionId);103104 expect(allowList).to.not.deep.contain({Substrate: bob.address});105 });106107 itSub('Admin can remove address from allow list', async ({helper}) => {108 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});109 await helper.nft.addAdmin(alice, collectionId, {Substrate: charlie.address});110 await helper.nft.addToAllowList(alice, collectionId, {Substrate: bob.address});111 await helper.collection.removeFromAllowList(charlie, collectionId, {Substrate: bob.address});112113 const allowList = await helper.nft.getAllowList(collectionId);114 expect(allowList).to.not.deep.contain({Substrate: bob.address});115 });116117 itSub('If address is already removed from allow list, nothing happens', async ({helper}) => {118 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});119 await helper.nft.addToAllowList(alice, collectionId, {Substrate: bob.address});120 await helper.collection.removeFromAllowList(alice, collectionId, {Substrate: bob.address});121 const allowListBefore = await helper.nft.getAllowList(collectionId);122 expect(allowListBefore).to.not.deep.contain({Substrate: bob.address});123124 await helper.collection.removeFromAllowList(alice, collectionId, {Substrate: bob.address});125126 const allowListAfter = await helper.nft.getAllowList(collectionId);127 expect(allowListAfter).to.not.deep.contain({Substrate: bob.address});128 });129 });130131 describe('Negative', () => {132 itSub('Non-privileged user cannot remove address from allow list', async ({helper}) => {133 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});134 await helper.nft.addToAllowList(alice, collectionId, {Substrate: charlie.address});135 await expect(helper.collection.removeFromAllowList(charlie, collectionId, {Substrate: charlie.address}))136 .to.be.rejectedWith(/common\.NoPermission/);137138 const allowList = await helper.nft.getAllowList(collectionId);139 expect(allowList).to.deep.contain({Substrate: charlie.address});140 });141142 itSub('Nobody can remove address from allow list of non-existing collection', async ({helper}) => {143 const collectionId = NON_EXISTENT_COLLECTION_ID;144 await expect(helper.collection.removeFromAllowList(bob, collectionId, {Substrate: charlie.address}))145 .to.be.rejectedWith(/common\.CollectionNotFound/);146 });147148 itSub('Nobody can remove address from allow list of deleted collection', async ({helper}) => {149 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});150 await helper.nft.addToAllowList(alice, collectionId, {Substrate: bob.address});151 await helper.collection.burn(alice, collectionId);152153 await expect(helper.collection.removeFromAllowList(alice, collectionId, {Substrate: bob.address}))154 .to.be.rejectedWith(/common\.CollectionNotFound/);155 });156 });157});158159describe('Integration Test ext. Transfer if included in Allow List', () => {160 let alice: IKeyringPair;161 let bob: IKeyringPair;162 let charlie: IKeyringPair;163164 before(async () => {165 await usingPlaygrounds(async (helper, privateKey) => {166 const donor = await privateKey({url: import.meta.url});167 [alice, bob, charlie] = await helper.arrange.createAccounts([30n, 10n, 10n], donor);168 });169 });170171 describe('Positive', () => {172 itSub('If Public Access mode is set to AllowList, tokens can be transferred to a allowlisted address with transfer.', async ({helper}) => {173 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});174 const {tokenId} = await helper.nft.mintToken(alice, {collectionId: collectionId, owner: alice.address});175 await helper.nft.setPermissions(alice, collectionId, {access: 'AllowList'});176 await helper.nft.addToAllowList(alice, collectionId, {Substrate: alice.address});177 await helper.nft.addToAllowList(alice, collectionId, {Substrate: charlie.address});178 await helper.nft.transferToken(alice, collectionId, tokenId, {Substrate: charlie.address});179 const owner = await helper.nft.getTokenOwner(collectionId, tokenId);180 expect(owner.Substrate).to.be.equal(charlie.address);181 });182183 itSub('If Public Access mode is set to AllowList, tokens can be transferred to a allowlisted address with transferFrom.', async ({helper}) => {184 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});185 const {tokenId} = await helper.nft.mintToken(alice, {collectionId: collectionId, owner: alice.address});186 await helper.nft.setPermissions(alice, collectionId, {access: 'AllowList'});187 await helper.nft.addToAllowList(alice, collectionId, {Substrate: alice.address});188 await helper.nft.addToAllowList(alice, collectionId, {Substrate: charlie.address});189 await helper.nft.approveToken(alice, collectionId, tokenId, {Substrate: charlie.address});190191 await helper.nft.transferTokenFrom(alice, collectionId, tokenId, {Substrate: alice.address}, {Substrate: charlie.address});192 const owner = await helper.nft.getTokenOwner(collectionId, tokenId);193 expect(owner.Substrate).to.be.equal(charlie.address);194 });195196 itSub('If Public Access mode is set to AllowList, tokens can be transferred from a allowlisted address with transfer', async ({helper}) => {197 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});198 const {tokenId} = await helper.nft.mintToken(alice, {collectionId: collectionId, owner: alice.address});199 await helper.nft.setPermissions(alice, collectionId, {access: 'AllowList'});200 await helper.nft.addToAllowList(alice, collectionId, {Substrate: alice.address});201 await helper.nft.addToAllowList(alice, collectionId, {Substrate: charlie.address});202203 await helper.nft.transferToken(alice, collectionId, tokenId, {Substrate: charlie.address});204 const owner = await helper.nft.getTokenOwner(collectionId, tokenId);205 expect(owner.Substrate).to.be.equal(charlie.address);206 });207208 itSub('If Public Access mode is set to AllowList, tokens can be transferred from a allowlisted address with transferFrom', async ({helper}) => {209 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});210 const {tokenId} = await helper.nft.mintToken(alice, {collectionId: collectionId, owner: alice.address});211 await helper.nft.setPermissions(alice, collectionId, {access: 'AllowList'});212 await helper.nft.addToAllowList(alice, collectionId, {Substrate: alice.address});213 await helper.nft.addToAllowList(alice, collectionId, {Substrate: charlie.address});214 await helper.nft.approveToken(alice, collectionId, tokenId, {Substrate: charlie.address});215216 await helper.nft.transferTokenFrom(alice, collectionId, tokenId, {Substrate: alice.address}, {Substrate: charlie.address});217 const owner = await helper.nft.getTokenOwner(collectionId, tokenId);218 expect(owner.Substrate).to.be.equal(charlie.address);219 });220 });221222 describe('Negative', () => {223 itSub('If Public Access mode is set to AllowList, tokens can\'t be transferred from a non-allowlisted address with transfer or transferFrom. Test1', async ({helper}) => {224 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});225 const {tokenId} = await helper.nft.mintToken(alice, {collectionId: collectionId, owner: alice.address});226 await helper.nft.setPermissions(alice, collectionId, {access: 'AllowList'});227 await helper.nft.addToAllowList(alice, collectionId, {Substrate: charlie.address});228229 await expect(helper.nft.transferToken(alice, collectionId, tokenId, {Substrate: charlie.address}))230 .to.be.rejectedWith(/common\.AddressNotInAllowlist/);231 });232233 itSub('If Public Access mode is set to AllowList, tokens can\'t be transferred from a non-allowlisted address with transfer or transferFrom. Test2', async ({helper}) => {234 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});235 const {tokenId} = await helper.nft.mintToken(alice, {collectionId: collectionId, owner: alice.address});236 await helper.nft.setPermissions(alice, collectionId, {access: 'AllowList'});237 await helper.nft.addToAllowList(alice, collectionId, {Substrate: alice.address});238 await helper.nft.addToAllowList(alice, collectionId, {Substrate: charlie.address});239 await helper.nft.approveToken(alice, collectionId, tokenId, {Substrate: charlie.address});240 await helper.collection.removeFromAllowList(alice, collectionId, {Substrate: alice.address});241242 await expect(helper.nft.transferToken(alice, collectionId, tokenId, {Substrate: charlie.address}))243 .to.be.rejectedWith(/common\.AddressNotInAllowlist/);244 });245246 itSub('If Public Access mode is set to AllowList, tokens can\'t be transferred to a non-allowlisted address with transfer or transferFrom. Test1', async ({helper}) => {247 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});248 const {tokenId} = await helper.nft.mintToken(alice, {collectionId: collectionId, owner: alice.address});249 await helper.nft.setPermissions(alice, collectionId, {access: 'AllowList'});250 await helper.nft.addToAllowList(alice, collectionId, {Substrate: alice.address});251252 await expect(helper.nft.transferToken(alice, collectionId, tokenId, {Substrate: charlie.address}))253 .to.be.rejectedWith(/common\.AddressNotInAllowlist/);254 });255256 itSub('If Public Access mode is set to AllowList, tokens can\'t be transferred to a non-allowlisted address with transfer or transferFrom. Test2', async ({helper}) => {257 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});258 const {tokenId} = await helper.nft.mintToken(alice, {collectionId: collectionId, owner: alice.address});259 await helper.nft.setPermissions(alice, collectionId, {access: 'AllowList'});260 await helper.nft.addToAllowList(alice, collectionId, {Substrate: alice.address});261 await helper.nft.addToAllowList(alice, collectionId, {Substrate: charlie.address});262263 await helper.nft.approveToken(alice, collectionId, tokenId, {Substrate: charlie.address});264 await helper.collection.removeFromAllowList(alice, collectionId, {Substrate: alice.address});265266 await expect(helper.nft.transferToken(alice, collectionId, tokenId, {Substrate: charlie.address}))267 .to.be.rejectedWith(/common\.AddressNotInAllowlist/);268 });269270 itSub('If Public Access mode is set to AllowList, tokens can\'t be destroyed by a non-allowlisted address (even if it owned them before enabling AllowList mode)', async ({helper}) => {271 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});272 const {tokenId} = await helper.nft.mintToken(alice, {collectionId: collectionId, owner: alice.address});273 await helper.nft.setPermissions(alice, collectionId, {access: 'AllowList'});274 await expect(helper.nft.burnToken(bob, collectionId, tokenId))275 .to.be.rejectedWith(/common\.NoPermission/);276 });277278 itSub('If Public Access mode is set to AllowList, token transfers can\'t be Approved by a non-allowlisted address (see Approve method)', async ({helper}) => {279 const {collectionId} = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});280 const {tokenId} = await helper.nft.mintToken(alice, {collectionId: collectionId, owner: alice.address});281 await helper.nft.setPermissions(alice, collectionId, {access: 'AllowList'});282 await expect(helper.nft.approveToken(alice, collectionId, tokenId, {Substrate: bob.address}))283 .to.be.rejectedWith(/common\.AddressNotInAllowlist/);284 });285 });286});287288describe('Integration Test ext. Mint if included in Allow List', () => {289 let alice: IKeyringPair;290 let bob: IKeyringPair;291292 before(async () => {293 await usingPlaygrounds(async (helper, privateKey) => {294 const donor = await privateKey({url: import.meta.url});295 [alice, bob] = await helper.arrange.createAccounts([100n, 100n], donor);296 });297 });298299 const permissionSet: ICollectionPermissions[] = [300 {access: 'Normal', mintMode: false},301 {access: 'Normal', mintMode: true},302 {access: 'AllowList', mintMode: false},303 {access: 'AllowList', mintMode: true},304 ];305306 const testPermissionSuite = (permissions: ICollectionPermissions) => {307 const allowlistedMintingShouldFail = !permissions.mintMode!;308309 const appropriateRejectionMessage = permissions.mintMode! ? /common\.AddressNotInAllowlist/ : /common\.PublicMintingNotAllowed/;310311 const allowlistedMintingTest = () => itSub(312 `With the condtions above, tokens can${allowlistedMintingShouldFail ? '\'t' : ''} be created by allow-listed addresses`,313 async ({helper}) => {314 const collection = await helper.nft.mintCollection(alice, {});315 await collection.setPermissions(alice, permissions);316 await collection.addToAllowList(alice, {Substrate: bob.address});317318 if(allowlistedMintingShouldFail)319 await expect(collection.mintToken(bob, {Substrate: bob.address})).to.be.rejectedWith(appropriateRejectionMessage);320 else321 await expect(collection.mintToken(bob, {Substrate: bob.address})).to.not.be.rejected;322 },323 );324325326 describe(`Public Access Mode = ${permissions.access}, Mint Mode = ${permissions.mintMode}`, () => {327 describe('Positive', () => {328 itSub('With the condtions above, tokens can be created by owner', async ({helper}) => {329 const collection = await helper.nft.mintCollection(alice, {});330 await collection.setPermissions(alice, permissions);331 await expect(collection.mintToken(alice, {Substrate: alice.address})).to.not.be.rejected;332 });333334 itSub('With the condtions above, tokens can be created by admin', async ({helper}) => {335 const collection = await helper.nft.mintCollection(alice, {});336 await collection.setPermissions(alice, permissions);337 await collection.addAdmin(alice, {Substrate: bob.address});338 await expect(collection.mintToken(bob, {Substrate: bob.address})).to.not.be.rejected;339 });340341 if(!allowlistedMintingShouldFail) allowlistedMintingTest();342 });343344 describe('Negative', () => {345 itSub('With the condtions above, tokens can\'t be created by non-priviliged non-allow-listed address', async ({helper}) => {346 const collection = await helper.nft.mintCollection(alice, {name: 'col', description: 'descr', tokenPrefix: 'COL'});347 await collection.setPermissions(alice, permissions);348 await expect(collection.mintToken(bob, {Substrate: bob.address}))349 .to.be.rejectedWith(appropriateRejectionMessage);350 });351352 if(allowlistedMintingShouldFail) allowlistedMintingTest();353 });354 });355 };356357 for(const permissions of permissionSet) {358 testPermissionSuite(permissions);359 }360});