123456import { ApiPromise, Keyring } from '@polkadot/api';7import { Enum, Struct } from '@polkadot/types/codec';8import type { AccountId, EventRecord } from '@polkadot/types/interfaces';9import { u128 } from '@polkadot/types/primitive';10import { IKeyringPair } from '@polkadot/types/types';11import { BigNumber } from 'bignumber.js';12import BN from 'bn.js';13import chai from 'chai';14import chaiAsPromised from 'chai-as-promised';15import { alicesPublicKey, nullPublicKey } from '../accounts';16import privateKey from '../substrate/privateKey';17import { default as usingApi, submitTransactionAsync, submitTransactionExpectFailAsync } from '../substrate/substrate-api';18import { ICollectionInterface } from '../types';19import { hexToStr, strToUTF16, utf16ToStr } from './util';2021chai.use(chaiAsPromised);22const expect = chai.expect;2324type GenericResult = {25 success: boolean,26};2728interface CreateCollectionResult {29 success: boolean;30 collectionId: number;31}3233interface CreateItemResult {34 success: boolean;35 collectionId: number;36 itemId: number;37}3839interface IReFungibleOwner {40 Fraction: BN;41 Owner: number[];42}4344interface ITokenDataType {45 Owner: number[];46 ConstData: number[];47 VariableData: number[];48}4950interface IFungibleTokenDataType {51 Value: BN;52}5354interface IReFungibleTokenDataType {55 Owner: IReFungibleOwner[];56 ConstData: number[];57 VariableData: number[];58}5960export function getGenericResult(events: EventRecord[]): GenericResult {61 const result: GenericResult = {62 success: false,63 };64 events.forEach(({ phase, event: { data, method, section } }) => {65 66 if (method === 'ExtrinsicSuccess') {67 result.success = true;68 }69 });70 return result;71}7273export function getCreateCollectionResult(events: EventRecord[]): CreateCollectionResult {74 let success = false;75 let collectionId: number = 0;76 events.forEach(({ phase, event: { data, method, section } }) => {77 78 if (method == 'ExtrinsicSuccess') {79 success = true;80 } else if ((section == 'nft') && (method == 'Created')) {81 collectionId = parseInt(data[0].toString());82 }83 });84 const result: CreateCollectionResult = {85 success,86 collectionId,87 };88 return result;89}9091export function getCreateItemResult(events: EventRecord[]): CreateItemResult {92 let success = false;93 let collectionId: number = 0;94 let itemId: number = 0;95 events.forEach(({ phase, event: { data, method, section } }) => {96 97 if (method == 'ExtrinsicSuccess') {98 success = true;99 } else if ((section == 'nft') && (method == 'ItemCreated')) {100 collectionId = parseInt(data[0].toString());101 itemId = parseInt(data[1].toString());102 }103 });104 const result: CreateItemResult = {105 success,106 collectionId,107 itemId,108 };109 return result;110}111112interface Invalid {113 type: 'Invalid';114}115116interface Nft {117 type: 'NFT';118}119120interface Fungible {121 type: 'Fungible';122 decimalPoints: number;123}124125interface ReFungible {126 type: 'ReFungible';127 decimalPoints: number;128}129130interface Nft {131 type: 'NFT'132}133134interface Fungible {135 type: 'Fungible',136 decimalPoints: number137}138139interface ReFungible {140 type: 'ReFungible',141 decimalPoints: number142}143144type CollectionMode = Nft | Fungible | ReFungible | Invalid;145146export type CreateCollectionParams = {147 mode: CollectionMode,148 name: string,149 description: string,150 tokenPrefix: string,151};152153const defaultCreateCollectionParams: CreateCollectionParams = {154 description: 'description',155 mode: { type: 'NFT' },156 name: 'name',157 tokenPrefix: 'prefix',158}159160export async function createCollectionExpectSuccess(params: Partial<CreateCollectionParams> = {}): Promise<number> {161 const {name, description, mode, tokenPrefix } = {...defaultCreateCollectionParams, ...params};162163 let collectionId: number = 0;164 await usingApi(async (api) => {165 166 const AcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString(), 10);167168 169 const alicePrivateKey = privateKey('//Alice');170171 let modeprm = {};172 if (mode.type === 'NFT') {173 modeprm = {nft: null};174 } else if (mode.type === 'Fungible') {175 modeprm = {fungible: mode.decimalPoints};176 } else if (mode.type === 'ReFungible') {177 modeprm = {refungible: mode.decimalPoints};178 } else if (mode.type === 'Invalid') {179 modeprm = {invalid: null};180 }181182 const tx = api.tx.nft.createCollection(strToUTF16(name), strToUTF16(description), strToUTF16(tokenPrefix), modeprm);183 const events = await submitTransactionAsync(alicePrivateKey, tx);184 const result = getCreateCollectionResult(events);185186 187 const BcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString(), 10);188189 190 const collection: any = (await api.query.nft.collection(result.collectionId)).toJSON();191192 193 194 expect(result.success).to.be.true;195 expect(result.collectionId).to.be.equal(BcollectionCount);196 197 expect(collection).to.be.not.null;198 expect(BcollectionCount).to.be.equal(AcollectionCount + 1, 'Error: NFT collection NOT created.');199 expect(collection.Owner).to.be.equal(alicesPublicKey);200 expect(utf16ToStr(collection.Name)).to.be.equal(name);201 expect(utf16ToStr(collection.Description)).to.be.equal(description);202 expect(hexToStr(collection.TokenPrefix)).to.be.equal(tokenPrefix);203204 collectionId = result.collectionId;205 });206207 return collectionId;208}209210export async function createCollectionExpectFailure(params: Partial<CreateCollectionParams> = {}) {211 const {name, description, mode, tokenPrefix } = {...defaultCreateCollectionParams, ...params};212213 let modeprm = {};214 if (mode.type === 'NFT') {215 modeprm = {nft: null};216 } else if (mode.type === 'Fungible') {217 modeprm = {fungible: mode.decimalPoints};218 } else if (mode.type === 'ReFungible') {219 modeprm = {refungible: mode.decimalPoints};220 } else if (mode.type === 'Invalid') {221 modeprm = {invalid: null};222 }223224 await usingApi(async (api) => {225 226 const AcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString());227228 229 const alicePrivateKey = privateKey('//Alice');230 const tx = api.tx.nft.createCollection(strToUTF16(name), strToUTF16(description), strToUTF16(tokenPrefix), modeprm);231 const events = await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;232 const result = getCreateCollectionResult(events);233234 235 const BcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString());236237 238 239 expect(result.success).to.be.false;240 expect(BcollectionCount).to.be.equal(AcollectionCount, 'Error: Collection with incorrect data created.');241 });242}243244export async function findUnusedAddress(api: ApiPromise): Promise<IKeyringPair> {245 let bal = new BigNumber(0);246 let unused;247 do {248 const randomSeed = 'seed' + Math.floor(Math.random() * Math.floor(10000));249 const keyring = new Keyring({ type: 'sr25519' });250 unused = keyring.addFromUri(`//${randomSeed}`);251 bal = new BigNumber((await api.query.system.account(unused.address)).data.free.toString());252 } while (bal.toFixed() != '0');253 return unused;254}255256export async function findNotExistingCollection(api: ApiPromise): Promise<number> {257 const totalNumber = parseInt((await api.query.nft.createdCollectionCount()).toString(), 10) as unknown as number;258 const newCollection: number = totalNumber + 1;259 return newCollection;260}261262function getDestroyResult(events: EventRecord[]): boolean {263 let success: boolean = false;264 events.forEach(({ phase, event: { data, method, section } }) => {265 266 if (method == 'ExtrinsicSuccess') {267 success = true;268 }269 });270 return success;271}272273export async function destroyCollectionExpectFailure(collectionId: number, senderSeed: string = '//Alice') {274 await usingApi(async (api) => {275 276 const alicePrivateKey = privateKey(senderSeed);277 const tx = api.tx.nft.destroyCollection(collectionId);278 await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;279 });280}281282export async function destroyCollectionExpectSuccess(collectionId: number, senderSeed: string = '//Alice') {283 await usingApi(async (api) => {284 285 const alicePrivateKey = privateKey(senderSeed);286 const tx = api.tx.nft.destroyCollection(collectionId);287 const events = await submitTransactionAsync(alicePrivateKey, tx);288 const result = getDestroyResult(events);289290 291 const collection: any = (await api.query.nft.collection(collectionId)).toJSON();292293 294 expect(result).to.be.true;295 expect(collection).to.be.not.null;296 expect(collection.Owner).to.be.equal(nullPublicKey);297 });298}299300export async function setCollectionSponsorExpectSuccess(collectionId: number, sponsor: string) {301 await usingApi(async (api) => {302303 304 const alicePrivateKey = privateKey('//Alice');305 const tx = api.tx.nft.setCollectionSponsor(collectionId, sponsor);306 const events = await submitTransactionAsync(alicePrivateKey, tx);307 const result = getGenericResult(events);308309 310 const collection: any = (await api.query.nft.collection(collectionId)).toJSON();311312 313 expect(result.success).to.be.true;314 expect(collection.Sponsor.toString()).to.be.equal(sponsor.toString());315 expect(collection.SponsorConfirmed).to.be.false;316 });317}318319export async function removeCollectionSponsorExpectSuccess(collectionId: number) {320 await usingApi(async (api) => {321322 323 const alicePrivateKey = privateKey('//Alice');324 const tx = api.tx.nft.removeCollectionSponsor(collectionId);325 const events = await submitTransactionAsync(alicePrivateKey, tx);326 const result = getGenericResult(events);327328 329 const collection: any = (await api.query.nft.collection(collectionId)).toJSON();330331 332 expect(result.success).to.be.true;333 expect(collection.Sponsor).to.be.equal(nullPublicKey);334 expect(collection.SponsorConfirmed).to.be.false;335 });336}337338export async function removeCollectionSponsorExpectFailure(collectionId: number) {339 await usingApi(async (api) => {340341 342 const alicePrivateKey = privateKey('//Alice');343 const tx = api.tx.nft.removeCollectionSponsor(collectionId);344 await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;345 });346}347348export async function setCollectionSponsorExpectFailure(collectionId: number, sponsor: string, senderSeed: string = '//Alice') {349 await usingApi(async (api) => {350351 352 const alicePrivateKey = privateKey(senderSeed);353 const tx = api.tx.nft.setCollectionSponsor(collectionId, sponsor);354 await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;355 });356}357358export async function confirmSponsorshipExpectSuccess(collectionId: number, senderSeed: string = '//Alice') {359 await usingApi(async (api) => {360361 362 const sender = privateKey(senderSeed);363 const tx = api.tx.nft.confirmSponsorship(collectionId);364 const events = await submitTransactionAsync(sender, tx);365 const result = getGenericResult(events);366367 368 const collection: any = (await api.query.nft.collection(collectionId)).toJSON();369370 371 expect(result.success).to.be.true;372 expect(collection.Sponsor).to.be.equal(sender.address);373 expect(collection.SponsorConfirmed).to.be.true;374 });375}376377export async function confirmSponsorshipExpectFailure(collectionId: number, senderSeed: string = '//Alice') {378 await usingApi(async (api) => {379380 381 const sender = privateKey(senderSeed);382 const tx = api.tx.nft.confirmSponsorship(collectionId);383 await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;384 });385}386387export async function enableContractSponsoringExpectSuccess(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 submitTransactionAsync(sender, tx);391 const result = getGenericResult(events);392393 expect(result.success).to.be.true;394 });395}396397export async function enableContractSponsoringExpectFailure(sender: IKeyringPair, contractAddress: AccountId | string, enable: boolean) {398 await usingApi(async (api) => {399 const tx = api.tx.nft.enableContractSponsoring(contractAddress, enable);400 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;401 const result = getGenericResult(events);402403 expect(result.success).to.be.false;404 });405}406407export async function setContractSponsoringRateLimitExpectSuccess(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 submitTransactionAsync(sender, tx);411 const result = getGenericResult(events);412413 expect(result.success).to.be.true;414 });415}416417export async function setVariableMetaDataExpectSuccess(sender: IKeyringPair, collectionId: number, itemId: number, data: number[]) {418 await usingApi(async (api) => {419 const tx = api.tx.nft.setVariableMetaData(collectionId, itemId, '0x' + Buffer.from(data).toString('hex'));420 const events = await submitTransactionAsync(sender, tx);421 const result = getGenericResult(events);422423 expect(result.success).to.be.true;424 });425}426427export async function setVariableMetaDataExpectFailure(sender: IKeyringPair, collectionId: number, itemId: number, data: number[]) {428 await usingApi(async (api) => {429 const tx = api.tx.nft.setVariableMetaData(collectionId, itemId, '0x' + Buffer.from(data).toString('hex'));430 await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;431 });432}433434export interface CreateFungibleData extends Struct {435 readonly value: u128;436}437438export interface CreateReFungibleData extends Struct {}439export interface CreateNftData extends Struct {}440441export interface CreateItemData extends Enum {442 NFT: CreateNftData;443 Fungible: CreateFungibleData;444 ReFungible: CreateReFungibleData;445}446447export async function burnItemExpectSuccess(owner: IKeyringPair, collectionId: number, tokenId: number, value = 0) {448 await usingApi(async (api) => {449 const tx = api.tx.nft.burnItem(collectionId, tokenId, value);450 const events = await submitTransactionAsync(owner, tx);451 const result = getGenericResult(events);452 453 const item: any = (await api.query.nft.nftItemList(collectionId, tokenId)).toJSON();454 455 456 expect(result.success).to.be.true;457 458 expect(item).to.be.not.null;459 expect(item.Owner).to.be.equal(nullPublicKey);460 });461}462463export async function464approveExpectSuccess(collectionId: number,465 tokenId: number, owner: IKeyringPair, approved: IKeyringPair, amount: number = 1) {466 await usingApi(async (api: ApiPromise) => {467 const allowanceBefore =468 await api.query.nft.allowances(collectionId, [tokenId, owner.address, approved.address]) as unknown as BN;469 const approveNftTx = await api.tx.nft.approve(approved.address, collectionId, tokenId, amount);470 const events = await submitTransactionAsync(owner, approveNftTx);471 const result = getCreateItemResult(events);472 473 expect(result.success).to.be.true;474 const allowanceAfter =475 await api.query.nft.allowances(collectionId, [tokenId, owner.address, approved.address]) as unknown as BN;476 expect(allowanceAfter.toNumber() - allowanceBefore.toNumber()).to.be.equal(amount);477 });478}479480export async function481transferFromExpectSuccess(collectionId: number,482 tokenId: number,483 accountApproved: IKeyringPair,484 accountFrom: IKeyringPair,485 accountTo: IKeyringPair,486 value: number = 1,487 type: string = 'NFT') {488 await usingApi(async (api: ApiPromise) => {489 let balanceBefore = new BN(0);490 if (type === 'Fungible') {491 balanceBefore = await api.query.nft.balance(collectionId, accountTo.address) as unknown as BN;492 }493 const transferFromTx = await api.tx.nft.transferFrom(494 accountFrom.address, accountTo.address, collectionId, tokenId, value);495 const events = await submitTransactionAsync(accountFrom, transferFromTx);496 const result = getCreateItemResult(events);497 498 expect(result.success).to.be.true;499 if (type === 'NFT') {500 const nftItemData = await api.query.nft.nftItemList(collectionId, tokenId) as unknown as ITokenDataType;501 expect(nftItemData.Owner.toString()).to.be.equal(accountTo.address);502 }503 if (type === 'Fungible') {504 const balanceAfter = await api.query.nft.balance(collectionId, accountTo.address) as unknown as BN;505 expect(balanceAfter.sub(balanceBefore).toNumber()).to.be.equal(value);506 }507 if (type === 'ReFungible') {508 const nftItemData =509 await api.query.nft.reFungibleItemList(collectionId, tokenId) as unknown as IReFungibleTokenDataType;510 expect(nftItemData.Owner[0].Owner.toString()).to.be.equal(accountTo.address);511 expect(nftItemData.Owner[0].Fraction.toNumber()).to.be.equal(value);512 }513 });514}515516export async function517transferFromExpectFail(collectionId: number,518 tokenId: number,519 accountApproved: IKeyringPair,520 accountFrom: IKeyringPair,521 accountTo: IKeyringPair,522 value: number = 1) {523 await usingApi(async (api: ApiPromise) => {524 const transferFromTx = await api.tx.nft.transferFrom(525 accountFrom.address, accountTo.address, collectionId, tokenId, value);526 const events = await expect(submitTransactionExpectFailAsync(accountApproved, transferFromTx)).to.be.rejected;527 const result = getCreateCollectionResult(events);528 529 expect(result.success).to.be.false;530 });531}532533export async function534transferExpectSuccess(collectionId: number,535 tokenId: number,536 sender: IKeyringPair,537 recipient: IKeyringPair,538 value: number = 1,539 type: string = 'NFT') {540 await usingApi(async (api: ApiPromise) => {541 let balanceBefore = new BN(0);542 if (type === 'Fungible') {543 balanceBefore = await api.query.nft.balance(collectionId, recipient.address) as unknown as BN;544 }545 const transferTx = await api.tx.nft.transfer(recipient.address, collectionId, tokenId, value);546 const events = await submitTransactionAsync(sender, transferTx);547 const result = getCreateItemResult(events);548 549 expect(result.success).to.be.true;550 if (type === 'NFT') {551 const nftItemData = await api.query.nft.nftItemList(collectionId, tokenId) as unknown as ITokenDataType;552 expect(nftItemData.Owner.toString()).to.be.equal(recipient.address);553 }554 if (type === 'Fungible') {555 const balanceAfter = await api.query.nft.balance(collectionId, recipient.address) as unknown as BN;556 expect(balanceAfter.sub(balanceBefore).toNumber()).to.be.equal(value);557 }558 if (type === 'ReFungible') {559 const nftItemData =560 await api.query.nft.reFungibleItemList(collectionId, tokenId) as unknown as IReFungibleTokenDataType;561 expect(nftItemData.Owner[0].Owner.toString()).to.be.equal(recipient.address);562 expect(nftItemData.Owner[0].Fraction.toNumber()).to.be.equal(value);563 }564 });565}566567export async function568transferExpectFail(collectionId: number,569 tokenId: number,570 sender: IKeyringPair,571 recipient: IKeyringPair,572 value: number = 1,573 type: string = 'NFT') {574 await usingApi(async (api: ApiPromise) => {575 const transferTx = await api.tx.nft.transfer(recipient.address, collectionId, tokenId, value);576 const events = await expect(submitTransactionExpectFailAsync(sender, transferTx)).to.be.rejected;577 if (events && Array.isArray(events)) {578 const result = getCreateCollectionResult(events);579 580 expect(result.success).to.be.false;581 }582 });583}584585export async function586approveExpectFail(collectionId: number,587 tokenId: number, owner: IKeyringPair, approved: IKeyringPair, amount: number = 1) {588 await usingApi(async (api: ApiPromise) => {589 const approveNftTx = await api.tx.nft.approve(approved.address, collectionId, tokenId, amount);590 const events = await expect(submitTransactionExpectFailAsync(owner, approveNftTx)).to.be.rejected;591 const result = getCreateCollectionResult(events);592 593 expect(result.success).to.be.false;594 });595}596597export async function createItemExpectSuccess(598 sender: IKeyringPair, collectionId: number, createMode: string, owner: string = '') {599 let newItemId: number = 0;600 await usingApi(async (api) => {601 const AItemCount = parseInt((await api.query.nft.itemListIndex(collectionId)).toString(), 10);602 const Aitem: any = (await api.query.nft.fungibleItemList(collectionId, owner)).toJSON();603 const AItemBalance = new BigNumber(Aitem.Value);604605 if (owner === '') {606 owner = sender.address;607 }608609 let tx;610 if (createMode === 'Fungible') {611 const createData = {fungible: {value: 10}};612 tx = api.tx.nft.createItem(collectionId, owner, createData);613 } else {614 tx = api.tx.nft.createItem(collectionId, owner, createMode);615 }616 const events = await submitTransactionAsync(sender, tx);617 const result = getCreateItemResult(events);618619 const BItemCount = parseInt((await api.query.nft.itemListIndex(collectionId)).toString(), 10);620 const Bitem: any = (await api.query.nft.fungibleItemList(collectionId, owner)).toJSON();621 const BItemBalance = new BigNumber(Bitem.Value);622623 624 625 expect(result.success).to.be.true;626 if (createMode === 'Fungible') {627 expect(BItemBalance.minus(AItemBalance).toNumber()).to.be.equal(10);628 } else {629 expect(BItemCount).to.be.equal(AItemCount + 1);630 }631 expect(collectionId).to.be.equal(result.collectionId);632 expect(BItemCount).to.be.equal(result.itemId);633 newItemId = result.itemId;634 });635 return newItemId;636}637638export async function enableWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number) {639 await usingApi(async (api) => {640641 642 const tx = api.tx.nft.setPublicAccessMode(collectionId, 'WhiteList');643 const events = await submitTransactionAsync(sender, tx);644 const result = getGenericResult(events);645646 647 const collection: any = (await api.query.nft.collection(collectionId)).toJSON();648649 650 651 expect(result.success).to.be.true;652 expect(collection.Access).to.be.equal('WhiteList');653 });654}655656export async function enablePublicMintingExpectSuccess(sender: IKeyringPair, collectionId: number) {657 await usingApi(async (api) => {658659 660 const tx = api.tx.nft.setMintPermission(collectionId, true);661 const events = await submitTransactionAsync(sender, tx);662 const result = getGenericResult(events);663664 665 const collection: any = (await api.query.nft.collection(collectionId)).toJSON();666667 668 expect(result.success).to.be.true;669 expect(collection.MintMode).to.be.equal(true);670 });671}672673export async function addToWhiteListExpectSuccess(sender: IKeyringPair, collectionId: number, address: string) {674 await usingApi(async (api) => {675676 const whiteListedBefore = (await api.query.nft.whiteList(collectionId, address)).toJSON();677678 679 const tx = api.tx.nft.addToWhiteList(collectionId, address);680 const events = await submitTransactionAsync(sender, tx);681 const result = getGenericResult(events);682683 const whiteListedAfter = (await api.query.nft.whiteList(collectionId, address)).toJSON();684685 686 687 expect(result.success).to.be.true;688 689 expect(whiteListedBefore).to.be.false;690 691 expect(whiteListedAfter).to.be.true;692 });693}694695export const getDetailedCollectionInfo = async (api: ApiPromise, collectionId: number)696 : Promise<ICollectionInterface | null> => {697 return await api.query.nft.collection(collectionId) as unknown as ICollectionInterface;698};699700export const getCreatedCollectionCount = async (api: ApiPromise): Promise<number> => {701 702 return (await api.query.nft.createdCollectionCount() as unknown as BN).toNumber();703};