123456import { ApiPromise, Keyring } from '@polkadot/api';7import { Enum, Struct } from '@polkadot/types/codec';8import type { AccountId, EventRecord } from '@polkadot/types/interfaces';9import { u128 } from '@polkadot/types/primitive';10import { IKeyringPair } from '@polkadot/types/types';11import { BigNumber } from 'bignumber.js';12import BN from 'bn.js';13import chai from 'chai';14import chaiAsPromised from 'chai-as-promised';15import { alicesPublicKey, nullPublicKey } from '../accounts';16import privateKey from '../substrate/privateKey';17import { default as usingApi, submitTransactionAsync, submitTransactionExpectFailAsync } from '../substrate/substrate-api';18import { ICollectionInterface } from '../types';19import { hexToStr, strToUTF16, utf16ToStr } from './util';2021chai.use(chaiAsPromised);22const expect = chai.expect;2324type GenericResult = {25 success: boolean,26};2728interface CreateCollectionResult {29 success: boolean;30 collectionId: number;31}3233interface CreateItemResult {34 success: boolean;35 collectionId: number;36 itemId: number;37}3839interface IReFungibleOwner {40 Fraction: BN;41 Owner: number[];42}4344interface ITokenDataType {45 Owner: number[];46 ConstData: number[];47 VariableData: number[];48}4950interface IFungibleTokenDataType {51 Value: BN;52}5354export interface IReFungibleTokenDataType {55 Owner: IReFungibleOwner[];56 ConstData: number[];57 VariableData: number[];58}5960export function getGenericResult(events: EventRecord[]): GenericResult {61 const result: GenericResult = {62 success: false,63 };64 events.forEach(({ phase, event: { data, method, section } }) => {65 66 if (method === 'ExtrinsicSuccess') {67 result.success = true;68 }69 });70 return result;71}7273export function getCreateCollectionResult(events: EventRecord[]): CreateCollectionResult {74 let success = false;75 let collectionId: number = 0;76 events.forEach(({ phase, event: { data, method, section } }) => {77 78 if (method == 'ExtrinsicSuccess') {79 success = true;80 } else if ((section == 'nft') && (method == 'Created')) {81 collectionId = parseInt(data[0].toString());82 }83 });84 const result: CreateCollectionResult = {85 success,86 collectionId,87 };88 return result;89}9091export function getCreateItemResult(events: EventRecord[]): CreateItemResult {92 let success = false;93 let collectionId: number = 0;94 let itemId: number = 0;95 events.forEach(({ phase, event: { data, method, section } }) => {96 97 if (method == 'ExtrinsicSuccess') {98 success = true;99 } else if ((section == 'nft') && (method == 'ItemCreated')) {100 collectionId = parseInt(data[0].toString());101 itemId = parseInt(data[1].toString());102 }103 });104 const result: CreateItemResult = {105 success,106 collectionId,107 itemId,108 };109 return result;110}111112interface Invalid {113 type: 'Invalid';114}115116interface Nft {117 type: 'NFT';118}119120interface Fungible {121 type: 'Fungible';122 decimalPoints: number;123}124125interface ReFungible {126 type: 'ReFungible';127}128129type CollectionMode = Nft | Fungible | ReFungible | Invalid;130131export type CreateCollectionParams = {132 mode: CollectionMode,133 name: string,134 description: string,135 tokenPrefix: string,136};137138const defaultCreateCollectionParams: CreateCollectionParams = {139 description: 'description',140 mode: { type: 'NFT' },141 name: 'name',142 tokenPrefix: 'prefix',143}144145export async function createCollectionExpectSuccess(params: Partial<CreateCollectionParams> = {}): Promise<number> {146 const {name, description, mode, tokenPrefix } = {...defaultCreateCollectionParams, ...params};147148 let collectionId: number = 0;149 await usingApi(async (api) => {150 151 const AcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString(), 10);152153 154 const alicePrivateKey = privateKey('//Alice');155156 let modeprm = {};157 if (mode.type === 'NFT') {158 modeprm = {nft: null};159 } else if (mode.type === 'Fungible') {160 modeprm = {fungible: mode.decimalPoints};161 } else if (mode.type === 'ReFungible') {162 modeprm = {refungible: null};163 } else if (mode.type === 'Invalid') {164 modeprm = {invalid: null};165 }166167 const tx = api.tx.nft.createCollection(strToUTF16(name), strToUTF16(description), strToUTF16(tokenPrefix), modeprm);168 const events = await submitTransactionAsync(alicePrivateKey, tx);169 const result = getCreateCollectionResult(events);170171 172 const BcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString(), 10);173174 175 const collection: any = (await api.query.nft.collection(result.collectionId)).toJSON();176177 178 179 expect(result.success).to.be.true;180 expect(result.collectionId).to.be.equal(BcollectionCount);181 182 expect(collection).to.be.not.null;183 expect(BcollectionCount).to.be.equal(AcollectionCount + 1, 'Error: NFT collection NOT created.');184 expect(collection.Owner).to.be.equal(alicesPublicKey);185 expect(utf16ToStr(collection.Name)).to.be.equal(name);186 expect(utf16ToStr(collection.Description)).to.be.equal(description);187 expect(hexToStr(collection.TokenPrefix)).to.be.equal(tokenPrefix);188189 collectionId = result.collectionId;190 });191192 return collectionId;193}194195export async function createCollectionExpectFailure(params: Partial<CreateCollectionParams> = {}) {196 const {name, description, mode, tokenPrefix } = {...defaultCreateCollectionParams, ...params};197198 let modeprm = {};199 if (mode.type === 'NFT') {200 modeprm = {nft: null};201 } else if (mode.type === 'Fungible') {202 modeprm = {fungible: mode.decimalPoints};203 } else if (mode.type === 'ReFungible') {204 modeprm = {refungible: null};205 } else if (mode.type === 'Invalid') {206 modeprm = {invalid: null};207 }208209 await usingApi(async (api) => {210 211 const AcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString());212213 214 const alicePrivateKey = privateKey('//Alice');215 const tx = api.tx.nft.createCollection(strToUTF16(name), strToUTF16(description), strToUTF16(tokenPrefix), modeprm);216 const events = await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;217 const result = getCreateCollectionResult(events);218219 220 const BcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString());221222 223 224 expect(result.success).to.be.false;225 expect(BcollectionCount).to.be.equal(AcollectionCount, 'Error: Collection with incorrect data created.');226 });227}228229export async function findUnusedAddress(api: ApiPromise): Promise<IKeyringPair> {230 let bal = new BigNumber(0);231 let unused;232 do {233 const randomSeed = 'seed' + Math.floor(Math.random() * Math.floor(10000));234 const keyring = new Keyring({ type: 'sr25519' });235 unused = keyring.addFromUri(`//${randomSeed}`);236 bal = new BigNumber((await api.query.system.account(unused.address)).data.free.toString());237 } while (bal.toFixed() != '0');238 return unused;239}240241export async function findNotExistingCollection(api: ApiPromise): Promise<number> {242 const totalNumber = parseInt((await api.query.nft.createdCollectionCount()).toString(), 10) as unknown as number;243 const newCollection: number = totalNumber + 1;244 return newCollection;245}246247function getDestroyResult(events: EventRecord[]): boolean {248 let success: boolean = false;249 events.forEach(({ phase, event: { data, method, section } }) => {250 251 if (method == 'ExtrinsicSuccess') {252 success = true;253 }254 });255 return success;256}257258export async function destroyCollectionExpectFailure(collectionId: number, senderSeed: string = '//Alice') {259 await usingApi(async (api) => {260 261 const alicePrivateKey = privateKey(senderSeed);262 const tx = api.tx.nft.destroyCollection(collectionId);263 await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;264 });265}266267export async function destroyCollectionExpectSuccess(collectionId: number, senderSeed: string = '//Alice') {268 await usingApi(async (api) => {269 270 const alicePrivateKey = privateKey(senderSeed);271 const tx = api.tx.nft.destroyCollection(collectionId);272 const events = await submitTransactionAsync(alicePrivateKey, tx);273 const result = getDestroyResult(events);274275 276 const collection: any = (await api.query.nft.collection(collectionId)).toJSON();277278 279 expect(result).to.be.true;280 expect(collection).to.be.not.null;281 expect(collection.Owner).to.be.equal(nullPublicKey);282 });283}284285export async function setCollectionSponsorExpectSuccess(collectionId: number, sponsor: string) {286 await usingApi(async (api) => {287288 289 const alicePrivateKey = privateKey('//Alice');290 const tx = api.tx.nft.setCollectionSponsor(collectionId, sponsor);291 const events = await submitTransactionAsync(alicePrivateKey, tx);292 const result = getGenericResult(events);293294 295 const collection: any = (await api.query.nft.collection(collectionId)).toJSON();296297 298 expect(result.success).to.be.true;299 expect(collection.Sponsor.toString()).to.be.equal(sponsor.toString());300 expect(collection.SponsorConfirmed).to.be.false;301 });302}303304export async function removeCollectionSponsorExpectSuccess(collectionId: number) {305 await usingApi(async (api) => {306307 308 const alicePrivateKey = privateKey('//Alice');309 const tx = api.tx.nft.removeCollectionSponsor(collectionId);310 const events = await submitTransactionAsync(alicePrivateKey, tx);311 const result = getGenericResult(events);312313 314 const collection: any = (await api.query.nft.collection(collectionId)).toJSON();315316 317 expect(result.success).to.be.true;318 expect(collection.Sponsor).to.be.equal(nullPublicKey);319 expect(collection.SponsorConfirmed).to.be.false;320 });321}322323export async function removeCollectionSponsorExpectFailure(collectionId: number) {324 await usingApi(async (api) => {325326 327 const alicePrivateKey = privateKey('//Alice');328 const tx = api.tx.nft.removeCollectionSponsor(collectionId);329 await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;330 });331}332333export async function setCollectionSponsorExpectFailure(collectionId: number, sponsor: string, senderSeed: string = '//Alice') {334 await usingApi(async (api) => {335336 337 const alicePrivateKey = privateKey(senderSeed);338 const tx = api.tx.nft.setCollectionSponsor(collectionId, sponsor);339 await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;340 });341}342343export async function confirmSponsorshipExpectSuccess(collectionId: number, senderSeed: string = '//Alice') {344 await usingApi(async (api) => {345346 347 const sender = privateKey(senderSeed);348 const tx = api.tx.nft.confirmSponsorship(collectionId);349 const events = await submitTransactionAsync(sender, tx);350 const result = getGenericResult(events);351352 353 const collection: any = (await api.query.nft.collection(collectionId)).toJSON();354355 356 expect(result.success).to.be.true;357 expect(collection.Sponsor).to.be.equal(sender.address);358 expect(collection.SponsorConfirmed).to.be.true;359 });360}361362export async function confirmSponsorshipExpectFailure(collectionId: number, senderSeed: string = '//Alice') {363 await usingApi(async (api) => {364365 366 const sender = privateKey(senderSeed);367 const tx = api.tx.nft.confirmSponsorship(collectionId);368 await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;369 });370}371372export async function enableContractSponsoringExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, enable: boolean) {373 await usingApi(async (api) => {374 const tx = api.tx.nft.enableContractSponsoring(contractAddress, enable);375 const events = await submitTransactionAsync(sender, tx);376 const result = getGenericResult(events);377378 expect(result.success).to.be.true;379 });380}381382export async function enableContractSponsoringExpectFailure(sender: IKeyringPair, contractAddress: AccountId | string, enable: boolean) {383 await usingApi(async (api) => {384 const tx = api.tx.nft.enableContractSponsoring(contractAddress, enable);385 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;386 const result = getGenericResult(events);387388 expect(result.success).to.be.false;389 });390}391392export async function setContractSponsoringRateLimitExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, rateLimit: number) {393 await usingApi(async (api) => {394 const tx = api.tx.nft.setContractSponsoringRateLimit(contractAddress, rateLimit);395 const events = await submitTransactionAsync(sender, tx);396 const result = getGenericResult(events);397398 expect(result.success).to.be.true;399 });400}401402export async function setContractSponsoringRateLimitExpectFailure(sender: IKeyringPair, contractAddress: AccountId | string, rateLimit: number) {403 await usingApi(async (api) => {404 const tx = api.tx.nft.setContractSponsoringRateLimit(contractAddress, rateLimit);405 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;406 const result = getGenericResult(events);407408 expect(result.success).to.be.false;409 });410}411412export async function toggleContractWhitelistExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, enabled: boolean) {413 await usingApi(async (api) => {414 const tx = api.tx.nft.toggleContractWhiteList(contractAddress, true);415 const events = await submitTransactionAsync(sender, tx);416 const result = getGenericResult(events);417418 expect(result.success).to.be.true;419 });420}421422export async function isWhitelistedInContract(contractAddress: AccountId | string, user: string) {423 let whitelisted: boolean = false;424 await usingApi(async (api) => {425 whitelisted = (await api.query.nft.contractWhiteList(contractAddress, user)).toJSON() as boolean;426 });427 return whitelisted;428}429430export async function addToContractWhiteListExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, user: string) {431 await usingApi(async (api) => {432 const tx = api.tx.nft.addToContractWhiteList(contractAddress, user);433 const events = await submitTransactionAsync(sender, tx);434 const result = getGenericResult(events);435436 expect(result.success).to.be.true;437 });438}439440export async function removeFromContractWhiteListExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, user: string) {441 await usingApi(async (api) => {442 const tx = api.tx.nft.removeFromContractWhiteList(contractAddress, user);443 const events = await submitTransactionAsync(sender, tx);444 const result = getGenericResult(events);445446 expect(result.success).to.be.true;447 });448}449450export async function removeFromContractWhiteListExpectFailure(sender: IKeyringPair, contractAddress: AccountId | string, user: string) {451 await usingApi(async (api) => {452 const tx = api.tx.nft.removeFromContractWhiteList(contractAddress, user);453 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;454 const result = getGenericResult(events);455456 expect(result.success).to.be.false;457 });458}459460export async function setVariableMetaDataExpectSuccess(sender: IKeyringPair, collectionId: number, itemId: number, data: number[]) {461 await usingApi(async (api) => {462 const tx = api.tx.nft.setVariableMetaData(collectionId, itemId, '0x' + Buffer.from(data).toString('hex'));463 const events = await submitTransactionAsync(sender, tx);464 const result = getGenericResult(events);465466 expect(result.success).to.be.true;467 });468}469470export async function setVariableMetaDataExpectFailure(sender: IKeyringPair, collectionId: number, itemId: number, data: number[]) {471 await usingApi(async (api) => {472 const tx = api.tx.nft.setVariableMetaData(collectionId, itemId, '0x' + Buffer.from(data).toString('hex'));473 await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;474 });475}476477export interface CreateFungibleData extends Struct {478 readonly value: u128;479}480481export interface CreateReFungibleData extends Struct {}482export interface CreateNftData extends Struct {}483484export interface CreateItemData extends Enum {485 NFT: CreateNftData;486 Fungible: CreateFungibleData;487 ReFungible: CreateReFungibleData;488}489490export async function burnItemExpectSuccess(owner: IKeyringPair, collectionId: number, tokenId: number, value = 0) {491 await usingApi(async (api) => {492 const tx = api.tx.nft.burnItem(collectionId, tokenId, value);493 const events = await submitTransactionAsync(owner, tx);494 const result = getGenericResult(events);495 496 const item: any = (await api.query.nft.nftItemList(collectionId, tokenId)).toJSON();497 498 499 expect(result.success).to.be.true;500 501 expect(item).to.be.not.null;502 expect(item.Owner).to.be.equal(nullPublicKey);503 });504}505506export async function507approveExpectSuccess(collectionId: number,508 tokenId: number, owner: IKeyringPair, approved: IKeyringPair, amount: number = 1) {509 await usingApi(async (api: ApiPromise) => {510 const allowanceBefore =511 await api.query.nft.allowances(collectionId, [tokenId, owner.address, approved.address]) as unknown as BN;512 const approveNftTx = await api.tx.nft.approve(approved.address, collectionId, tokenId, amount);513 const events = await submitTransactionAsync(owner, approveNftTx);514 const result = getCreateItemResult(events);515 516 expect(result.success).to.be.true;517 const allowanceAfter =518 await api.query.nft.allowances(collectionId, [tokenId, owner.address, approved.address]) as unknown as BN;519 expect(allowanceAfter.toNumber() - allowanceBefore.toNumber()).to.be.equal(amount);520 });521}522523export async function524transferFromExpectSuccess(collectionId: number,525 tokenId: number,526 accountApproved: IKeyringPair,527 accountFrom: IKeyringPair,528 accountTo: IKeyringPair,529 value: number = 1,530 type: string = 'NFT') {531 await usingApi(async (api: ApiPromise) => {532 let balanceBefore = new BN(0);533 if (type === 'Fungible') {534 balanceBefore = await api.query.nft.balance(collectionId, accountTo.address) as unknown as BN;535 }536 const transferFromTx = await api.tx.nft.transferFrom(537 accountFrom.address, accountTo.address, collectionId, tokenId, value);538 const events = await submitTransactionAsync(accountApproved, transferFromTx);539 const result = getCreateItemResult(events);540 541 expect(result.success).to.be.true;542 if (type === 'NFT') {543 const nftItemData = await api.query.nft.nftItemList(collectionId, tokenId) as unknown as ITokenDataType;544 expect(nftItemData.Owner.toString()).to.be.equal(accountTo.address);545 }546 if (type === 'Fungible') {547 const balanceAfter = await api.query.nft.balance(collectionId, accountTo.address) as unknown as BN;548 expect(balanceAfter.sub(balanceBefore).toNumber()).to.be.equal(value);549 }550 if (type === 'ReFungible') {551 const nftItemData =552 await api.query.nft.reFungibleItemList(collectionId, tokenId) as unknown as IReFungibleTokenDataType;553 expect(nftItemData.Owner[0].Owner.toString()).to.be.equal(accountTo.address);554 expect(nftItemData.Owner[0].Fraction.toNumber()).to.be.equal(value);555 }556 });557}558559export async function560transferFromExpectFail(collectionId: number,561 tokenId: number,562 accountApproved: IKeyringPair,563 accountFrom: IKeyringPair,564 accountTo: IKeyringPair,565 value: number = 1) {566 await usingApi(async (api: ApiPromise) => {567 const transferFromTx = await api.tx.nft.transferFrom(568 accountFrom.address, accountTo.address, collectionId, tokenId, value);569 const events = await expect(submitTransactionExpectFailAsync(accountApproved, transferFromTx)).to.be.rejected;570 const result = getCreateCollectionResult(events);571 572 expect(result.success).to.be.false;573 });574}575576export async function577transferExpectSuccess(collectionId: number,578 tokenId: number,579 sender: IKeyringPair,580 recipient: IKeyringPair,581 value: number = 1,582 type: string = 'NFT') {583 await usingApi(async (api: ApiPromise) => {584 let balanceBefore = new BN(0);585 if (type === 'Fungible') {586 balanceBefore = await api.query.nft.balance(collectionId, recipient.address) as unknown as BN;587 }588 const transferTx = await api.tx.nft.transfer(recipient.address, collectionId, tokenId, value);589 const events = await submitTransactionAsync(sender, transferTx);590 const result = getCreateItemResult(events);591 592 expect(result.success).to.be.true;593 if (type === 'NFT') {594 const nftItemData = await api.query.nft.nftItemList(collectionId, tokenId) as unknown as ITokenDataType;595 expect(nftItemData.Owner.toString()).to.be.equal(recipient.address);596 }597 if (type === 'Fungible') {598 const balanceAfter = await api.query.nft.balance(collectionId, recipient.address) as unknown as BN;599 expect(balanceAfter.sub(balanceBefore).toNumber()).to.be.equal(value);600 }601 if (type === 'ReFungible') {602 const nftItemData =603 await api.query.nft.reFungibleItemList(collectionId, tokenId) as unknown as IReFungibleTokenDataType;604 expect(nftItemData.Owner[0].Owner.toString()).to.be.equal(recipient.address);605 expect(nftItemData.Owner[0].Fraction.toNumber()).to.be.equal(value);606 }607 });608}609610export async function611transferExpectFail(collectionId: number,612 tokenId: number,613 sender: IKeyringPair,614 recipient: IKeyringPair,615 value: number = 1,616 type: string = 'NFT') {617 await usingApi(async (api: ApiPromise) => {618 const transferTx = await api.tx.nft.transfer(recipient.address, collectionId, tokenId, value);619 const events = await expect(submitTransactionExpectFailAsync(sender, transferTx)).to.be.rejected;620 if (events && Array.isArray(events)) {621 const result = getCreateCollectionResult(events);622 623 expect(result.success).to.be.false;624 }625 });626}627628export async function629approveExpectFail(collectionId: number,630 tokenId: number, owner: IKeyringPair, approved: IKeyringPair, amount: number = 1) {631 await usingApi(async (api: ApiPromise) => {632 const approveNftTx = await api.tx.nft.approve(approved.address, collectionId, tokenId, amount);633 const events = await expect(submitTransactionExpectFailAsync(owner, approveNftTx)).to.be.rejected;634 const result = getCreateCollectionResult(events);635 636 expect(result.success).to.be.false;637 });638}639640export async function createItemExpectSuccess(641 sender: IKeyringPair, collectionId: number, createMode: string, owner: string = '') {642 let newItemId: number = 0;643 await usingApi(async (api) => {644 const AItemCount = parseInt((await api.query.nft.itemListIndex(collectionId)).toString(), 10);645 const Aitem: any = (await api.query.nft.fungibleItemList(collectionId, owner)).toJSON();646 const AItemBalance = new BigNumber(Aitem.Value);647648 if (owner === '') {649 owner = sender.address;650 }651652 let tx;653 if (createMode === 'Fungible') {654 const createData = {fungible: {value: 10}};655 tx = api.tx.nft.createItem(collectionId, owner, createData);656 } else if (createMode === 'ReFungible') {657 const createData = {refungible: {const_data: [], variable_data: [], pieces: 100}};658 tx = api.tx.nft.createItem(collectionId, owner, createData);659 } else {660 tx = api.tx.nft.createItem(collectionId, owner, createMode);661 }662 const events = await submitTransactionAsync(sender, tx);663 const result = getCreateItemResult(events);664665 const BItemCount = parseInt((await api.query.nft.itemListIndex(collectionId)).toString(), 10);666 const Bitem: any = (await api.query.nft.fungibleItemList(collectionId, owner)).toJSON();667 const BItemBalance = new BigNumber(Bitem.Value);668669 670 671 expect(result.success).to.be.true;672 if (createMode === 'Fungible') {673 expect(BItemBalance.minus(AItemBalance).toNumber()).to.be.equal(10);674 } else {675 expect(BItemCount).to.be.equal(AItemCount + 1);676 }677 expect(collectionId).to.be.equal(result.collectionId);678 expect(BItemCount).to.be.equal(result.itemId);679 newItemId = result.itemId;680 });681 return newItemId;682}683684export async function createItemExpectFailure(685 sender: IKeyringPair, collectionId: number, createMode: string, owner: string = sender.address) {686 await usingApi(async (api) => {687 const tx = api.tx.nft.createItem(collectionId, owner, createMode);688 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;689 const result = getCreateItemResult(events);690691 expect(result.success).to.be.false;692 });693}694695export async function setPublicAccessModeExpectSuccess(696 sender: IKeyringPair, collectionId: number,697 accessMode: 'Normal' | 'WhiteList',698) {699 await usingApi(async (api) => {700701 702 const tx = api.tx.nft.setPublicAccessMode(collectionId, accessMode);703 const events = await submitTransactionAsync(sender, tx);704 const result = getGenericResult(events);705706 707 const collection: any = (await api.query.nft.collection(collectionId)).toJSON();708709 710 711 expect(result.success).to.be.true;712 expect(collection.Access).to.be.equal(accessMode);713 });714}715716export async function enableWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number) {717 await setPublicAccessModeExpectSuccess(sender, collectionId, 'WhiteList');718}719720export async function disableWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number) {721 await setPublicAccessModeExpectSuccess(sender, collectionId, 'Normal');722}723724export async function setMintPermissionExpectSuccess(sender: IKeyringPair, collectionId: number, enabled: boolean) {725 await usingApi(async (api) => {726727 728 const tx = api.tx.nft.setMintPermission(collectionId, enabled);729 const events = await submitTransactionAsync(sender, tx);730 const result = getGenericResult(events);731732 733 const collection: any = (await api.query.nft.collection(collectionId)).toJSON();734735 736 737 expect(result.success).to.be.true;738 expect(collection.MintMode).to.be.equal(enabled);739 });740}741742export async function enablePublicMintingExpectSuccess(sender: IKeyringPair, collectionId: number) {743 await setMintPermissionExpectSuccess(sender, collectionId, true);744}745746export async function setMintPermissionExpectFailure(sender: IKeyringPair, collectionId: number, enabled: boolean) {747 await usingApi(async (api) => {748 749 const tx = api.tx.nft.setMintPermission(collectionId, enabled);750 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;751 const result = getCreateCollectionResult(events);752 753 expect(result.success).to.be.false;754 });755}756757export async function isWhitelisted(collectionId: number, address: string) {758 let whitelisted: boolean = false;759 await usingApi(async (api) => {760 whitelisted = (await api.query.nft.whiteList(collectionId, address)).toJSON() as unknown as boolean;761 });762 return whitelisted;763}764765export async function addToWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number, address: string) {766 await usingApi(async (api) => {767768 const whiteListedBefore = (await api.query.nft.whiteList(collectionId, address)).toJSON();769770 771 const tx = api.tx.nft.addToWhiteList(collectionId, address);772 const events = await submitTransactionAsync(sender, tx);773 const result = getGenericResult(events);774775 const whiteListedAfter = (await api.query.nft.whiteList(collectionId, address)).toJSON();776777 778 779 expect(result.success).to.be.true;780 781 expect(whiteListedBefore).to.be.false;782 783 expect(whiteListedAfter).to.be.true;784 });785}786787export async function removeFromWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number, address: string) {788 await usingApi(async (api) => {789 790 const tx = api.tx.nft.removeFromWhiteList(collectionId, address);791 const events = await submitTransactionAsync(sender, tx);792 const result = getGenericResult(events);793794 795 796 expect(result.success).to.be.true;797 });798}799800export async function removeFromWhiteListExpectFailure(sender: IKeyringPair, collectionId: number, address: string) {801 await usingApi(async (api) => {802 803 const tx = api.tx.nft.removeFromWhiteList(collectionId, address);804 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;805 const result = getGenericResult(events);806807 808 809 expect(result.success).to.be.false;810 });811}812813export const getDetailedCollectionInfo = async (api: ApiPromise, collectionId: number)814 : Promise<ICollectionInterface | null> => {815 return await api.query.nft.collection(collectionId) as unknown as ICollectionInterface;816};817818export const getCreatedCollectionCount = async (api: ApiPromise): Promise<number> => {819 820 return (await api.query.nft.createdCollectionCount() as unknown as BN).toNumber();821};