git.delta.rocks / unique-network / refs/commits / 800a78ff8f6e

difftreelog

source

tests/src/util/helpers.ts31.0 KiBsourcehistory
1//2// This file is subject to the terms and conditions defined in3// file 'LICENSE', which is part of this source code package.4//56import { 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    // console.log(`    ${phase}: ${section}.${method}:: ${data}`);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    // console.log(`    ${phase}: ${section}.${method}:: ${data}`);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    // console.log(`    ${phase}: ${section}.${method}:: ${data}`);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    // Get number of collections before the transaction151    const AcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString(), 10);152153    // Run the CreateCollection transaction154    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    // Get number of collections after the transaction172    const BcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString(), 10);173174    // Get the collection175    const collection: any = (await api.query.nft.collection(result.collectionId)).toJSON();176177    // What to expect178    // tslint:disable-next-line:no-unused-expression179    expect(result.success).to.be.true;180    expect(result.collectionId).to.be.equal(BcollectionCount);181    // tslint:disable-next-line:no-unused-expression182    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    // Get number of collections before the transaction211    const AcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString());212213    // Run the CreateCollection transaction214    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    // Get number of collections after the transaction220    const BcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString());221222    // What to expect223    // tslint:disable-next-line:no-unused-expression224    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, seedAddition = ''): Promise<IKeyringPair> {230  let bal = new BigNumber(0);231  let unused;232  do {233    const randomSeed = 'seed' +  Math.floor(Math.random() * Math.floor(10000)) + seedAddition;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 function findUnusedAddresses(api: ApiPromise, amount: number): Promise<IKeyringPair[]> {242  return Promise.all(new Array(amount).fill(null).map(() => findUnusedAddress(api, '_' + Date.now())));243}244245export async function findNotExistingCollection(api: ApiPromise): Promise<number> {246  const totalNumber = parseInt((await api.query.nft.createdCollectionCount()).toString(), 10) as unknown as number;247  const newCollection: number = totalNumber + 1;248  return newCollection;249}250251function getDestroyResult(events: EventRecord[]): boolean {252  let success: boolean = false;253  events.forEach(({ phase, event: { data, method, section } }) => {254    // console.log(`    ${phase}: ${section}.${method}:: ${data}`);255    if (method == 'ExtrinsicSuccess') {256      success = true;257    }258  });259  return success;260}261262export async function destroyCollectionExpectFailure(collectionId: number, senderSeed: string = '//Alice') {263  await usingApi(async (api) => {264    // Run the DestroyCollection transaction265    const alicePrivateKey = privateKey(senderSeed);266    const tx = api.tx.nft.destroyCollection(collectionId);267    await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;268  });269}270271export async function destroyCollectionExpectSuccess(collectionId: number, senderSeed: string = '//Alice') {272  await usingApi(async (api) => {273    // Run the DestroyCollection transaction274    const alicePrivateKey = privateKey(senderSeed);275    const tx = api.tx.nft.destroyCollection(collectionId);276    const events = await submitTransactionAsync(alicePrivateKey, tx);277    const result = getDestroyResult(events);278279    // Get the collection280    const collection: any = (await api.query.nft.collection(collectionId)).toJSON();281282    // What to expect283    expect(result).to.be.true;284    expect(collection).to.be.not.null;285    expect(collection.Owner).to.be.equal(nullPublicKey);286  });287}288289export async function setCollectionSponsorExpectSuccess(collectionId: number, sponsor: string) {290  await usingApi(async (api) => {291292    // Run the transaction293    const alicePrivateKey = privateKey('//Alice');294    const tx = api.tx.nft.setCollectionSponsor(collectionId, sponsor);295    const events = await submitTransactionAsync(alicePrivateKey, tx);296    const result = getGenericResult(events);297298    // Get the collection299    const collection: any = (await api.query.nft.collection(collectionId)).toJSON();300301    // What to expect302    expect(result.success).to.be.true;303    expect(collection.Sponsor.toString()).to.be.equal(sponsor.toString());304    expect(collection.SponsorConfirmed).to.be.false;305  });306}307308export async function removeCollectionSponsorExpectSuccess(collectionId: number) {309  await usingApi(async (api) => {310311    // Run the transaction312    const alicePrivateKey = privateKey('//Alice');313    const tx = api.tx.nft.removeCollectionSponsor(collectionId);314    const events = await submitTransactionAsync(alicePrivateKey, tx);315    const result = getGenericResult(events);316317    // Get the collection318    const collection: any = (await api.query.nft.collection(collectionId)).toJSON();319320    // What to expect321    expect(result.success).to.be.true;322    expect(collection.Sponsor).to.be.equal(nullPublicKey);323    expect(collection.SponsorConfirmed).to.be.false;324  });325}326327export async function removeCollectionSponsorExpectFailure(collectionId: number) {328  await usingApi(async (api) => {329330    // Run the transaction331    const alicePrivateKey = privateKey('//Alice');332    const tx = api.tx.nft.removeCollectionSponsor(collectionId);333    await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;334  });335}336337export async function setCollectionSponsorExpectFailure(collectionId: number, sponsor: string, senderSeed: string = '//Alice') {338  await usingApi(async (api) => {339340    // Run the transaction341    const alicePrivateKey = privateKey(senderSeed);342    const tx = api.tx.nft.setCollectionSponsor(collectionId, sponsor);343    await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;344  });345}346347export async function confirmSponsorshipExpectSuccess(collectionId: number, senderSeed: string = '//Alice') {348  await usingApi(async (api) => {349350    // Run the transaction351    const sender = privateKey(senderSeed);352    const tx = api.tx.nft.confirmSponsorship(collectionId);353    const events = await submitTransactionAsync(sender, tx);354    const result = getGenericResult(events);355356    // Get the collection357    const collection: any = (await api.query.nft.collection(collectionId)).toJSON();358359    // What to expect360    expect(result.success).to.be.true;361    expect(collection.Sponsor).to.be.equal(sender.address);362    expect(collection.SponsorConfirmed).to.be.true;363  });364}365366367export async function confirmSponsorshipExpectFailure(collectionId: number, senderSeed: string = '//Alice') {368  await usingApi(async (api) => {369370    // Run the transaction371    const sender = privateKey(senderSeed);372    const tx = api.tx.nft.confirmSponsorship(collectionId);373    await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;374  });375}376377export async function enableContractSponsoringExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, enable: boolean) {378  await usingApi(async (api) => {379    const tx = api.tx.nft.enableContractSponsoring(contractAddress, enable);380    const events = await submitTransactionAsync(sender, tx);381    const result = getGenericResult(events);382383    expect(result.success).to.be.true;384  });385}386387export async function enableContractSponsoringExpectFailure(sender: IKeyringPair, contractAddress: AccountId | string, enable: boolean) {388  await usingApi(async (api) => {389    const tx = api.tx.nft.enableContractSponsoring(contractAddress, enable);390    const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;391    const result = getGenericResult(events);392393    expect(result.success).to.be.false;394  });395}396397export async function setContractSponsoringRateLimitExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, rateLimit: number) {398  await usingApi(async (api) => {399    const tx = api.tx.nft.setContractSponsoringRateLimit(contractAddress, rateLimit);400    const events = await submitTransactionAsync(sender, tx);401    const result = getGenericResult(events);402403    expect(result.success).to.be.true;404  });405}406407export async function setContractSponsoringRateLimitExpectFailure(sender: IKeyringPair, contractAddress: AccountId | string, rateLimit: number) {408  await usingApi(async (api) => {409    const tx = api.tx.nft.setContractSponsoringRateLimit(contractAddress, rateLimit);410    const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;411    const result = getGenericResult(events);412413    expect(result.success).to.be.false;414  });415}416417export async function toggleContractWhitelistExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, enabled: boolean) {418  await usingApi(async (api) => {419    const tx = api.tx.nft.toggleContractWhiteList(contractAddress, true);420    const events = await submitTransactionAsync(sender, tx);421    const result = getGenericResult(events);422423    expect(result.success).to.be.true;424  });425}426427export async function isWhitelistedInContract(contractAddress: AccountId | string, user: string) {428  let whitelisted: boolean = false;429  await usingApi(async (api) => {430    whitelisted = (await api.query.nft.contractWhiteList(contractAddress, user)).toJSON() as boolean;431  });432  return whitelisted;433}434435export async function addToContractWhiteListExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, user: string) {436  await usingApi(async (api) => {437    const tx = api.tx.nft.addToContractWhiteList(contractAddress, user);438    const events = await submitTransactionAsync(sender, tx);439    const result = getGenericResult(events);440441    expect(result.success).to.be.true;442  });443}444445export async function removeFromContractWhiteListExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, user: string) {446  await usingApi(async (api) => {447    const tx = api.tx.nft.removeFromContractWhiteList(contractAddress, user);448    const events = await submitTransactionAsync(sender, tx);449    const result = getGenericResult(events);450451    expect(result.success).to.be.true;452  });453}454455export async function removeFromContractWhiteListExpectFailure(sender: IKeyringPair, contractAddress: AccountId | string, user: string) {456  await usingApi(async (api) => {457    const tx = api.tx.nft.removeFromContractWhiteList(contractAddress, user);458    const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;459    const result = getGenericResult(events);460461    expect(result.success).to.be.false;462  });463}464465export async function setVariableMetaDataExpectSuccess(sender: IKeyringPair, collectionId: number, itemId: number, data: number[]) {466  await usingApi(async (api) => {467    const tx = api.tx.nft.setVariableMetaData(collectionId, itemId, '0x' + Buffer.from(data).toString('hex'));468    const events = await submitTransactionAsync(sender, tx);469    const result = getGenericResult(events);470471    expect(result.success).to.be.true;472  });473}474475export async function setVariableMetaDataExpectFailure(sender: IKeyringPair, collectionId: number, itemId: number, data: number[]) {476  await usingApi(async (api) => {477    const tx = api.tx.nft.setVariableMetaData(collectionId, itemId, '0x' + Buffer.from(data).toString('hex'));478    await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;479  });480}481482export interface CreateFungibleData extends Struct {483  readonly value: u128;484}485486export interface CreateReFungibleData extends Struct {}487export interface CreateNftData extends Struct {}488489export interface CreateItemData extends Enum {490  NFT: CreateNftData;491  Fungible: CreateFungibleData;492  ReFungible: CreateReFungibleData;493}494495export async function burnItemExpectSuccess(owner: IKeyringPair, collectionId: number, tokenId: number, value = 0) {496  await usingApi(async (api) => {497    const tx = api.tx.nft.burnItem(collectionId, tokenId, value);498    const events = await submitTransactionAsync(owner, tx);499    const result = getGenericResult(events);500    // Get the item501    const item: any = (await api.query.nft.nftItemList(collectionId, tokenId)).toJSON();502    // What to expect503    // tslint:disable-next-line:no-unused-expression504    expect(result.success).to.be.true;505    // tslint:disable-next-line:no-unused-expression506    expect(item).to.be.not.null;507    expect(item.Owner).to.be.equal(nullPublicKey);508  });509}510511export async function512approveExpectSuccess(collectionId: number,513                     tokenId: number, owner: IKeyringPair, approved: IKeyringPair, amount: number = 1) { //alice,bob514  await usingApi(async (api: ApiPromise) => {515    const allowanceBefore =516      await api.query.nft.allowances(collectionId, [tokenId, owner.address, approved.address]) as unknown as BN;517    const approveNftTx = await api.tx.nft.approve(approved.address, collectionId, tokenId, amount);518    const events = await submitTransactionAsync(owner, approveNftTx);519    const result = getCreateItemResult(events);520    // tslint:disable-next-line:no-unused-expression521    expect(result.success).to.be.true;522    const allowanceAfter =523      await api.query.nft.allowances(collectionId, [tokenId, owner.address, approved.address]) as unknown as BN;524    expect(allowanceAfter.toNumber() - allowanceBefore.toNumber()).to.be.equal(amount);525  });526}527528export async function529transferFromExpectSuccess(collectionId: number,530                          tokenId: number,531                          accountApproved: IKeyringPair, //bob532                          accountFrom: IKeyringPair, //alice533                          accountTo: IKeyringPair, //charlie534                          value: number = 1,535                          type: string = 'NFT') {536  await usingApi(async (api: ApiPromise) => {537    let balanceBefore = new BN(0);538    if (type === 'Fungible') {539      balanceBefore = await api.query.nft.balance(collectionId, accountTo.address) as unknown as BN;540    }541    const transferFromTx = await api.tx.nft.transferFrom(542      accountFrom.address, accountTo.address, collectionId, tokenId, value);543    const events = await submitTransactionAsync(accountApproved, transferFromTx);544    const result = getCreateItemResult(events);545    // tslint:disable-next-line:no-unused-expression546    expect(result.success).to.be.true;547    if (type === 'NFT') {548      const nftItemData = await api.query.nft.nftItemList(collectionId, tokenId) as unknown as ITokenDataType;549      expect(nftItemData.Owner.toString()).to.be.equal(accountTo.address);550    }551    if (type === 'Fungible') {552      const balanceAfter = await api.query.nft.balance(collectionId, accountTo.address) as unknown as BN;553      expect(balanceAfter.sub(balanceBefore).toNumber()).to.be.equal(value);554    }555    if (type === 'ReFungible') {556      const nftItemData =557        await api.query.nft.reFungibleItemList(collectionId, tokenId) as unknown as IReFungibleTokenDataType;558      expect(nftItemData.Owner[0].Owner.toString()).to.be.equal(accountTo.address);559      expect(nftItemData.Owner[0].Fraction.toNumber()).to.be.equal(value);560    }561  });562}563564export async function565transferFromExpectFail(collectionId: number,566                       tokenId: number,567                       accountApproved: IKeyringPair,568                       accountFrom: IKeyringPair,569                       accountTo: IKeyringPair,570                       value: number = 1) {571  await usingApi(async (api: ApiPromise) => {572    const transferFromTx = await api.tx.nft.transferFrom(573      accountFrom.address, accountTo.address, collectionId, tokenId, value);574    const events = await expect(submitTransactionExpectFailAsync(accountApproved, transferFromTx)).to.be.rejected;575    const result = getCreateCollectionResult(events);576    // tslint:disable-next-line:no-unused-expression577    expect(result.success).to.be.false;578  });579}580581export async function582transferExpectSuccess(collectionId: number,583                      tokenId: number,584                      sender: IKeyringPair,585                      recipient: IKeyringPair,586                      value: number = 1,587                      type: string = 'NFT') {588  await usingApi(async (api: ApiPromise) => {589    let balanceBefore = new BN(0);590    if (type === 'Fungible') {591      balanceBefore = await api.query.nft.balance(collectionId, recipient.address) as unknown as BN;592    }593    const transferTx = await api.tx.nft.transfer(recipient.address, collectionId, tokenId, value);594    const events = await submitTransactionAsync(sender, transferTx);595    const result = getCreateItemResult(events);596    // tslint:disable-next-line:no-unused-expression597    expect(result.success).to.be.true;598    if (type === 'NFT') {599      const nftItemData = await api.query.nft.nftItemList(collectionId, tokenId) as unknown as ITokenDataType;600      expect(nftItemData.Owner.toString()).to.be.equal(recipient.address);601    }602    if (type === 'Fungible') {603      const balanceAfter = await api.query.nft.balance(collectionId, recipient.address) as unknown as BN;604      expect(balanceAfter.sub(balanceBefore).toNumber()).to.be.equal(value);605    }606    if (type === 'ReFungible') {607      const nftItemData =608        await api.query.nft.reFungibleItemList(collectionId, tokenId) as unknown as IReFungibleTokenDataType;609      expect(nftItemData.Owner[0].Owner.toString()).to.be.equal(recipient.address);610      expect(nftItemData.Owner[0].Fraction.toNumber()).to.be.equal(value);611    }612  });613}614615export async function616transferExpectFail(collectionId: number,617                   tokenId: number,618                   sender: IKeyringPair,619                   recipient: IKeyringPair,620                   value: number = 1,621                   type: string = 'NFT') {622  await usingApi(async (api: ApiPromise) => {623    const transferTx = await api.tx.nft.transfer(recipient.address, collectionId, tokenId, value);624    const events = await expect(submitTransactionExpectFailAsync(sender, transferTx)).to.be.rejected;625    if (events && Array.isArray(events)) {626      const result = getCreateCollectionResult(events);627      // tslint:disable-next-line:no-unused-expression628      expect(result.success).to.be.false;629    }630  });631}632633export async function634approveExpectFail(collectionId: number,635                  tokenId: number, owner: IKeyringPair, approved: IKeyringPair, amount: number = 1) {636  await usingApi(async (api: ApiPromise) => {637    const approveNftTx = await api.tx.nft.approve(approved.address, collectionId, tokenId, amount);638    const events = await expect(submitTransactionExpectFailAsync(owner, approveNftTx)).to.be.rejected;639    const result = getCreateCollectionResult(events);640    // tslint:disable-next-line:no-unused-expression641    expect(result.success).to.be.false;642  });643}644645export async function createItemExpectSuccess(646  sender: IKeyringPair, collectionId: number, createMode: string, owner: string = '') {647  let newItemId: number = 0;648  await usingApi(async (api) => {649    const AItemCount = parseInt((await api.query.nft.itemListIndex(collectionId)).toString(), 10);650    const Aitem: any = (await api.query.nft.fungibleItemList(collectionId, owner)).toJSON();651    const AItemBalance = new BigNumber(Aitem.Value);652653    if (owner === '') {654      owner = sender.address;655    }656657    let tx;658    if (createMode === 'Fungible') {659      const createData = {fungible: {value: 10}};660      tx = api.tx.nft.createItem(collectionId, owner, createData);661    } else if (createMode === 'ReFungible') {662      const createData = {refungible: {const_data: [], variable_data: [], pieces: 100}};663      tx = api.tx.nft.createItem(collectionId, owner, createData);664    } else {665      tx = api.tx.nft.createItem(collectionId, owner, createMode);666    }667    const events = await submitTransactionAsync(sender, tx);668    const result = getCreateItemResult(events);669670    const BItemCount = parseInt((await api.query.nft.itemListIndex(collectionId)).toString(), 10);671    const Bitem: any = (await api.query.nft.fungibleItemList(collectionId, owner)).toJSON();672    const BItemBalance = new BigNumber(Bitem.Value);673674    // What to expect675    // tslint:disable-next-line:no-unused-expression676    expect(result.success).to.be.true;677    if (createMode === 'Fungible') {678      expect(BItemBalance.minus(AItemBalance).toNumber()).to.be.equal(10);679    } else {680      expect(BItemCount).to.be.equal(AItemCount + 1);681    }682    expect(collectionId).to.be.equal(result.collectionId);683    expect(BItemCount).to.be.equal(result.itemId);684    newItemId = result.itemId;685  });686  return newItemId;687}688689export async function createItemExpectFailure(690  sender: IKeyringPair, collectionId: number, createMode: string, owner: string = sender.address) {691  await usingApi(async (api) => {692    const tx = api.tx.nft.createItem(collectionId, owner, createMode);693    const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;694    const result = getCreateItemResult(events);695696    expect(result.success).to.be.false;697  });698}699700export async function setPublicAccessModeExpectSuccess(701  sender: IKeyringPair, collectionId: number,702  accessMode: 'Normal' | 'WhiteList',703) {704  await usingApi(async (api) => {705706    // Run the transaction707    const tx = api.tx.nft.setPublicAccessMode(collectionId, accessMode);708    const events = await submitTransactionAsync(sender, tx);709    const result = getGenericResult(events);710711    // Get the collection712    const collection: any = (await api.query.nft.collection(collectionId)).toJSON();713714    // What to expect715    // tslint:disable-next-line:no-unused-expression716    expect(result.success).to.be.true;717    expect(collection.Access).to.be.equal(accessMode);718  });719}720721export async function enableWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number) {722  await setPublicAccessModeExpectSuccess(sender, collectionId, 'WhiteList');723}724725export async function disableWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number) {726  await setPublicAccessModeExpectSuccess(sender, collectionId, 'Normal');727}728729export async function setMintPermissionExpectSuccess(sender: IKeyringPair, collectionId: number, enabled: boolean) {730  await usingApi(async (api) => {731732    // Run the transaction733    const tx = api.tx.nft.setMintPermission(collectionId, enabled);734    const events = await submitTransactionAsync(sender, tx);735    const result = getGenericResult(events);736737    // Get the collection738    const collection: any = (await api.query.nft.collection(collectionId)).toJSON();739740    // What to expect741    // tslint:disable-next-line:no-unused-expression742    expect(result.success).to.be.true;743    expect(collection.MintMode).to.be.equal(enabled);744  });745}746747export async function enablePublicMintingExpectSuccess(sender: IKeyringPair, collectionId: number) {748  await setMintPermissionExpectSuccess(sender, collectionId, true);749}750751export async function setMintPermissionExpectFailure(sender: IKeyringPair, collectionId: number, enabled: boolean) {752  await usingApi(async (api) => {753    // Run the transaction754    const tx = api.tx.nft.setMintPermission(collectionId, enabled);755    const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;756    const result = getCreateCollectionResult(events);757    // tslint:disable-next-line:no-unused-expression758    expect(result.success).to.be.false;759  });760}761762export async function isWhitelisted(collectionId: number, address: string) {763  let whitelisted: boolean = false;764  await usingApi(async (api) => {765    whitelisted = (await api.query.nft.whiteList(collectionId, address)).toJSON() as unknown as boolean;766  });767  return whitelisted;768}769770export async function addToWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number, address: string) {771  await usingApi(async (api) => {772773    const whiteListedBefore = (await api.query.nft.whiteList(collectionId, address)).toJSON();774775    // Run the transaction776    const tx = api.tx.nft.addToWhiteList(collectionId, address);777    const events = await submitTransactionAsync(sender, tx);778    const result = getGenericResult(events);779780    const whiteListedAfter = (await api.query.nft.whiteList(collectionId, address)).toJSON();781782    // What to expect783    // tslint:disable-next-line:no-unused-expression784    expect(result.success).to.be.true;785    // tslint:disable-next-line: no-unused-expression786    expect(whiteListedBefore).to.be.false;787    // tslint:disable-next-line: no-unused-expression788    expect(whiteListedAfter).to.be.true;789  });790}791792export async function removeFromWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number, address: string) {793  await usingApi(async (api) => {794    // Run the transaction795    const tx = api.tx.nft.removeFromWhiteList(collectionId, address);796    const events = await submitTransactionAsync(sender, tx);797    const result = getGenericResult(events);798799    // What to expect800    // tslint:disable-next-line:no-unused-expression801    expect(result.success).to.be.true;802  });803}804805export async function removeFromWhiteListExpectFailure(sender: IKeyringPair, collectionId: number, address: string) {806  await usingApi(async (api) => {807    // Run the transaction808    const tx = api.tx.nft.removeFromWhiteList(collectionId, address);809    const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;810    const result = getGenericResult(events);811812    // What to expect813    // tslint:disable-next-line:no-unused-expression814    expect(result.success).to.be.false;815  });816}817818export const getDetailedCollectionInfo = async (api: ApiPromise, collectionId: number)819  : Promise<ICollectionInterface | null> => {820  return await api.query.nft.collection(collectionId) as unknown as ICollectionInterface;821};822823export const getCreatedCollectionCount = async (api: ApiPromise): Promise<number> => {824  // set global object - collectionsCount825  return (await api.query.nft.createdCollectionCount() as unknown as BN).toNumber();826};