123456import { ApiPromise, Keyring } from '@polkadot/api';7import type { AccountId, EventRecord } from '@polkadot/types/interfaces';8import { IKeyringPair } from '@polkadot/types/types';9import { evmToAddress } from '@polkadot/util-crypto';10import { BigNumber } from 'bignumber.js';11import BN from 'bn.js';12import chai from 'chai';13import chaiAsPromised from 'chai-as-promised';14import { alicesPublicKey } from '../accounts';15import privateKey from '../substrate/privateKey';16import { default as usingApi, submitTransactionAsync, submitTransactionExpectFailAsync } from '../substrate/substrate-api';17import { ICollectionInterface } from '../types';18import { hexToStr, strToUTF16, utf16ToStr } from './util';1920chai.use(chaiAsPromised);21const expect = chai.expect;2223export type CrossAccountId = {24 substrate: string,25} | {26 ethereum: string,27};28export function normalizeAccountId(input: string | AccountId | CrossAccountId | IKeyringPair): CrossAccountId {29 if (typeof input === 'string')30 return { substrate: input };31 if ('address' in input) {32 return { substrate: input.address };33 }34 if ('ethereum' in input) {35 input.ethereum = input.ethereum.toLowerCase();36 return input;37 }38 if ('substrate' in input) {39 return input;40 }4142 43 return {substrate: input.toString()};44}45export function toSubstrateAddress(input: string | CrossAccountId | IKeyringPair): string {46 input = normalizeAccountId(input);47 if ('substrate' in input) {48 return input.substrate;49 } else {50 return evmToAddress(input.ethereum);51 }52}5354export const U128_MAX = (1n << 128n) - 1n;5556const MICROUNIQUE = 1_000_000_000n;57const MILLIUNIQUE = 1_000n * MICROUNIQUE;58const CENTIUNIQUE = 10n * MILLIUNIQUE;59export const UNIQUE = 100n * CENTIUNIQUE;6061type GenericResult = {62 success: boolean,63};6465interface CreateCollectionResult {66 success: boolean;67 collectionId: number;68}6970interface CreateItemResult {71 success: boolean;72 collectionId: number;73 itemId: number;74 recipient?: CrossAccountId;75}7677interface TransferResult {78 success: boolean;79 collectionId: number;80 itemId: number;81 sender?: CrossAccountId;82 recipient?: CrossAccountId;83 value: bigint;84}8586interface IReFungibleOwner {87 Fraction: BN;88 Owner: number[];89}9091interface ITokenDataType {92 Owner: IKeyringPair;93 ConstData: number[];94 VariableData: number[];95}9697interface IGetMessage {98 checkMsgNftMethod: string;99 checkMsgTrsMethod: string;100 checkMsgSysMethod: string;101}102103export interface IFungibleTokenDataType {104 Value: number;105}106107export interface IChainLimits {108 CollectionNumbersLimit: number;109 AccountTokenOwnershipLimit: number;110 CollectionsAdminsLimit: number;111 CustomDataLimit: number;112 NftSponsorTransferTimeout: number;113 FungibleSponsorTransferTimeout: number;114 RefungibleSponsorTransferTimeout: number;115 OffchainSchemaLimit: number;116 VariableOnChainSchemaLimit: number;117 ConstOnChainSchemaLimit: number;118}119120export interface IReFungibleTokenDataType {121 Owner: IReFungibleOwner[];122 ConstData: number[];123 VariableData: number[];124}125126export function nftEventMessage(events: EventRecord[]): IGetMessage {127 let checkMsgNftMethod = '';128 let checkMsgTrsMethod = '';129 let checkMsgSysMethod = '';130 events.forEach(({ event: { method, section } }) => {131 if (section === 'nft') {132 checkMsgNftMethod = method;133 } else if (section === 'treasury') {134 checkMsgTrsMethod = method;135 } else if (section === 'system') {136 checkMsgSysMethod = method;137 } else { return null; }138 });139 const result: IGetMessage = {140 checkMsgNftMethod,141 checkMsgTrsMethod,142 checkMsgSysMethod,143 };144 return result;145}146147export function getGenericResult(events: EventRecord[]): GenericResult {148 const result: GenericResult = {149 success: false,150 };151 events.forEach(({ event: { method } }) => {152 153 if (method === 'ExtrinsicSuccess') {154 result.success = true;155 }156 });157 return result;158}159160161162export function getCreateCollectionResult(events: EventRecord[]): CreateCollectionResult {163 let success = false;164 let collectionId = 0;165 events.forEach(({ event: { data, method, section } }) => {166 167 if (method == 'ExtrinsicSuccess') {168 success = true;169 } else if ((section == 'nft') && (method == 'CollectionCreated')) {170 collectionId = parseInt(data[0].toString());171 }172 });173 const result: CreateCollectionResult = {174 success,175 collectionId,176 };177 return result;178}179180export function getCreateItemResult(events: EventRecord[]): CreateItemResult {181 let success = false;182 let collectionId = 0;183 let itemId = 0;184 let recipient;185 events.forEach(({ event: { data, method, section } }) => {186 187 if (method == 'ExtrinsicSuccess') {188 success = true;189 } else if ((section == 'nft') && (method == 'ItemCreated')) {190 collectionId = parseInt(data[0].toString());191 itemId = parseInt(data[1].toString());192 recipient = data[2].toJSON();193 }194 });195 const result: CreateItemResult = {196 success,197 collectionId,198 itemId,199 recipient,200 };201 return result;202}203204export function getTransferResult(events: EventRecord[]): TransferResult {205 const result: TransferResult = {206 success: false,207 collectionId: 0,208 itemId: 0,209 value: 0n,210 };211212 events.forEach(({ event: { data, method, section } }) => {213 if (method === 'ExtrinsicSuccess') {214 result.success = true;215 } else if (section === 'nft' && method === 'Transfer') {216 result.collectionId = +data[0].toString();217 result.itemId = +data[1].toString();218 result.sender = data[2].toJSON() as CrossAccountId;219 result.recipient = data[3].toJSON() as CrossAccountId;220 result.value = BigInt(data[4].toString());221 }222 });223224 return result;225}226227interface Invalid {228 type: 'Invalid';229}230231interface Nft {232 type: 'NFT';233}234235interface Fungible {236 type: 'Fungible';237 decimalPoints: number;238}239240interface ReFungible {241 type: 'ReFungible';242}243244type CollectionMode = Nft | Fungible | ReFungible | Invalid;245246export type CreateCollectionParams = {247 mode: CollectionMode,248 name: string,249 description: string,250 tokenPrefix: string,251};252253const defaultCreateCollectionParams: CreateCollectionParams = {254 description: 'description',255 mode: { type: 'NFT' },256 name: 'name',257 tokenPrefix: 'prefix',258};259260export async function createCollectionExpectSuccess(params: Partial<CreateCollectionParams> = {}): Promise<number> {261 const { name, description, mode, tokenPrefix } = { ...defaultCreateCollectionParams, ...params };262263 let collectionId = 0;264 await usingApi(async (api) => {265 266 const AcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString(), 10);267268 269 const alicePrivateKey = privateKey('//Alice');270271 let modeprm = {};272 if (mode.type === 'NFT') {273 modeprm = { nft: null };274 } else if (mode.type === 'Fungible') {275 modeprm = { fungible: mode.decimalPoints };276 } else if (mode.type === 'ReFungible') {277 modeprm = { refungible: null };278 } else if (mode.type === 'Invalid') {279 modeprm = { invalid: null };280 }281282 const tx = api.tx.nft.createCollection(strToUTF16(name), strToUTF16(description), strToUTF16(tokenPrefix), modeprm);283 const events = await submitTransactionAsync(alicePrivateKey, tx);284 const result = getCreateCollectionResult(events);285286 287 const BcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString(), 10);288289 290 const collection: any = (await api.query.nft.collectionById(result.collectionId) as any).toJSON();291292 293 294 expect(result.success).to.be.true;295 expect(result.collectionId).to.be.equal(BcollectionCount);296 297 expect(collection).to.be.not.null;298 expect(BcollectionCount).to.be.equal(AcollectionCount + 1, 'Error: NFT collection NOT created.');299 expect(collection.Owner).to.be.equal(toSubstrateAddress(alicesPublicKey));300 expect(utf16ToStr(collection.Name)).to.be.equal(name);301 expect(utf16ToStr(collection.Description)).to.be.equal(description);302 expect(hexToStr(collection.TokenPrefix)).to.be.equal(tokenPrefix);303304 collectionId = result.collectionId;305 });306307 return collectionId;308}309310export async function createCollectionExpectFailure(params: Partial<CreateCollectionParams> = {}) {311 const { name, description, mode, tokenPrefix } = { ...defaultCreateCollectionParams, ...params };312313 let modeprm = {};314 if (mode.type === 'NFT') {315 modeprm = { nft: null };316 } else if (mode.type === 'Fungible') {317 modeprm = { fungible: mode.decimalPoints };318 } else if (mode.type === 'ReFungible') {319 modeprm = { refungible: null };320 } else if (mode.type === 'Invalid') {321 modeprm = { invalid: null };322 }323324 await usingApi(async (api) => {325 326 const AcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString());327328 329 const alicePrivateKey = privateKey('//Alice');330 const tx = api.tx.nft.createCollection(strToUTF16(name), strToUTF16(description), strToUTF16(tokenPrefix), modeprm);331 const events = await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;332 const result = getCreateCollectionResult(events);333334 335 const BcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString());336337 338 339 expect(result.success).to.be.false;340 expect(BcollectionCount).to.be.equal(AcollectionCount, 'Error: Collection with incorrect data created.');341 });342}343344export async function findUnusedAddress(api: ApiPromise, seedAddition = ''): Promise<IKeyringPair> {345 let bal = new BigNumber(0);346 let unused;347 do {348 const randomSeed = 'seed' + Math.floor(Math.random() * Math.floor(10000)) + seedAddition;349 const keyring = new Keyring({ type: 'sr25519' });350 unused = keyring.addFromUri(`//${randomSeed}`);351 bal = new BigNumber((await api.query.system.account(unused.address)).data.free.toString());352 } while (bal.toFixed() != '0');353 return unused;354}355356export async function getAllowance(collectionId: number, tokenId: number, owner: string, approved: string) {357 return await usingApi(async (api) => {358 const bn = await api.query.nft.allowances(collectionId, [tokenId, owner, approved]) as unknown as BN;359 return BigInt(bn.toString());360 });361}362363export function findUnusedAddresses(api: ApiPromise, amount: number): Promise<IKeyringPair[]> {364 return Promise.all(new Array(amount).fill(null).map(() => findUnusedAddress(api, '_' + Date.now())));365}366367export async function findNotExistingCollection(api: ApiPromise): Promise<number> {368 const totalNumber = parseInt((await api.query.nft.createdCollectionCount()).toString(), 10) as unknown as number;369 const newCollection: number = totalNumber + 1;370 return newCollection;371}372373function getDestroyResult(events: EventRecord[]): boolean {374 let success = false;375 events.forEach(({ event: { method } }) => {376 if (method == 'ExtrinsicSuccess') {377 success = true;378 }379 });380 return success;381}382383export async function destroyCollectionExpectFailure(collectionId: number, senderSeed = '//Alice') {384 await usingApi(async (api) => {385 386 const alicePrivateKey = privateKey(senderSeed);387 const tx = api.tx.nft.destroyCollection(collectionId);388 await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;389 });390}391392export async function destroyCollectionExpectSuccess(collectionId: number, senderSeed = '//Alice') {393 await usingApi(async (api) => {394 395 const alicePrivateKey = privateKey(senderSeed);396 const tx = api.tx.nft.destroyCollection(collectionId);397 const events = await submitTransactionAsync(alicePrivateKey, tx);398 const result = getDestroyResult(events);399400 401 const collection: any = (await api.query.nft.collectionById(collectionId)).toJSON();402403 404 expect(result).to.be.true;405 expect(collection).to.be.null;406 });407}408409export async function queryCollectionLimits(collectionId: number) {410 return await usingApi(async (api) => {411 return ((await api.query.nft.collectionById(collectionId)).toJSON() as any).Limits;412 });413}414415export async function setCollectionLimitsExpectSuccess(sender: IKeyringPair, collectionId: number, limits: any) {416 await usingApi(async (api) => {417 const oldLimits = await queryCollectionLimits(collectionId);418 const newLimits = { ...oldLimits as any, ...limits };419 const tx = api.tx.nft.setCollectionLimits(collectionId, newLimits);420 const events = await submitTransactionAsync(sender, tx);421 const result = getGenericResult(events);422423 expect(result.success).to.be.true;424 });425}426427export async function setCollectionLimitsExpectFailure(sender: IKeyringPair, collectionId: number, limits: any) {428 await usingApi(async (api) => {429 const oldLimits = await queryCollectionLimits(collectionId);430 const newLimits = { ...oldLimits as any, ...limits };431 const tx = api.tx.nft.setCollectionLimits(collectionId, newLimits);432 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;433 const result = getGenericResult(events);434435 expect(result.success).to.be.false;436 });437}438439export async function setCollectionSponsorExpectSuccess(collectionId: number, sponsor: string, sender = '//Alice') {440 await usingApi(async (api) => {441442 443 const senderPrivateKey = privateKey(sender);444 const tx = api.tx.nft.setCollectionSponsor(collectionId, sponsor);445 const events = await submitTransactionAsync(senderPrivateKey, tx);446 const result = getGenericResult(events);447448 449 const collection: any = (await api.query.nft.collectionById(collectionId)).toJSON();450451 452 expect(result.success).to.be.true;453 expect(collection.Sponsorship).to.deep.equal({454 unconfirmed: sponsor,455 });456 });457}458459export async function removeCollectionSponsorExpectSuccess(collectionId: number, sender = '//Alice') {460 await usingApi(async (api) => {461462 463 const alicePrivateKey = privateKey(sender);464 const tx = api.tx.nft.removeCollectionSponsor(collectionId);465 const events = await submitTransactionAsync(alicePrivateKey, tx);466 const result = getGenericResult(events);467468 469 const collection: any = (await api.query.nft.collectionById(collectionId)).toJSON();470471 472 expect(result.success).to.be.true;473 expect(collection.Sponsorship).to.be.deep.equal({ disabled: null });474 });475}476477export async function removeCollectionSponsorExpectFailure(collectionId: number, senderSeed = '//Alice') {478 await usingApi(async (api) => {479480 481 const alicePrivateKey = privateKey(senderSeed);482 const tx = api.tx.nft.removeCollectionSponsor(collectionId);483 await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;484 });485}486487export async function setCollectionSponsorExpectFailure(collectionId: number, sponsor: string, senderSeed = '//Alice') {488 await usingApi(async (api) => {489490 491 const alicePrivateKey = privateKey(senderSeed);492 const tx = api.tx.nft.setCollectionSponsor(collectionId, sponsor);493 await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;494 });495}496497export async function confirmSponsorshipExpectSuccess(collectionId: number, senderSeed = '//Alice') {498 await usingApi(async (api) => {499500 501 const sender = privateKey(senderSeed);502 const tx = api.tx.nft.confirmSponsorship(collectionId);503 const events = await submitTransactionAsync(sender, tx);504 const result = getGenericResult(events);505506 507 const collection: any = (await api.query.nft.collectionById(collectionId)).toJSON();508509 510 expect(result.success).to.be.true;511 expect(collection.Sponsorship).to.be.deep.equal({512 confirmed: sender.address,513 });514 });515}516517518export async function confirmSponsorshipExpectFailure(collectionId: number, senderSeed = '//Alice') {519 await usingApi(async (api) => {520521 522 const sender = privateKey(senderSeed);523 const tx = api.tx.nft.confirmSponsorship(collectionId);524 await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;525 });526}527528export async function enableContractSponsoringExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, enable: boolean) {529 await usingApi(async (api) => {530 const tx = api.tx.nft.enableContractSponsoring(contractAddress, enable);531 const events = await submitTransactionAsync(sender, tx);532 const result = getGenericResult(events);533534 expect(result.success).to.be.true;535 });536}537538export async function enableContractSponsoringExpectFailure(sender: IKeyringPair, contractAddress: AccountId | string, enable: boolean) {539 await usingApi(async (api) => {540 const tx = api.tx.nft.enableContractSponsoring(contractAddress, enable);541 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;542 const result = getGenericResult(events);543544 expect(result.success).to.be.false;545 });546}547548export async function setTransferFlagExpectSuccess(sender: IKeyringPair, collectionId: number, enabled: boolean) {549550 await usingApi(async (api) => {551552 const tx = api.tx.nft.setTransfersEnabledFlag (collectionId, enabled);553 const events = await submitTransactionAsync(sender, tx);554 const result = getGenericResult(events);555556 expect(result.success).to.be.true;557 }); 558}559560export async function setTransferFlagExpectFailure(sender: IKeyringPair, collectionId: number, enabled: boolean) {561562 await usingApi(async (api) => {563564 const tx = api.tx.nft.setTransfersEnabledFlag (collectionId, enabled);565 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;566 const result = getGenericResult(events);567568 expect(result.success).to.be.false;569 }); 570}571572export async function setContractSponsoringRateLimitExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, rateLimit: number) {573 await usingApi(async (api) => {574 const tx = api.tx.nft.setContractSponsoringRateLimit(contractAddress, rateLimit);575 const events = await submitTransactionAsync(sender, tx);576 const result = getGenericResult(events);577578 expect(result.success).to.be.true;579 });580}581582export async function setContractSponsoringRateLimitExpectFailure(sender: IKeyringPair, contractAddress: AccountId | string, rateLimit: number) {583 await usingApi(async (api) => {584 const tx = api.tx.nft.setContractSponsoringRateLimit(contractAddress, rateLimit);585 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;586 const result = getGenericResult(events);587588 expect(result.success).to.be.false;589 });590}591592export async function toggleContractWhitelistExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, value = true) {593 await usingApi(async (api) => {594 const tx = api.tx.nft.toggleContractWhiteList(contractAddress, value);595 const events = await submitTransactionAsync(sender, tx);596 const result = getGenericResult(events);597598 expect(result.success).to.be.true;599 });600}601602export async function isWhitelistedInContract(contractAddress: AccountId | string, user: string) {603 let whitelisted = false;604 await usingApi(async (api) => {605 whitelisted = (await api.query.nft.contractWhiteList(contractAddress, user)).toJSON() as boolean;606 });607 return whitelisted;608}609610export async function addToContractWhiteListExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, user: AccountId | string) {611 await usingApi(async (api) => {612 const tx = api.tx.nft.addToContractWhiteList(contractAddress.toString(), user.toString());613 const events = await submitTransactionAsync(sender, tx);614 const result = getGenericResult(events);615616 expect(result.success).to.be.true;617 });618}619620export async function removeFromContractWhiteListExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, user: AccountId | string) {621 await usingApi(async (api) => {622 const tx = api.tx.nft.removeFromContractWhiteList(contractAddress.toString(), user.toString());623 const events = await submitTransactionAsync(sender, tx);624 const result = getGenericResult(events);625626 expect(result.success).to.be.true;627 });628}629630export async function removeFromContractWhiteListExpectFailure(sender: IKeyringPair, contractAddress: AccountId | string, user: AccountId | string) {631 await usingApi(async (api) => {632 const tx = api.tx.nft.removeFromContractWhiteList(contractAddress.toString(), user.toString());633 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;634 const result = getGenericResult(events);635636 expect(result.success).to.be.false;637 });638}639640export async function setVariableMetaDataExpectSuccess(sender: IKeyringPair, collectionId: number, itemId: number, data: number[]) {641 await usingApi(async (api) => {642 const tx = api.tx.nft.setVariableMetaData(collectionId, itemId, '0x' + Buffer.from(data).toString('hex'));643 const events = await submitTransactionAsync(sender, tx);644 const result = getGenericResult(events);645646 expect(result.success).to.be.true;647 });648}649650export async function setVariableMetaDataExpectFailure(sender: IKeyringPair, collectionId: number, itemId: number, data: number[]) {651 await usingApi(async (api) => {652 const tx = api.tx.nft.setVariableMetaData(collectionId, itemId, '0x' + Buffer.from(data).toString('hex'));653 await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;654 });655}656657export async function setOffchainSchemaExpectSuccess(sender: IKeyringPair, collectionId: number, data: number[]) {658 await usingApi(async (api) => {659 const tx = api.tx.nft.setOffchainSchema(collectionId, '0x' + Buffer.from(data).toString('hex'));660 const events = await submitTransactionAsync(sender, tx);661 const result = getGenericResult(events);662663 expect(result.success).to.be.true;664 });665}666667export async function setOffchainSchemaExpectFailure(sender: IKeyringPair, collectionId: number, data: number[]) {668 await usingApi(async (api) => {669 const tx = api.tx.nft.setOffchainSchema(collectionId, '0x' + Buffer.from(data).toString('hex'));670 await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;671 });672}673674export interface CreateFungibleData {675 readonly Value: bigint;676}677678export interface CreateReFungibleData { }679export interface CreateNftData { }680681export type CreateItemData = {682 NFT: CreateNftData;683} | {684 Fungible: CreateFungibleData;685} | {686 ReFungible: CreateReFungibleData;687};688689export async function burnItemExpectSuccess(owner: IKeyringPair, collectionId: number, tokenId: number, value = 0) {690 await usingApi(async (api) => {691 const tx = api.tx.nft.burnItem(collectionId, tokenId, value);692 const events = await submitTransactionAsync(owner, tx);693 const result = getGenericResult(events);694 695 const item: any = (await api.query.nft.nftItemList(collectionId, tokenId)).toJSON();696 697 698 expect(result.success).to.be.true;699 700 expect(item).to.be.null;701 });702}703704export async function705approveExpectSuccess(706 collectionId: number,707 tokenId: number, owner: IKeyringPair, approved: IKeyringPair | CrossAccountId | string, amount: number | bigint = 1,708) {709 await usingApi(async (api: ApiPromise) => {710 approved = normalizeAccountId(approved);711 const allowanceBefore =712 await api.query.nft.allowances(collectionId, [tokenId, owner.address, toSubstrateAddress(approved)]) as unknown as BN;713 const approveNftTx = api.tx.nft.approve(approved, collectionId, tokenId, amount);714 const events = await submitTransactionAsync(owner, approveNftTx);715 const result = getCreateItemResult(events);716 717 expect(result.success).to.be.true;718 const allowanceAfter =719 await api.query.nft.allowances(collectionId, [tokenId, owner.address, toSubstrateAddress(approved)]) as unknown as BN;720 expect(allowanceAfter.sub(allowanceBefore).toString()).to.be.equal(amount.toString());721 });722}723724export async function725transferFromExpectSuccess(726 collectionId: number,727 tokenId: number,728 accountApproved: IKeyringPair,729 accountFrom: IKeyringPair | CrossAccountId,730 accountTo: IKeyringPair | CrossAccountId,731 value: number | bigint = 1,732 type = 'NFT',733) {734 await usingApi(async (api: ApiPromise) => {735 const to = normalizeAccountId(accountTo);736 let balanceBefore = new BN(0);737 if (type === 'Fungible') {738 balanceBefore = await api.query.nft.balance(collectionId, toSubstrateAddress(to)) as unknown as BN;739 }740 const transferFromTx = api.tx.nft.transferFrom(normalizeAccountId(accountFrom), to, collectionId, tokenId, value);741 const events = await submitTransactionAsync(accountApproved, transferFromTx);742 const result = getCreateItemResult(events);743 744 expect(result.success).to.be.true;745 if (type === 'NFT') {746 const nftItemData = (await api.query.nft.nftItemList(collectionId, tokenId) as any).toJSON() as ITokenDataType;747 expect(nftItemData.Owner).to.be.deep.equal(to);748 }749 if (type === 'Fungible') {750 const balanceAfter = (await api.query.nft.fungibleItemList(collectionId, toSubstrateAddress(to)) as any).Value as unknown as BN;751 expect(balanceAfter.sub(balanceBefore).toString()).to.be.equal(value.toString());752 }753 if (type === 'ReFungible') {754 const nftItemData =755 (await api.query.nft.reFungibleItemList(collectionId, tokenId) as any).toJSON() as IReFungibleTokenDataType;756 expect(nftItemData.Owner[0].Owner).to.be.deep.equal(normalizeAccountId(to));757 expect(nftItemData.Owner[0].Fraction).to.be.equal(value);758 }759 });760}761762export async function763transferFromExpectFail(764 collectionId: number,765 tokenId: number,766 accountApproved: IKeyringPair,767 accountFrom: IKeyringPair,768 accountTo: IKeyringPair,769 value: number | bigint = 1,770) {771 await usingApi(async (api: ApiPromise) => {772 const transferFromTx = api.tx.nft.transferFrom(normalizeAccountId(accountFrom.address), normalizeAccountId(accountTo.address), collectionId, tokenId, value);773 const events = await expect(submitTransactionExpectFailAsync(accountApproved, transferFromTx)).to.be.rejected;774 const result = getCreateCollectionResult(events);775 776 expect(result.success).to.be.false;777 });778}779780781async function getBlockNumber(api: ApiPromise): Promise<number> {782 return new Promise<number>(async (resolve) => {783 const unsubscribe = await api.rpc.chain.subscribeNewHeads((head) => {784 unsubscribe();785 resolve(head.number.toNumber());786 });787 });788}789790export async function addCollectionAdminExpectSuccess(sender: IKeyringPair, collectionId: number, address: IKeyringPair) {791 await usingApi(async (api) => {792 const changeAdminTx = api.tx.nft.addCollectionAdmin(collectionId, normalizeAccountId(address.address));793 const events = await submitTransactionAsync(sender, changeAdminTx);794 const result = getCreateCollectionResult(events);795 expect(result.success).to.be.true;796 });797}798799export async function800scheduleTransferExpectSuccess(801 collectionId: number,802 tokenId: number,803 sender: IKeyringPair,804 recipient: IKeyringPair,805 value: number | bigint = 1,806 blockTimeMs: number,807 blockSchedule: number,808) {809 await usingApi(async (api: ApiPromise) => {810 const blockNumber: number | undefined = await getBlockNumber(api);811 const expectedBlockNumber = blockNumber + blockSchedule;812813 expect(blockNumber).to.be.greaterThan(0);814 const transferTx = await api.tx.nft.transfer(normalizeAccountId(recipient.address), collectionId, tokenId, value); 815 const scheduleTx = await api.tx.scheduler.schedule(expectedBlockNumber, null, 0, transferTx);816817 await submitTransactionAsync(sender, scheduleTx);818819 const recipientBalanceBefore = new BigNumber((await api.query.system.account(recipient.address)).data.free.toString());820821 const nftItemDataBefore = (await api.query.nft.nftItemList(collectionId, tokenId)).toJSON() as any as ITokenDataType;822 expect(toSubstrateAddress(nftItemDataBefore.Owner)).to.be.equal(sender.address);823824 825 await new Promise(resolve => setTimeout(resolve, blockTimeMs * (blockSchedule + 1)));826827 const recipientBalanceAfter = new BigNumber((await api.query.system.account(recipient.address)).data.free.toString());828829 const nftItemData = (await api.query.nft.nftItemList(collectionId, tokenId)).toJSON() as unknown as ITokenDataType;830 expect(toSubstrateAddress(nftItemData.Owner)).to.be.equal(recipient.address);831 expect(recipientBalanceAfter.toNumber()).to.be.equal(recipientBalanceBefore.toNumber());832 });833}834835836export async function837transferExpectSuccess(838 collectionId: number,839 tokenId: number,840 sender: IKeyringPair,841 recipient: IKeyringPair | CrossAccountId,842 value: number | bigint = 1,843 type = 'NFT',844) {845 await usingApi(async (api: ApiPromise) => {846 const to = normalizeAccountId(recipient);847848 let balanceBefore = new BN(0);849 if (type === 'Fungible') {850 balanceBefore = await api.query.nft.balance(collectionId, toSubstrateAddress(to)) as unknown as BN;851 }852 const transferTx = api.tx.nft.transfer(to, collectionId, tokenId, value);853 const events = await submitTransactionAsync(sender, transferTx);854 const result = getTransferResult(events);855 856 expect(result.success).to.be.true;857 expect(result.collectionId).to.be.equal(collectionId);858 expect(result.itemId).to.be.equal(tokenId);859 expect(result.sender).to.be.deep.equal(normalizeAccountId(sender.address));860 expect(result.recipient).to.be.deep.equal(to);861 expect(result.value.toString()).to.be.equal(value.toString());862 if (type === 'NFT') {863 const nftItemData = (await api.query.nft.nftItemList(collectionId, tokenId)).toJSON() as unknown as ITokenDataType;864 expect(nftItemData.Owner).to.be.deep.equal(to);865 }866 if (type === 'Fungible') {867 const balanceAfter = (await api.query.nft.fungibleItemList(collectionId, toSubstrateAddress(to)) as any).Value as unknown as BN;868 expect(balanceAfter.sub(balanceBefore).toString()).to.be.equal(value.toString());869 }870 if (type === 'ReFungible') {871 const nftItemData =872 (await api.query.nft.reFungibleItemList(collectionId, tokenId)).toJSON() as unknown as IReFungibleTokenDataType;873 expect(nftItemData.Owner[0].Owner).to.be.deep.equal(to);874 expect(nftItemData.Owner[0].Fraction.toString()).to.be.equal(value.toString());875 }876 });877}878879export async function880transferExpectFailure(881 collectionId: number,882 tokenId: number,883 sender: IKeyringPair,884 recipient: IKeyringPair,885 value: number | bigint = 1,886) {887 await usingApi(async (api: ApiPromise) => {888 const transferTx = api.tx.nft.transfer(normalizeAccountId(recipient.address), collectionId, tokenId, value);889 const events = await expect(submitTransactionExpectFailAsync(sender, transferTx)).to.be.rejected;890 if (events && Array.isArray(events)) {891 const result = getCreateCollectionResult(events);892 893 expect(result.success).to.be.false;894 }895 });896}897898export async function899approveExpectFail(900 collectionId: number,901 tokenId: number, owner: IKeyringPair, approved: IKeyringPair, amount: number | bigint = 1,902) {903 await usingApi(async (api: ApiPromise) => {904 const approveNftTx = api.tx.nft.approve(normalizeAccountId(approved.address), collectionId, tokenId, amount);905 const events = await expect(submitTransactionExpectFailAsync(owner, approveNftTx)).to.be.rejected;906 const result = getCreateCollectionResult(events);907 908 expect(result.success).to.be.false;909 });910}911912export async function getFungibleBalance(913 collectionId: number,914 owner: string,915) {916 return await usingApi(async (api) => {917 const response = (await api.query.nft.fungibleItemList(collectionId, owner)).toJSON() as unknown as { Value: string };918 return BigInt(response.Value);919 });920}921922export async function createFungibleItemExpectSuccess(923 sender: IKeyringPair,924 collectionId: number,925 data: CreateFungibleData,926 owner: CrossAccountId | string = sender.address,927) {928 return await usingApi(async (api) => {929 const tx = api.tx.nft.createItem(collectionId, normalizeAccountId(owner), { Fungible: data });930931 const events = await submitTransactionAsync(sender, tx);932 const result = getCreateItemResult(events);933934 expect(result.success).to.be.true;935 return result.itemId;936 });937}938939export async function createItemExpectSuccess(sender: IKeyringPair, collectionId: number, createMode: string, owner: CrossAccountId | string = sender.address) {940 let newItemId = 0;941 await usingApi(async (api) => {942 const to = normalizeAccountId(owner);943 const AItemCount = parseInt((await api.query.nft.itemListIndex(collectionId)).toString(), 10);944 const Aitem: any = (await api.query.nft.fungibleItemList(collectionId, toSubstrateAddress(to))).toJSON();945 const AItemBalance = new BigNumber(Aitem.Value);946947 let tx;948 if (createMode === 'Fungible') {949 const createData = { fungible: { value: 10 } };950 tx = api.tx.nft.createItem(collectionId, to, createData);951 } else if (createMode === 'ReFungible') {952 const createData = { refungible: { const_data: [], variable_data: [], pieces: 100 } };953 tx = api.tx.nft.createItem(collectionId, to, createData);954 } else {955 const createData = { nft: { const_data: [], variable_data: [] } };956 tx = api.tx.nft.createItem(collectionId, to, createData);957 }958959 const events = await submitTransactionAsync(sender, tx);960 const result = getCreateItemResult(events);961962 const BItemCount = parseInt((await api.query.nft.itemListIndex(collectionId)).toString(), 10);963 const Bitem: any = (await api.query.nft.fungibleItemList(collectionId, toSubstrateAddress(to))).toJSON();964 const BItemBalance = new BigNumber(Bitem.Value);965966 967 968 expect(result.success).to.be.true;969 if (createMode === 'Fungible') {970 expect(BItemBalance.minus(AItemBalance).toNumber()).to.be.equal(10);971 } else {972 expect(BItemCount).to.be.equal(AItemCount + 1);973 }974 expect(collectionId).to.be.equal(result.collectionId);975 expect(BItemCount.toString()).to.be.equal(result.itemId.toString());976 expect(to).to.be.deep.equal(result.recipient);977 newItemId = result.itemId;978 });979 return newItemId;980}981982export async function createItemExpectFailure(sender: IKeyringPair, collectionId: number, createMode: string, owner: string = sender.address) {983 await usingApi(async (api) => {984 const tx = api.tx.nft.createItem(collectionId, normalizeAccountId(owner), createMode);985 986 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;987 const result = getCreateItemResult(events);988989 expect(result.success).to.be.false;990 });991}992993export async function setPublicAccessModeExpectSuccess(994 sender: IKeyringPair, collectionId: number,995 accessMode: 'Normal' | 'WhiteList',996) {997 await usingApi(async (api) => {998999 1000 const tx = api.tx.nft.setPublicAccessMode(collectionId, accessMode);1001 const events = await submitTransactionAsync(sender, tx);1002 const result = getGenericResult(events);10031004 1005 const collection: any = (await api.query.nft.collectionById(collectionId)).toJSON();10061007 1008 1009 expect(result.success).to.be.true;1010 expect(collection.Access).to.be.equal(accessMode);1011 });1012}10131014export async function setPublicAccessModeExpectFail(1015 sender: IKeyringPair, collectionId: number,1016 accessMode: 'Normal' | 'WhiteList',1017) {1018 await usingApi(async (api) => {10191020 1021 const tx = api.tx.nft.setPublicAccessMode(collectionId, accessMode);1022 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;1023 const result = getGenericResult(events);10241025 1026 1027 expect(result.success).to.be.false;1028 });1029}10301031export async function enableWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number) {1032 await setPublicAccessModeExpectSuccess(sender, collectionId, 'WhiteList');1033}10341035export async function enableWhiteListExpectFail(sender: IKeyringPair, collectionId: number) {1036 await setPublicAccessModeExpectFail(sender, collectionId, 'WhiteList');1037}10381039export async function disableWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number) {1040 await setPublicAccessModeExpectSuccess(sender, collectionId, 'Normal');1041}10421043export async function setMintPermissionExpectSuccess(sender: IKeyringPair, collectionId: number, enabled: boolean) {1044 await usingApi(async (api) => {10451046 1047 const tx = api.tx.nft.setMintPermission(collectionId, enabled);1048 const events = await submitTransactionAsync(sender, tx);1049 const result = getGenericResult(events);10501051 1052 const collection: any = (await api.query.nft.collectionById(collectionId)).toJSON();10531054 1055 1056 expect(result.success).to.be.true;1057 expect(collection.MintMode).to.be.equal(enabled);1058 });1059}10601061export async function enablePublicMintingExpectSuccess(sender: IKeyringPair, collectionId: number) {1062 await setMintPermissionExpectSuccess(sender, collectionId, true);1063}10641065export async function setMintPermissionExpectFailure(sender: IKeyringPair, collectionId: number, enabled: boolean) {1066 await usingApi(async (api) => {1067 1068 const tx = api.tx.nft.setMintPermission(collectionId, enabled);1069 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;1070 const result = getCreateCollectionResult(events);1071 1072 expect(result.success).to.be.false;1073 });1074}10751076export async function setChainLimitsExpectFailure(sender: IKeyringPair, limits: IChainLimits) {1077 await usingApi(async (api) => {1078 1079 const tx = api.tx.nft.setChainLimits(limits);1080 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;1081 const result = getCreateCollectionResult(events);1082 1083 expect(result.success).to.be.false;1084 });1085}10861087export async function isWhitelisted(collectionId: number, address: string) {1088 let whitelisted = false;1089 await usingApi(async (api) => {1090 whitelisted = (await api.query.nft.whiteList(collectionId, address)).toJSON() as unknown as boolean;1091 });1092 return whitelisted;1093}10941095export async function addToWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number, address: string | AccountId) {1096 await usingApi(async (api) => {10971098 const whiteListedBefore = (await api.query.nft.whiteList(collectionId, address)).toJSON();10991100 1101 const tx = api.tx.nft.addToWhiteList(collectionId, normalizeAccountId(address));1102 const events = await submitTransactionAsync(sender, tx);1103 const result = getGenericResult(events);11041105 const whiteListedAfter = (await api.query.nft.whiteList(collectionId, address)).toJSON();11061107 1108 1109 expect(result.success).to.be.true;1110 1111 expect(whiteListedBefore).to.be.false;1112 1113 expect(whiteListedAfter).to.be.true;1114 });1115}11161117export async function addToWhiteListExpectFail(sender: IKeyringPair, collectionId: number, address: string | AccountId) {1118 await usingApi(async (api) => {1119 1120 const tx = api.tx.nft.addToWhiteList(collectionId, normalizeAccountId(address));1121 const events = await expect(submitTransactionAsync(sender, tx)).to.be.rejected;1122 const result = getGenericResult(events);11231124 1125 1126 expect(result.success).to.be.false;1127 });1128}11291130export async function removeFromWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number, address: CrossAccountId) {1131 await usingApi(async (api) => {1132 1133 const tx = api.tx.nft.removeFromWhiteList(collectionId, normalizeAccountId(address));1134 const events = await submitTransactionAsync(sender, tx);1135 const result = getGenericResult(events);11361137 1138 1139 expect(result.success).to.be.true;1140 });1141}11421143export async function removeFromWhiteListExpectFailure(sender: IKeyringPair, collectionId: number, address: CrossAccountId) {1144 await usingApi(async (api) => {1145 1146 const tx = api.tx.nft.removeFromWhiteList(collectionId, normalizeAccountId(address));1147 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;1148 const result = getGenericResult(events);11491150 1151 1152 expect(result.success).to.be.false;1153 });1154}11551156export const getDetailedCollectionInfo = async (api: ApiPromise, collectionId: number)1157 : Promise<ICollectionInterface | null> => {1158 return (await api.query.nft.collectionById(collectionId)).toJSON() as unknown as ICollectionInterface;1159};11601161export const getCreatedCollectionCount = async (api: ApiPromise): Promise<number> => {1162 1163 return (await api.query.nft.createdCollectionCount() as unknown as BN).toNumber();1164};11651166export async function queryCollectionExpectSuccess(collectionId: number): Promise<ICollectionInterface> {1167 return await usingApi(async (api) => {1168 return (await api.query.nft.collectionById(collectionId)).toJSON() as unknown as ICollectionInterface;1169 });1170}11711172export async function queryNftOwner(api: ApiPromise, collectionId: number, tokenId: number): Promise<CrossAccountId> {1173 return normalizeAccountId((await api.query.nft.nftItemList(collectionId, tokenId) as any).toJSON().Owner);1174}