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 function800getFreeBalance(account: IKeyringPair) : Promise<BigNumber>801{802 let balance = new BigNumber(0) ;803 await usingApi(async (api) => { 804 balance = new BigNumber((await api.query.system.account(account.address)).data.free.toString()); 805 });806807 return balance;808}809810export async function811scheduleTransferExpectSuccess(812 collectionId: number,813 tokenId: number,814 sender: IKeyringPair,815 recipient: IKeyringPair,816 value: number | bigint = 1,817 blockTimeMs: number,818 blockSchedule: number,819) {820 await usingApi(async (api: ApiPromise) => {821 const blockNumber: number | undefined = await getBlockNumber(api);822 const expectedBlockNumber = blockNumber + blockSchedule;823824 expect(blockNumber).to.be.greaterThan(0);825 const transferTx = await api.tx.nft.transfer(normalizeAccountId(recipient.address), collectionId, tokenId, value); 826 const scheduleTx = await api.tx.scheduler.schedule(expectedBlockNumber, null, 0, transferTx);827828 await submitTransactionAsync(sender, scheduleTx);829830 const recipientBalanceBefore = new BigNumber((await api.query.system.account(recipient.address)).data.free.toString());831832 const nftItemDataBefore = (await api.query.nft.nftItemList(collectionId, tokenId)).toJSON() as any as ITokenDataType;833 expect(toSubstrateAddress(nftItemDataBefore.Owner)).to.be.equal(sender.address);834835 836 await new Promise(resolve => setTimeout(resolve, blockTimeMs * (blockSchedule + 1)));837838 const recipientBalanceAfter = new BigNumber((await api.query.system.account(recipient.address)).data.free.toString());839840 const nftItemData = (await api.query.nft.nftItemList(collectionId, tokenId)).toJSON() as unknown as ITokenDataType;841 expect(toSubstrateAddress(nftItemData.Owner)).to.be.equal(recipient.address);842 expect(recipientBalanceAfter.toNumber()).to.be.equal(recipientBalanceBefore.toNumber());843 });844}845846847export async function848transferExpectSuccess(849 collectionId: number,850 tokenId: number,851 sender: IKeyringPair,852 recipient: IKeyringPair | CrossAccountId,853 value: number | bigint = 1,854 type = 'NFT',855) {856 await usingApi(async (api: ApiPromise) => {857 const to = normalizeAccountId(recipient);858859 let balanceBefore = new BN(0);860 if (type === 'Fungible') {861 balanceBefore = await api.query.nft.balance(collectionId, toSubstrateAddress(to)) as unknown as BN;862 }863 const transferTx = api.tx.nft.transfer(to, collectionId, tokenId, value);864 const events = await submitTransactionAsync(sender, transferTx);865 const result = getTransferResult(events);866 867 expect(result.success).to.be.true;868 expect(result.collectionId).to.be.equal(collectionId);869 expect(result.itemId).to.be.equal(tokenId);870 expect(result.sender).to.be.deep.equal(normalizeAccountId(sender.address));871 expect(result.recipient).to.be.deep.equal(to);872 expect(result.value.toString()).to.be.equal(value.toString());873 if (type === 'NFT') {874 const nftItemData = (await api.query.nft.nftItemList(collectionId, tokenId)).toJSON() as unknown as ITokenDataType;875 expect(nftItemData.Owner).to.be.deep.equal(to);876 }877 if (type === 'Fungible') {878 const balanceAfter = (await api.query.nft.fungibleItemList(collectionId, toSubstrateAddress(to)) as any).Value as unknown as BN;879 expect(balanceAfter.sub(balanceBefore).toString()).to.be.equal(value.toString());880 }881 if (type === 'ReFungible') {882 const nftItemData =883 (await api.query.nft.reFungibleItemList(collectionId, tokenId)).toJSON() as unknown as IReFungibleTokenDataType;884 const expectedOwner = toSubstrateAddress(to);885 const ownerIndex = nftItemData.Owner.findIndex(v => toSubstrateAddress(v.Owner as any as string) == expectedOwner);886 expect(ownerIndex).to.not.equal(-1);887 expect(nftItemData.Owner[ownerIndex].Owner).to.be.deep.equal(normalizeAccountId(to));888 expect(nftItemData.Owner[ownerIndex].Fraction).to.be.greaterThanOrEqual(value as number);889 }890 });891}892893export async function894transferExpectFailure(895 collectionId: number,896 tokenId: number,897 sender: IKeyringPair,898 recipient: IKeyringPair,899 value: number | bigint = 1,900) {901 await usingApi(async (api: ApiPromise) => {902 const transferTx = api.tx.nft.transfer(normalizeAccountId(recipient.address), collectionId, tokenId, value);903 const events = await expect(submitTransactionExpectFailAsync(sender, transferTx)).to.be.rejected;904 if (events && Array.isArray(events)) {905 const result = getCreateCollectionResult(events);906 907 expect(result.success).to.be.false;908 }909 });910}911912export async function913approveExpectFail(914 collectionId: number,915 tokenId: number, owner: IKeyringPair, approved: IKeyringPair, amount: number | bigint = 1,916) {917 await usingApi(async (api: ApiPromise) => {918 const approveNftTx = api.tx.nft.approve(normalizeAccountId(approved.address), collectionId, tokenId, amount);919 const events = await expect(submitTransactionExpectFailAsync(owner, approveNftTx)).to.be.rejected;920 const result = getCreateCollectionResult(events);921 922 expect(result.success).to.be.false;923 });924}925926export async function getFungibleBalance(927 collectionId: number,928 owner: string,929) {930 return await usingApi(async (api) => {931 const response = (await api.query.nft.fungibleItemList(collectionId, owner)).toJSON() as unknown as { Value: string };932 return BigInt(response.Value);933 });934}935936export async function createFungibleItemExpectSuccess(937 sender: IKeyringPair,938 collectionId: number,939 data: CreateFungibleData,940 owner: CrossAccountId | string = sender.address,941) {942 return await usingApi(async (api) => {943 const tx = api.tx.nft.createItem(collectionId, normalizeAccountId(owner), { Fungible: data });944945 const events = await submitTransactionAsync(sender, tx);946 const result = getCreateItemResult(events);947948 expect(result.success).to.be.true;949 return result.itemId;950 });951}952953export async function createItemExpectSuccess(sender: IKeyringPair, collectionId: number, createMode: string, owner: CrossAccountId | string = sender.address) {954 let newItemId = 0;955 await usingApi(async (api) => {956 const to = normalizeAccountId(owner);957 const AItemCount = parseInt((await api.query.nft.itemListIndex(collectionId)).toString(), 10);958 const Aitem: any = (await api.query.nft.fungibleItemList(collectionId, toSubstrateAddress(to))).toJSON();959 const AItemBalance = new BigNumber(Aitem.Value);960961 let tx;962 if (createMode === 'Fungible') {963 const createData = { fungible: { value: 10 } };964 tx = api.tx.nft.createItem(collectionId, to, createData);965 } else if (createMode === 'ReFungible') {966 const createData = { refungible: { const_data: [], variable_data: [], pieces: 100 } };967 tx = api.tx.nft.createItem(collectionId, to, createData);968 } else {969 const createData = { nft: { const_data: [], variable_data: [] } };970 tx = api.tx.nft.createItem(collectionId, to, createData);971 }972973 const events = await submitTransactionAsync(sender, tx);974 const result = getCreateItemResult(events);975976 const BItemCount = parseInt((await api.query.nft.itemListIndex(collectionId)).toString(), 10);977 const Bitem: any = (await api.query.nft.fungibleItemList(collectionId, toSubstrateAddress(to))).toJSON();978 const BItemBalance = new BigNumber(Bitem.Value);979980 981 982 expect(result.success).to.be.true;983 if (createMode === 'Fungible') {984 expect(BItemBalance.minus(AItemBalance).toNumber()).to.be.equal(10);985 } else {986 expect(BItemCount).to.be.equal(AItemCount + 1);987 }988 expect(collectionId).to.be.equal(result.collectionId);989 expect(BItemCount.toString()).to.be.equal(result.itemId.toString());990 expect(to).to.be.deep.equal(result.recipient);991 newItemId = result.itemId;992 });993 return newItemId;994}995996export async function createItemExpectFailure(sender: IKeyringPair, collectionId: number, createMode: string, owner: string = sender.address) {997 await usingApi(async (api) => {998 const tx = api.tx.nft.createItem(collectionId, normalizeAccountId(owner), createMode);999 1000 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;1001 const result = getCreateItemResult(events);10021003 expect(result.success).to.be.false;1004 });1005}10061007export async function setPublicAccessModeExpectSuccess(1008 sender: IKeyringPair, collectionId: number,1009 accessMode: 'Normal' | 'WhiteList',1010) {1011 await usingApi(async (api) => {10121013 1014 const tx = api.tx.nft.setPublicAccessMode(collectionId, accessMode);1015 const events = await submitTransactionAsync(sender, tx);1016 const result = getGenericResult(events);10171018 1019 const collection: any = (await api.query.nft.collectionById(collectionId)).toJSON();10201021 1022 1023 expect(result.success).to.be.true;1024 expect(collection.Access).to.be.equal(accessMode);1025 });1026}10271028export async function setPublicAccessModeExpectFail(1029 sender: IKeyringPair, collectionId: number,1030 accessMode: 'Normal' | 'WhiteList',1031) {1032 await usingApi(async (api) => {10331034 1035 const tx = api.tx.nft.setPublicAccessMode(collectionId, accessMode);1036 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;1037 const result = getGenericResult(events);10381039 1040 1041 expect(result.success).to.be.false;1042 });1043}10441045export async function enableWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number) {1046 await setPublicAccessModeExpectSuccess(sender, collectionId, 'WhiteList');1047}10481049export async function enableWhiteListExpectFail(sender: IKeyringPair, collectionId: number) {1050 await setPublicAccessModeExpectFail(sender, collectionId, 'WhiteList');1051}10521053export async function disableWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number) {1054 await setPublicAccessModeExpectSuccess(sender, collectionId, 'Normal');1055}10561057export async function setMintPermissionExpectSuccess(sender: IKeyringPair, collectionId: number, enabled: boolean) {1058 await usingApi(async (api) => {10591060 1061 const tx = api.tx.nft.setMintPermission(collectionId, enabled);1062 const events = await submitTransactionAsync(sender, tx);1063 const result = getGenericResult(events);10641065 1066 const collection: any = (await api.query.nft.collectionById(collectionId)).toJSON();10671068 1069 1070 expect(result.success).to.be.true;1071 expect(collection.MintMode).to.be.equal(enabled);1072 });1073}10741075export async function enablePublicMintingExpectSuccess(sender: IKeyringPair, collectionId: number) {1076 await setMintPermissionExpectSuccess(sender, collectionId, true);1077}10781079export async function setMintPermissionExpectFailure(sender: IKeyringPair, collectionId: number, enabled: boolean) {1080 await usingApi(async (api) => {1081 1082 const tx = api.tx.nft.setMintPermission(collectionId, enabled);1083 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;1084 const result = getCreateCollectionResult(events);1085 1086 expect(result.success).to.be.false;1087 });1088}10891090export async function setChainLimitsExpectFailure(sender: IKeyringPair, limits: IChainLimits) {1091 await usingApi(async (api) => {1092 1093 const tx = api.tx.nft.setChainLimits(limits);1094 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;1095 const result = getCreateCollectionResult(events);1096 1097 expect(result.success).to.be.false;1098 });1099}11001101export async function isWhitelisted(collectionId: number, address: string) {1102 let whitelisted = false;1103 await usingApi(async (api) => {1104 whitelisted = (await api.query.nft.whiteList(collectionId, address)).toJSON() as unknown as boolean;1105 });1106 return whitelisted;1107}11081109export async function addToWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number, address: string | AccountId) {1110 await usingApi(async (api) => {11111112 const whiteListedBefore = (await api.query.nft.whiteList(collectionId, address)).toJSON();11131114 1115 const tx = api.tx.nft.addToWhiteList(collectionId, normalizeAccountId(address));1116 const events = await submitTransactionAsync(sender, tx);1117 const result = getGenericResult(events);11181119 const whiteListedAfter = (await api.query.nft.whiteList(collectionId, address)).toJSON();11201121 1122 1123 expect(result.success).to.be.true;1124 1125 expect(whiteListedBefore).to.be.false;1126 1127 expect(whiteListedAfter).to.be.true;1128 });1129}11301131export async function addToWhiteListExpectFail(sender: IKeyringPair, collectionId: number, address: string | AccountId) {1132 await usingApi(async (api) => {1133 1134 const tx = api.tx.nft.addToWhiteList(collectionId, normalizeAccountId(address));1135 const events = await expect(submitTransactionAsync(sender, tx)).to.be.rejected;1136 const result = getGenericResult(events);11371138 1139 1140 expect(result.success).to.be.false;1141 });1142}11431144export async function removeFromWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number, address: CrossAccountId) {1145 await usingApi(async (api) => {1146 1147 const tx = api.tx.nft.removeFromWhiteList(collectionId, normalizeAccountId(address));1148 const events = await submitTransactionAsync(sender, tx);1149 const result = getGenericResult(events);11501151 1152 1153 expect(result.success).to.be.true;1154 });1155}11561157export async function removeFromWhiteListExpectFailure(sender: IKeyringPair, collectionId: number, address: CrossAccountId) {1158 await usingApi(async (api) => {1159 1160 const tx = api.tx.nft.removeFromWhiteList(collectionId, normalizeAccountId(address));1161 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;1162 const result = getGenericResult(events);11631164 1165 1166 expect(result.success).to.be.false;1167 });1168}11691170export const getDetailedCollectionInfo = async (api: ApiPromise, collectionId: number)1171 : Promise<ICollectionInterface | null> => {1172 return (await api.query.nft.collectionById(collectionId)).toJSON() as unknown as ICollectionInterface;1173};11741175export const getCreatedCollectionCount = async (api: ApiPromise): Promise<number> => {1176 1177 return (await api.query.nft.createdCollectionCount() as unknown as BN).toNumber();1178};11791180export async function queryCollectionExpectSuccess(collectionId: number): Promise<ICollectionInterface> {1181 return await usingApi(async (api) => {1182 return (await api.query.nft.collectionById(collectionId)).toJSON() as unknown as ICollectionInterface;1183 });1184}11851186export async function queryNftOwner(api: ApiPromise, collectionId: number, tokenId: number): Promise<CrossAccountId> {1187 return normalizeAccountId((await api.query.nft.nftItemList(collectionId, tokenId) as any).toJSON().Owner);1188}11891190export async function waitNewBlocks(blocksCount = 1): Promise<void> {1191 await usingApi(async (api) => {1192 const promise = new Promise<void>(async (resolve) => {1193 1194 const unsubscribe = await api.rpc.chain.subscribeNewHeads(() => {1195 if (blocksCount > 0) {1196 blocksCount--;1197 } else {1198 unsubscribe();1199 resolve();1200 }1201 });1202 });1203 return promise;1204 });1205}