`96967bb10fde`

source

tests/src/util/helpers.ts7.3 KiBsource history
1//2// This file is subject to the terms and conditions defined in3// file 'LICENSE', which is part of this source code package.4//56import chai from 'chai';7import chaiAsPromised from 'chai-as-promised';8import type { AccountId, EventRecord } from '@polkadot/types/interfaces';9import { ApiPromise, Keyring } from "@polkadot/api";10import { default as usingApi, submitTransactionAsync } from "../substrate/substrate-api";11import privateKey from '../substrate/privateKey';12import { alicesPublicKey, nullPublicKey } from "../accounts";13import { strToUTF16, utf16ToStr, hexToStr } from '../util/util';14import { IKeyringPair } from "@polkadot/types/types";15import { BigNumber } from 'bignumber.js';1617chai.use(chaiAsPromised);18const expect = chai.expect;1920type GenericResult = {21  success: boolean,22};2324type CreateCollectionResult = {25  success: boolean,26  collectionId: number27};2829export function getGenericResult(events: EventRecord[]): GenericResult {30  let result: GenericResult = {31    success: false32  }33  events.forEach(({ phase, event: { data, method, section } }) => {34    // console.log(`    ${phase}: ${section}.${method}:: ${data}`);35    if (method == 'ExtrinsicSuccess') {36      result.success = true;37    }38  });39  return result;40}4142function getCreateCollectionResult(events: EventRecord[]): CreateCollectionResult {43  let success = false;44  let collectionId: number = 0;45  events.forEach(({ phase, event: { data, method, section } }) => {46    // console.log(`    ${phase}: ${section}.${method}:: ${data}`);47    if (method == 'ExtrinsicSuccess') {48      success = true;49    } else if ((section == 'nft') && (method == 'Created')) {50      collectionId = parseInt(data[0].toString());51    }52  });53  let result: CreateCollectionResult = {54    success,55    collectionId56  }57  return result;58}5960export async function createCollectionExpectSuccess(name: string, description: string, tokenPrefix: string, mode: string): Promise<number> {61  let collectionId: number = 0;62  await usingApi(async (api) => {63    // Get number of collections before the transaction64    const AcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString());6566    // Run the CreateCollection transaction67    const alicePrivateKey = privateKey('//Alice');68    const tx = api.tx.nft.createCollection(strToUTF16(name), strToUTF16(description), strToUTF16(tokenPrefix), mode);69    const events = await submitTransactionAsync(alicePrivateKey, tx);70    const result = getCreateCollectionResult(events);7172    // Get number of collections after the transaction73    const BcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString());7475    // Get the collection 76    const collection: any = (await api.query.nft.collection(result.collectionId)).toJSON();7778    // What to expect79    expect(result.success).to.be.true;80    expect(result.collectionId).to.be.equal(BcollectionCount);81    expect(collection).to.be.not.null;82    expect(BcollectionCount).to.be.equal(AcollectionCount+1, 'Error: NFT collection NOT created.');83    expect(collection.Owner).to.be.equal(alicesPublicKey);84    expect(utf16ToStr(collection.Name)).to.be.equal(name);85    expect(utf16ToStr(collection.Description)).to.be.equal(description);86    expect(hexToStr(collection.TokenPrefix)).to.be.equal(tokenPrefix);8788    collectionId = result.collectionId;89  });9091  return collectionId;92}93  94export async function createCollectionExpectFailure(name: string, description: string, tokenPrefix: string, mode: string) {95  await usingApi(async (api) => {96    // Get number of collections before the transaction97    const AcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString());9899    // Run the CreateCollection transaction100    const alicePrivateKey = privateKey('//Alice');101    const tx = api.tx.nft.createCollection(name, description, tokenPrefix, mode);102    const events = await submitTransactionAsync(alicePrivateKey, tx);103    const result = getCreateCollectionResult(events);104105    // Get number of collections after the transaction106    const BcollectionCount = parseInt((await api.query.nft.createdCollectionCount()).toString());107108    // What to expect109    expect(result.success).to.be.false;110    expect(BcollectionCount).to.be.equal(AcollectionCount, 'Error: Collection with incorrect data created.');111  });112}113  114export async function findUnusedAddress(api: ApiPromise): Promise<IKeyringPair> {115  let bal = new BigNumber(0);116  let unused;117  do {118    const randomSeed = 'seed' +  Math.floor(Math.random() * Math.floor(10000));119    const keyring = new Keyring({ type: 'sr25519' });120    unused = keyring.addFromUri(`//${randomSeed}`);121    bal = new BigNumber((await api.query.system.account(unused.address)).data.free.toString());122  } while (bal.toFixed() != '0');123  return unused; 124}125126function getDestroyResult(events: EventRecord[]): boolean {127  let success: boolean = false;128  events.forEach(({ phase, event: { data, method, section } }) => {129    // console.log(`    ${phase}: ${section}.${method}:: ${data}`);130    if (method == 'ExtrinsicSuccess') {131      success = true;132    }133  });134  return success;135}136137export async function setCollectionSponsorExpectSuccess(collectionId: number, sponsor: string) {138  await usingApi(async (api) => {139140    // Run the transaction141    const alicePrivateKey = privateKey('//Alice');142    const tx = api.tx.nft.setCollectionSponsor(collectionId, sponsor);143    const events = await submitTransactionAsync(alicePrivateKey, tx);144    const result = getGenericResult(events);145146    // Get the collection 147    const collection: any = (await api.query.nft.collection(collectionId)).toJSON();148149    // What to expect150    expect(result.success).to.be.true;151    expect(collection.Sponsor.toString()).to.be.equal(sponsor.toString());152    expect(collection.SponsorConfirmed).to.be.false;153  });154}155156export async function destroyCollectionExpectFailure(collectionId: number, senderSeed: string = '//Alice') {157  await usingApi(async (api) => {158    // Run the DestroyCollection transaction159    const alicePrivateKey = privateKey(senderSeed);160    const tx = api.tx.nft.destroyCollection(collectionId);161    const events = await submitTransactionAsync(alicePrivateKey, tx);162    const result = getDestroyResult(events);163164    // What to expect165    expect(result).to.be.false;166  });167}168169export async function destroyCollectionExpectSuccess(collectionId: number, senderSeed: string = '//Alice') {170  await usingApi(async (api) => {171    // Run the DestroyCollection transaction172    const alicePrivateKey = privateKey(senderSeed);173    const tx = api.tx.nft.destroyCollection(collectionId);174    const events = await submitTransactionAsync(alicePrivateKey, tx);175    const result = getDestroyResult(events);176177    // Get the collection 178    const collection: any = (await api.query.nft.collection(collectionId)).toJSON();179180    // What to expect181    expect(result).to.be.true;182    expect(collection).to.be.not.null;183    expect(collection.Owner).to.be.equal(nullPublicKey);184  });185}186187export async function setCollectionSponsorExpectFailure(collectionId: number, sponsor: string) {188  await usingApi(async (api) => {189190    // Run the transaction191    const alicePrivateKey = privateKey('//Alice');192    const tx = api.tx.nft.setCollectionSponsor(collectionId, sponsor);193    const events = await submitTransactionAsync(alicePrivateKey, tx);194    const result = getGenericResult(events);195196    // What to expect197    expect(result.success).to.be.false;198  });199}