difftreelog
fix misprint
in: master
1 file changed
tests/src/util/helpers.tsdiffbeforeafterboth1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// This file is part of Unique Network.34// Unique Network is free software: you can redistribute it and/or modify5// it under the terms of the GNU General Public License as published by6// the Free Software Foundation, either version 3 of the License, or7// (at your option) any later version.89// Unique Network is distributed in the hope that it will be useful,10// but WITHOUT ANY WARRANTY; without even the implied warranty of11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the12// GNU General Public License for more details.1314// You should have received a copy of the GNU General Public License15// along with Unique Network. If not, see <http://www.gnu.org/licenses/>.1617import '../interfaces/augment-api-rpc';18import '../interfaces/augment-api-query';19import {ApiPromise, Keyring} from '@polkadot/api';20import type {AccountId, EventRecord} from '@polkadot/types/interfaces';21import {IKeyringPair} from '@polkadot/types/types';22import {evmToAddress} from '@polkadot/util-crypto';23import BN from 'bn.js';24import chai from 'chai';25import chaiAsPromised from 'chai-as-promised';26import {alicesPublicKey} from '../accounts';27import privateKey from '../substrate/privateKey';28import {default as usingApi, submitTransactionAsync, submitTransactionExpectFailAsync} from '../substrate/substrate-api';29import {hexToStr, strToUTF16, utf16ToStr} from './util';30import {UpDataStructsCollection} from '@polkadot/types/lookup';3132chai.use(chaiAsPromised);33const expect = chai.expect;3435export type CrossAccountId = {36 Substrate: string,37} | {38 Ethereum: string,39};4041export function normalizeAccountId(input: string | AccountId | CrossAccountId | IKeyringPair): CrossAccountId {42 if (typeof input === 'string') {43 if (input.length === 48 || input.length === 47) {44 return {Substrate: input};45 } else if (input.length === 42 && input.startsWith('0x')) {46 return {Ethereum: input.toLowerCase()};47 } else if (input.length === 40 && !input.startsWith('0x')) {48 return {Ethereum: '0x' + input.toLowerCase()};49 } else {50 throw new Error(`Unknown address format: "${input}"`);51 }52 }53 if ('address' in input) {54 return {Substrate: input.address};55 }56 if ('Ethereum' in input) {57 return {58 Ethereum: input.Ethereum.toLowerCase(),59 };60 } else if ('ethereum' in input) {61 return {62 Ethereum: (input as any).ethereum.toLowerCase(),63 };64 } else if ('Substrate' in input) {65 return input;66 } else if ('substrate' in input) {67 return {68 Substrate: (input as any).substrate,69 };70 }7172 // AccountId73 return {Substrate: input.toString()};74}75export function toSubstrateAddress(input: string | CrossAccountId | IKeyringPair): string {76 input = normalizeAccountId(input);77 if ('Substrate' in input) {78 return input.Substrate;79 } else {80 return evmToAddress(input.Ethereum);81 }82}8384export const U128_MAX = (1n << 128n) - 1n;8586const MICROUNIQUE = 1_000_000_000_000n;87const MILLIUNIQUE = 1_000n * MICROUNIQUE;88const CENTIUNIQUE = 10n * MILLIUNIQUE;89export const UNIQUE = 100n * CENTIUNIQUE;9091type GenericResult = {92 success: boolean,93};9495interface CreateCollectionResult {96 success: boolean;97 collectionId: number;98}99100interface CreateItemResult {101 success: boolean;102 collectionId: number;103 itemId: number;104 recipient?: CrossAccountId;105}106107interface TransferResult {108 success: boolean;109 collectionId: number;110 itemId: number;111 sender?: CrossAccountId;112 recipient?: CrossAccountId;113 value: bigint;114}115116interface IReFungibleOwner {117 fraction: BN;118 owner: number[];119}120121interface IGetMessage {122 checkMsgUnqMethod: string;123 checkMsgTrsMethod: string;124 checkMsgSysMethod: string;125}126127export interface IFungibleTokenDataType {128 value: number;129}130131export interface IChainLimits {132 collectionNumbersLimit: number;133 accountTokenOwnershipLimit: number;134 collectionsAdminsLimit: number;135 customDataLimit: number;136 nftSponsorTransferTimeout: number;137 fungibleSponsorTransferTimeout: number;138 refungibleSponsorTransferTimeout: number;139 offchainSchemaLimit: number;140 variableOnChainSchemaLimit: number;141 constOnChainSchemaLimit: number;142}143144export interface IReFungibleTokenDataType {145 owner: IReFungibleOwner[];146 constData: number[];147 variableData: number[];148}149150export function uniqueEventMessage(events: EventRecord[]): IGetMessage {151 let checkMsgUnqMethod = '';152 let checkMsgTrsMethod = '';153 let checkMsgSysMethod = '';154 events.forEach(({event: {method, section}}) => {155 if (section === 'common') {156 checkMsgUnqMethod = method;157 } else if (section === 'treasury') {158 checkMsgTrsMethod = method;159 } else if (section === 'system') {160 checkMsgSysMethod = method;161 } else { return null; }162 });163 const result: IGetMessage = {164 checkMsgUnqMethod,165 checkMsgTrsMethod,166 checkMsgSysMethod,167 };168 return result;169}170171export function getGenericResult(events: EventRecord[]): GenericResult {172 const result: GenericResult = {173 success: false,174 };175 events.forEach(({event: {method}}) => {176 // console.log(` ${phase}: ${section}.${method}:: ${data}`);177 if (method === 'ExtrinsicSuccess') {178 result.success = true;179 }180 });181 return result;182}183184185186export function getCreateCollectionResult(events: EventRecord[]): CreateCollectionResult {187 let success = false;188 let collectionId = 0;189 events.forEach(({event: {data, method, section}}) => {190 // console.log(` ${phase}: ${section}.${method}:: ${data}`);191 if (method == 'ExtrinsicSuccess') {192 success = true;193 } else if ((section == 'common') && (method == 'CollectionCreated')) {194 collectionId = parseInt(data[0].toString(), 10);195 }196 });197 const result: CreateCollectionResult = {198 success,199 collectionId,200 };201 return result;202}203204export function getCreateItemResult(events: EventRecord[]): CreateItemResult {205 let success = false;206 let collectionId = 0;207 let itemId = 0;208 let recipient;209 events.forEach(({event: {data, method, section}}) => {210 // console.log(` ${phase}: ${section}.${method}:: ${data}`);211 if (method == 'ExtrinsicSuccess') {212 success = true;213 } else if ((section == 'common') && (method == 'ItemCreated')) {214 collectionId = parseInt(data[0].toString(), 10);215 itemId = parseInt(data[1].toString(), 10);216 recipient = normalizeAccountId(data[2].toJSON() as any);217 }218 });219 const result: CreateItemResult = {220 success,221 collectionId,222 itemId,223 recipient,224 };225 return result;226}227228export function getTransferResult(events: EventRecord[]): TransferResult {229 const result: TransferResult = {230 success: false,231 collectionId: 0,232 itemId: 0,233 value: 0n,234 };235236 events.forEach(({event: {data, method, section}}) => {237 if (method === 'ExtrinsicSuccess') {238 result.success = true;239 } else if (section === 'common' && method === 'Transfer') {240 result.collectionId = +data[0].toString();241 result.itemId = +data[1].toString();242 result.sender = normalizeAccountId(data[2].toJSON() as any);243 result.recipient = normalizeAccountId(data[3].toJSON() as any);244 result.value = BigInt(data[4].toString());245 }246 });247248 return result;249}250251interface Nft {252 type: 'NFT';253}254255interface Fungible {256 type: 'Fungible';257 decimalPoints: number;258}259260interface ReFungible {261 type: 'ReFungible';262}263264type CollectionMode = Nft | Fungible | ReFungible;265266export type CreateCollectionParams = {267 mode: CollectionMode,268 name: string,269 description: string,270 tokenPrefix: string,271 shemaVersion: string,272};273274const defaultCreateCollectionParams: CreateCollectionParams = {275 description: 'description',276 mode: {type: 'NFT'},277 name: 'name',278 tokenPrefix: 'prefix',279 shemaVersion: 'ImageURL',280};281282export async function createCollectionExpectSuccess(params: Partial<CreateCollectionParams> = {}): Promise<number> {283 const {name, description, mode, tokenPrefix, shemaVersion} = {...defaultCreateCollectionParams, ...params};284285 let collectionId = 0;286 await usingApi(async (api) => {287 // Get number of collections before the transaction288 const collectionCountBefore = await getCreatedCollectionCount(api);289290 // Run the CreateCollection transaction291 const alicePrivateKey = privateKey('//Alice');292293 let modeprm = {};294 if (mode.type === 'NFT') {295 modeprm = {nft: null};296 } else if (mode.type === 'Fungible') {297 modeprm = {fungible: mode.decimalPoints};298 } else if (mode.type === 'ReFungible') {299 modeprm = {refungible: null};300 }301302 const tx = api.tx.unique.createCollectionEx({303 name: strToUTF16(name), 304 description: strToUTF16(description), 305 tokenPrefix: strToUTF16(tokenPrefix), 306 mode: modeprm as any,307 schemaVersion: shemaVersion,308 });309 const events = await submitTransactionAsync(alicePrivateKey, tx);310 const result = getCreateCollectionResult(events);311312 // Get number of collections after the transaction313 const collectionCountAfter = await getCreatedCollectionCount(api);314315 // Get the collection316 const collection = await queryCollectionExpectSuccess(api, result.collectionId);317318 // What to expect319 // tslint:disable-next-line:no-unused-expression320 expect(result.success).to.be.true;321 expect(result.collectionId).to.be.equal(collectionCountAfter);322 // tslint:disable-next-line:no-unused-expression323 expect(collection).to.be.not.null;324 expect(collectionCountAfter).to.be.equal(collectionCountBefore + 1, 'Error: NFT collection NOT created.');325 expect(collection.owner.toString()).to.be.equal(toSubstrateAddress(alicesPublicKey));326 expect(utf16ToStr(collection.name.toJSON() as any)).to.be.equal(name);327 expect(utf16ToStr(collection.description.toJSON() as any)).to.be.equal(description);328 expect(hexToStr(collection.tokenPrefix.toJSON())).to.be.equal(tokenPrefix);329330 collectionId = result.collectionId;331 });332333 return collectionId;334}335336export async function createCollectionExpectFailure(params: Partial<CreateCollectionParams> = {}) {337 const {name, description, mode, tokenPrefix} = {...defaultCreateCollectionParams, ...params};338339 let modeprm = {};340 if (mode.type === 'NFT') {341 modeprm = {nft: null};342 } else if (mode.type === 'Fungible') {343 modeprm = {fungible: mode.decimalPoints};344 } else if (mode.type === 'ReFungible') {345 modeprm = {refungible: null};346 }347348 await usingApi(async (api) => {349 // Get number of collections before the transaction350 const collectionCountBefore = await getCreatedCollectionCount(api);351352 // Run the CreateCollection transaction353 const alicePrivateKey = privateKey('//Alice');354 const tx = api.tx.unique.createCollectionEx({name: strToUTF16(name), description: strToUTF16(description), tokenPrefix: strToUTF16(tokenPrefix), mode: modeprm as any});355 await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;356357 // Get number of collections after the transaction358 const collectionCountAfter = await getCreatedCollectionCount(api);359360 // What to expect361 expect(collectionCountAfter).to.be.equal(collectionCountBefore, 'Error: Collection with incorrect data created.');362 });363}364365export async function findUnusedAddress(api: ApiPromise, seedAddition = ''): Promise<IKeyringPair> {366 let bal = 0n;367 let unused;368 do {369 const randomSeed = 'seed' + Math.floor(Math.random() * Math.floor(10000)) + seedAddition;370 const keyring = new Keyring({type: 'sr25519'});371 unused = keyring.addFromUri(`//${randomSeed}`);372 bal = (await api.query.system.account(unused.address)).data.free.toBigInt();373 } while (bal !== 0n);374 return unused;375}376377export async function getAllowance(api: ApiPromise, collectionId: number, owner: CrossAccountId | string, approved: CrossAccountId | string, tokenId: number) {378 return (await api.rpc.unique.allowance(collectionId, normalizeAccountId(owner), normalizeAccountId(approved), tokenId)).toBigInt();379}380381export function findUnusedAddresses(api: ApiPromise, amount: number): Promise<IKeyringPair[]> {382 return Promise.all(new Array(amount).fill(null).map(() => findUnusedAddress(api, '_' + Date.now())));383}384385export async function findNotExistingCollection(api: ApiPromise): Promise<number> {386 const totalNumber = await getCreatedCollectionCount(api);387 const newCollection: number = totalNumber + 1;388 return newCollection;389}390391function getDestroyResult(events: EventRecord[]): boolean {392 let success = false;393 events.forEach(({event: {method}}) => {394 if (method == 'ExtrinsicSuccess') {395 success = true;396 }397 });398 return success;399}400401export async function destroyCollectionExpectFailure(collectionId: number, senderSeed = '//Alice') {402 await usingApi(async (api) => {403 // Run the DestroyCollection transaction404 const alicePrivateKey = privateKey(senderSeed);405 const tx = api.tx.unique.destroyCollection(collectionId);406 await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;407 });408}409410export async function destroyCollectionExpectSuccess(collectionId: number, senderSeed = '//Alice') {411 await usingApi(async (api) => {412 // Run the DestroyCollection transaction413 const alicePrivateKey = privateKey(senderSeed);414 const tx = api.tx.unique.destroyCollection(collectionId);415 const events = await submitTransactionAsync(alicePrivateKey, tx);416 const result = getDestroyResult(events);417 expect(result).to.be.true;418419 // What to expect420 expect(await getDetailedCollectionInfo(api, collectionId)).to.be.null;421 });422}423424export async function setCollectionLimitsExpectSuccess(sender: IKeyringPair, collectionId: number, limits: any) {425 await usingApi(async (api) => {426 const tx = api.tx.unique.setCollectionLimits(collectionId, limits);427 const events = await submitTransactionAsync(sender, tx);428 const result = getGenericResult(events);429430 expect(result.success).to.be.true;431 });432}433434export async function setCollectionLimitsExpectFailure(sender: IKeyringPair, collectionId: number, limits: any) {435 await usingApi(async (api) => {436 const tx = api.tx.unique.setCollectionLimits(collectionId, limits);437 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;438 const result = getGenericResult(events);439440 expect(result.success).to.be.false;441 });442}443444export async function setCollectionSponsorExpectSuccess(collectionId: number, sponsor: string, sender = '//Alice') {445 await usingApi(async (api) => {446447 // Run the transaction448 const senderPrivateKey = privateKey(sender);449 const tx = api.tx.unique.setCollectionSponsor(collectionId, sponsor);450 const events = await submitTransactionAsync(senderPrivateKey, tx);451 const result = getGenericResult(events);452453 // Get the collection454 const collection = await queryCollectionExpectSuccess(api, collectionId);455456 // What to expect457 expect(result.success).to.be.true;458 expect(collection.sponsorship.toJSON()).to.deep.equal({459 unconfirmed: sponsor,460 });461 });462}463464export async function removeCollectionSponsorExpectSuccess(collectionId: number, sender = '//Alice') {465 await usingApi(async (api) => {466467 // Run the transaction468 const alicePrivateKey = privateKey(sender);469 const tx = api.tx.unique.removeCollectionSponsor(collectionId);470 const events = await submitTransactionAsync(alicePrivateKey, tx);471 const result = getGenericResult(events);472473 // Get the collection474 const collection = await queryCollectionExpectSuccess(api, collectionId);475476 // What to expect477 expect(result.success).to.be.true;478 expect(collection.sponsorship.toJSON()).to.be.deep.equal({disabled: null});479 });480}481482export async function removeCollectionSponsorExpectFailure(collectionId: number, senderSeed = '//Alice') {483 await usingApi(async (api) => {484485 // Run the transaction486 const alicePrivateKey = privateKey(senderSeed);487 const tx = api.tx.unique.removeCollectionSponsor(collectionId);488 await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;489 });490}491492export async function setCollectionSponsorExpectFailure(collectionId: number, sponsor: string, senderSeed = '//Alice') {493 await usingApi(async (api) => {494495 // Run the transaction496 const alicePrivateKey = privateKey(senderSeed);497 const tx = api.tx.unique.setCollectionSponsor(collectionId, sponsor);498 await expect(submitTransactionExpectFailAsync(alicePrivateKey, tx)).to.be.rejected;499 });500}501502export async function confirmSponsorshipExpectSuccess(collectionId: number, senderSeed = '//Alice') {503 await usingApi(async (api) => {504505 // Run the transaction506 const sender = privateKey(senderSeed);507 const tx = api.tx.unique.confirmSponsorship(collectionId);508 const events = await submitTransactionAsync(sender, tx);509 const result = getGenericResult(events);510511 // Get the collection512 const collection = await queryCollectionExpectSuccess(api, collectionId);513514 // What to expect515 expect(result.success).to.be.true;516 expect(collection.sponsorship.toJSON()).to.be.deep.equal({517 confirmed: sender.address,518 });519 });520}521522523export async function confirmSponsorshipExpectFailure(collectionId: number, senderSeed = '//Alice') {524 await usingApi(async (api) => {525526 // Run the transaction527 const sender = privateKey(senderSeed);528 const tx = api.tx.unique.confirmSponsorship(collectionId);529 await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;530 });531}532533export async function setMetadataUpdatePermissionFlagExpectSuccess(sender: IKeyringPair, collectionId: number, flag: string) {534535 await usingApi(async (api) => {536 const tx = api.tx.unique.setMetaUpdatePermissionFlag(collectionId, flag as any);537 const events = await submitTransactionAsync(sender, tx);538 const result = getGenericResult(events);539540 expect(result.success).to.be.true;541 });542}543544export async function setMetadataUpdatePermissionFlagExpectFailure(sender: IKeyringPair, collectionId: number, flag: string) {545546 await usingApi(async (api) => {547 const tx = api.tx.unique.setMetaUpdatePermissionFlag(collectionId, flag as any);548 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;549 const result = getGenericResult(events);550551 expect(result.success).to.be.false;552 });553}554555export async function enableContractSponsoringExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, enable: boolean) {556 await usingApi(async (api) => {557 const tx = api.tx.unique.enableContractSponsoring(contractAddress, enable);558 const events = await submitTransactionAsync(sender, tx);559 const result = getGenericResult(events);560561 expect(result.success).to.be.true;562 });563}564565export async function enableContractSponsoringExpectFailure(sender: IKeyringPair, contractAddress: AccountId | string, enable: boolean) {566 await usingApi(async (api) => {567 const tx = api.tx.unique.enableContractSponsoring(contractAddress, enable);568 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;569 const result = getGenericResult(events);570571 expect(result.success).to.be.false;572 });573}574575export async function setTransferFlagExpectSuccess(sender: IKeyringPair, collectionId: number, enabled: boolean) {576577 await usingApi(async (api) => {578579 const tx = api.tx.unique.setTransfersEnabledFlag(collectionId, enabled);580 const events = await submitTransactionAsync(sender, tx);581 const result = getGenericResult(events);582583 expect(result.success).to.be.true;584 });585}586587export async function setTransferFlagExpectFailure(sender: IKeyringPair, collectionId: number, enabled: boolean) {588589 await usingApi(async (api) => {590591 const tx = api.tx.unique.setTransfersEnabledFlag(collectionId, enabled);592 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;593 const result = getGenericResult(events);594595 expect(result.success).to.be.false;596 });597}598599export async function setContractSponsoringRateLimitExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, rateLimit: number) {600 await usingApi(async (api) => {601 const tx = api.tx.unique.setContractSponsoringRateLimit(contractAddress, rateLimit);602 const events = await submitTransactionAsync(sender, tx);603 const result = getGenericResult(events);604605 expect(result.success).to.be.true;606 });607}608609export async function setContractSponsoringRateLimitExpectFailure(sender: IKeyringPair, contractAddress: AccountId | string, rateLimit: number) {610 await usingApi(async (api) => {611 const tx = api.tx.unique.setContractSponsoringRateLimit(contractAddress, rateLimit);612 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;613 const result = getGenericResult(events);614615 expect(result.success).to.be.false;616 });617}618619export async function getNextSponsored(620 api: ApiPromise,621 collectionId: number,622 account: string | CrossAccountId,623 tokenId: number,624): Promise<number> {625 return Number((await api.rpc.unique.nextSponsored(collectionId, account, tokenId)).unwrapOr(-1));626}627628export async function toggleContractAllowlistExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, value = true) {629 await usingApi(async (api) => {630 const tx = api.tx.unique.toggleContractAllowList(contractAddress, value);631 const events = await submitTransactionAsync(sender, tx);632 const result = getGenericResult(events);633634 expect(result.success).to.be.true;635 });636}637638export async function isAllowlistedInContract(contractAddress: AccountId | string, user: string) {639 let allowlisted = false;640 await usingApi(async (api) => {641 allowlisted = (await api.query.unique.contractAllowList(contractAddress, user)).toJSON() as boolean;642 });643 return allowlisted;644}645646export async function addToContractAllowListExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, user: AccountId | string) {647 await usingApi(async (api) => {648 const tx = api.tx.unique.addToContractAllowList(contractAddress.toString(), user.toString());649 const events = await submitTransactionAsync(sender, tx);650 const result = getGenericResult(events);651652 expect(result.success).to.be.true;653 });654}655656export async function removeFromContractAllowListExpectSuccess(sender: IKeyringPair, contractAddress: AccountId | string, user: AccountId | string) {657 await usingApi(async (api) => {658 const tx = api.tx.unique.removeFromContractAllowList(contractAddress.toString(), user.toString());659 const events = await submitTransactionAsync(sender, tx);660 const result = getGenericResult(events);661662 expect(result.success).to.be.true;663 });664}665666export async function removeFromContractAllowListExpectFailure(sender: IKeyringPair, contractAddress: AccountId | string, user: AccountId | string) {667 await usingApi(async (api) => {668 const tx = api.tx.unique.removeFromContractAllowList(contractAddress.toString(), user.toString());669 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;670 const result = getGenericResult(events);671672 expect(result.success).to.be.false;673 });674}675676export async function setVariableMetaDataExpectSuccess(sender: IKeyringPair, collectionId: number, itemId: number, data: number[]) {677 await usingApi(async (api) => {678 const tx = api.tx.unique.setVariableMetaData(collectionId, itemId, '0x' + Buffer.from(data).toString('hex'));679 const events = await submitTransactionAsync(sender, tx);680 const result = getGenericResult(events);681682 expect(result.success).to.be.true;683 });684}685686export async function setVariableMetaDataExpectFailure(sender: IKeyringPair, collectionId: number, itemId: number, data: number[]) {687 await usingApi(async (api) => {688 const tx = api.tx.unique.setVariableMetaData(collectionId, itemId, '0x' + Buffer.from(data).toString('hex'));689 await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;690 });691}692693export async function setOffchainSchemaExpectSuccess(sender: IKeyringPair, collectionId: number, data: number[]) {694 await usingApi(async (api) => {695 const tx = api.tx.unique.setOffchainSchema(collectionId, '0x' + Buffer.from(data).toString('hex'));696 const events = await submitTransactionAsync(sender, tx);697 const result = getGenericResult(events);698699 expect(result.success).to.be.true;700 });701}702703export async function setOffchainSchemaExpectFailure(sender: IKeyringPair, collectionId: number, data: number[]) {704 await usingApi(async (api) => {705 const tx = api.tx.unique.setOffchainSchema(collectionId, '0x' + Buffer.from(data).toString('hex'));706 await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;707 });708}709710export interface CreateFungibleData {711 readonly Value: bigint;712}713714export interface CreateReFungibleData { }715export interface CreateNftData { }716717export type CreateItemData = {718 NFT: CreateNftData;719} | {720 Fungible: CreateFungibleData;721} | {722 ReFungible: CreateReFungibleData;723};724725export async function burnItemExpectSuccess(sender: IKeyringPair, collectionId: number, tokenId: number, value = 1) {726 await usingApi(async (api) => {727 const balanceBefore = await getBalance(api, collectionId, normalizeAccountId(sender), tokenId);728 // if burning token by admin - use adminButnItemExpectSuccess729 expect(balanceBefore >= BigInt(value)).to.be.true;730731 const tx = api.tx.unique.burnItem(collectionId, tokenId, value);732 const events = await submitTransactionAsync(sender, tx);733 const result = getGenericResult(events);734 expect(result.success).to.be.true;735736 const balanceAfter = await getBalance(api, collectionId, normalizeAccountId(sender), tokenId);737 expect(balanceAfter + BigInt(value)).to.be.equal(balanceBefore);738 });739}740741export async function742approveExpectSuccess(743 collectionId: number,744 tokenId: number, owner: IKeyringPair, approved: CrossAccountId | string, amount: number | bigint = 1,745) {746 await usingApi(async (api: ApiPromise) => {747 const approveUniqueTx = api.tx.unique.approve(normalizeAccountId(approved), collectionId, tokenId, amount);748 const events = await submitTransactionAsync(owner, approveUniqueTx);749 const result = getGenericResult(events);750 expect(result.success).to.be.true;751752 expect(await getAllowance(api, collectionId, owner.address, approved, tokenId)).to.be.equal(BigInt(amount));753 });754}755756export async function adminApproveFromExpectSuccess(757 collectionId: number,758 tokenId: number, admin: IKeyringPair, owner: CrossAccountId | string, approved: CrossAccountId | string, amount: number | bigint = 1,759) {760 await usingApi(async (api: ApiPromise) => {761 const approveUniqueTx = api.tx.unique.approve(normalizeAccountId(approved), collectionId, tokenId, amount);762 const events = await submitTransactionAsync(admin, approveUniqueTx);763 const result = getGenericResult(events);764 expect(result.success).to.be.true;765766 expect(await getAllowance(api, collectionId, owner, approved, tokenId)).to.be.equal(BigInt(amount));767 });768}769770export async function771transferFromExpectSuccess(772 collectionId: number,773 tokenId: number,774 accountApproved: IKeyringPair,775 accountFrom: IKeyringPair | CrossAccountId,776 accountTo: IKeyringPair | CrossAccountId,777 value: number | bigint = 1,778 type = 'NFT',779) {780 await usingApi(async (api: ApiPromise) => {781 const from = normalizeAccountId(accountFrom);782 const to = normalizeAccountId(accountTo);783 let balanceBefore = 0n;784 if (type === 'Fungible') {785 balanceBefore = await getBalance(api, collectionId, to, tokenId);786 }787 const transferFromTx = api.tx.unique.transferFrom(normalizeAccountId(accountFrom), to, collectionId, tokenId, value);788 const events = await submitTransactionAsync(accountApproved, transferFromTx);789 const result = getCreateItemResult(events);790 // tslint:disable-next-line:no-unused-expression791 expect(result.success).to.be.true;792 if (type === 'NFT') {793 expect(await getTokenOwner(api, collectionId, tokenId)).to.be.deep.equal(to);794 }795 if (type === 'Fungible') {796 const balanceAfter = await getBalance(api, collectionId, to, tokenId);797 if (JSON.stringify(to) !== JSON.stringify(from)) {798 expect(balanceAfter - balanceBefore).to.be.equal(BigInt(value));799 } else {800 expect(balanceAfter).to.be.equal(balanceBefore);801 }802 }803 if (type === 'ReFungible') {804 expect(await getBalance(api, collectionId, to, tokenId)).to.be.equal(BigInt(value));805 }806 });807}808809export async function810transferFromExpectFail(811 collectionId: number,812 tokenId: number,813 accountApproved: IKeyringPair,814 accountFrom: IKeyringPair,815 accountTo: IKeyringPair,816 value: number | bigint = 1,817) {818 await usingApi(async (api: ApiPromise) => {819 const transferFromTx = api.tx.unique.transferFrom(normalizeAccountId(accountFrom.address), normalizeAccountId(accountTo.address), collectionId, tokenId, value);820 const events = await expect(submitTransactionExpectFailAsync(accountApproved, transferFromTx)).to.be.rejected;821 const result = getCreateCollectionResult(events);822 // tslint:disable-next-line:no-unused-expression823 expect(result.success).to.be.false;824 });825}826827/* eslint no-async-promise-executor: "off" */828async function getBlockNumber(api: ApiPromise): Promise<number> {829 return new Promise<number>(async (resolve) => {830 const unsubscribe = await api.rpc.chain.subscribeNewHeads((head) => {831 unsubscribe();832 resolve(head.number.toNumber());833 });834 });835}836837export async function addCollectionAdminExpectSuccess(sender: IKeyringPair, collectionId: number, address: string | CrossAccountId) {838 await usingApi(async (api) => {839 const changeAdminTx = api.tx.unique.addCollectionAdmin(collectionId, normalizeAccountId(address));840 const events = await submitTransactionAsync(sender, changeAdminTx);841 const result = getCreateCollectionResult(events);842 expect(result.success).to.be.true;843 });844}845846export async function847getFreeBalance(account: IKeyringPair): Promise<bigint> {848 let balance = 0n;849 await usingApi(async (api) => {850 balance = BigInt((await api.query.system.account(account.address)).data.free.toString());851 });852853 return balance;854}855856export async function transferBalanceTo(api: ApiPromise, source: IKeyringPair, target: string, amount = 1000n * UNIQUE) {857 const tx = api.tx.balances.transfer(target, amount);858 const events = await submitTransactionAsync(source, tx);859 const result = getGenericResult(events);860 expect(result.success).to.be.true;861}862863export async function864scheduleTransferExpectSuccess(865 collectionId: number,866 tokenId: number,867 sender: IKeyringPair,868 recipient: IKeyringPair,869 value: number | bigint = 1,870 blockSchedule: number,871) {872 await usingApi(async (api: ApiPromise) => {873 const blockNumber: number | undefined = await getBlockNumber(api);874 const expectedBlockNumber = blockNumber + blockSchedule;875876 expect(blockNumber).to.be.greaterThan(0);877 const transferTx = api.tx.unique.transfer(normalizeAccountId(recipient.address), collectionId, tokenId, value);878 const scheduleTx = api.tx.scheduler.schedule(expectedBlockNumber, null, 0, transferTx as any);879880 await submitTransactionAsync(sender, scheduleTx);881882 const recipientBalanceBefore = (await api.query.system.account(recipient.address)).data.free.toBigInt();883884 expect(await getTokenOwner(api, collectionId, tokenId)).to.be.deep.equal(normalizeAccountId(sender.address));885886 // sleep for 4 blocks887 await waitNewBlocks(blockSchedule + 1);888889 const recipientBalanceAfter = (await api.query.system.account(recipient.address)).data.free.toBigInt();890891 expect(await getTokenOwner(api, collectionId, tokenId)).to.be.deep.equal(normalizeAccountId(recipient.address));892 expect(recipientBalanceAfter).to.be.equal(recipientBalanceBefore);893 });894}895896897export async function898transferExpectSuccess(899 collectionId: number,900 tokenId: number,901 sender: IKeyringPair,902 recipient: IKeyringPair | CrossAccountId,903 value: number | bigint = 1,904 type = 'NFT',905) {906 await usingApi(async (api: ApiPromise) => {907 const from = normalizeAccountId(sender);908 const to = normalizeAccountId(recipient);909910 let balanceBefore = 0n;911 if (type === 'Fungible') {912 balanceBefore = await getBalance(api, collectionId, to, tokenId);913 }914 const transferTx = api.tx.unique.transfer(to, collectionId, tokenId, value);915 const events = await submitTransactionAsync(sender, transferTx);916 const result = getTransferResult(events);917 // tslint:disable-next-line:no-unused-expression918 expect(result.success).to.be.true;919 expect(result.collectionId).to.be.equal(collectionId);920 expect(result.itemId).to.be.equal(tokenId);921 expect(result.sender).to.be.deep.equal(normalizeAccountId(sender.address));922 expect(result.recipient).to.be.deep.equal(to);923 expect(result.value).to.be.equal(BigInt(value));924 if (type === 'NFT') {925 expect(await getTokenOwner(api, collectionId, tokenId)).to.be.deep.equal(to);926 }927 if (type === 'Fungible') {928 const balanceAfter = await getBalance(api, collectionId, to, tokenId);929 if (JSON.stringify(to) !== JSON.stringify(from)) {930 expect(balanceAfter - balanceBefore).to.be.equal(BigInt(value));931 } else {932 expect(balanceAfter).to.be.equal(balanceBefore);933 }934 }935 if (type === 'ReFungible') {936 expect(await getBalance(api, collectionId, to, tokenId) >= value).to.be.true;937 }938 });939}940941export async function942transferExpectFailure(943 collectionId: number,944 tokenId: number,945 sender: IKeyringPair,946 recipient: IKeyringPair,947 value: number | bigint = 1,948) {949 await usingApi(async (api: ApiPromise) => {950 const transferTx = api.tx.unique.transfer(normalizeAccountId(recipient.address), collectionId, tokenId, value);951 const events = await expect(submitTransactionExpectFailAsync(sender, transferTx)).to.be.rejected;952 const result = getGenericResult(events);953 // if (events && Array.isArray(events)) {954 // const result = getCreateCollectionResult(events);955 // tslint:disable-next-line:no-unused-expression956 expect(result.success).to.be.false;957 //}958 });959}960961export async function962approveExpectFail(963 collectionId: number,964 tokenId: number, owner: IKeyringPair, approved: IKeyringPair, amount: number | bigint = 1,965) {966 await usingApi(async (api: ApiPromise) => {967 const approveUniqueTx = api.tx.unique.approve(normalizeAccountId(approved.address), collectionId, tokenId, amount);968 const events = await expect(submitTransactionExpectFailAsync(owner, approveUniqueTx)).to.be.rejected;969 const result = getCreateCollectionResult(events);970 // tslint:disable-next-line:no-unused-expression971 expect(result.success).to.be.false;972 });973}974975export async function getBalance(976 api: ApiPromise,977 collectionId: number,978 owner: string | CrossAccountId,979 token: number,980): Promise<bigint> {981 return (await api.rpc.unique.balance(collectionId, normalizeAccountId(owner), token)).toBigInt();982}983export async function getTokenOwner(984 api: ApiPromise,985 collectionId: number,986 token: number,987): Promise<CrossAccountId> {988 return normalizeAccountId((await api.rpc.unique.tokenOwner(collectionId, token)).toJSON() as any);989}990export async function isTokenExists(991 api: ApiPromise,992 collectionId: number,993 token: number,994): Promise<boolean> {995 return (await api.rpc.unique.tokenExists(collectionId, token)).toJSON();996}997export async function getLastTokenId(998 api: ApiPromise,999 collectionId: number,1000): Promise<number> {1001 return (await api.rpc.unique.lastTokenId(collectionId)).toJSON();1002}1003export async function getAdminList(1004 api: ApiPromise,1005 collectionId: number,1006): Promise<string[]> {1007 return (await api.rpc.unique.adminlist(collectionId)).toHuman() as any;1008}1009export async function getVariableMetadata(1010 api: ApiPromise,1011 collectionId: number,1012 tokenId: number,1013): Promise<number[]> {1014 return [...(await api.rpc.unique.variableMetadata(collectionId, tokenId))];1015}1016export async function getConstMetadata(1017 api: ApiPromise,1018 collectionId: number,1019 tokenId: number,1020): Promise<number[]> {1021 return [...(await api.rpc.unique.constMetadata(collectionId, tokenId))];1022}10231024export async function createFungibleItemExpectSuccess(1025 sender: IKeyringPair,1026 collectionId: number,1027 data: CreateFungibleData,1028 owner: CrossAccountId | string = sender.address,1029) {1030 return await usingApi(async (api) => {1031 const tx = api.tx.unique.createItem(collectionId, normalizeAccountId(owner), {Fungible: data});10321033 const events = await submitTransactionAsync(sender, tx);1034 const result = getCreateItemResult(events);10351036 expect(result.success).to.be.true;1037 return result.itemId;1038 });1039}10401041export async function createItemExpectSuccess(sender: IKeyringPair, collectionId: number, createMode: string, owner: CrossAccountId | string = sender.address) {1042 let newItemId = 0;1043 await usingApi(async (api) => {1044 const to = normalizeAccountId(owner);1045 const itemCountBefore = await getLastTokenId(api, collectionId);1046 const itemBalanceBefore = await getBalance(api, collectionId, to, newItemId);10471048 let tx;1049 if (createMode === 'Fungible') {1050 const createData = {fungible: {value: 10}};1051 tx = api.tx.unique.createItem(collectionId, to, createData as any);1052 } else if (createMode === 'ReFungible') {1053 const createData = {refungible: {const_data: [], variable_data: [], pieces: 100}};1054 tx = api.tx.unique.createItem(collectionId, to, createData as any);1055 } else {1056 const createData = {nft: {const_data: [], variable_data: []}};1057 tx = api.tx.unique.createItem(collectionId, to, createData as any);1058 }10591060 const events = await submitTransactionAsync(sender, tx);1061 const result = getCreateItemResult(events);10621063 const itemCountAfter = await getLastTokenId(api, collectionId);1064 const itemBalanceAfter = await getBalance(api, collectionId, to, newItemId);10651066 // What to expect1067 // tslint:disable-next-line:no-unused-expression1068 expect(result.success).to.be.true;1069 if (createMode === 'Fungible') {1070 expect(itemBalanceAfter - itemBalanceBefore).to.be.equal(10n);1071 } else {1072 expect(itemCountAfter).to.be.equal(itemCountBefore + 1);1073 }1074 expect(collectionId).to.be.equal(result.collectionId);1075 expect(itemCountAfter.toString()).to.be.equal(result.itemId.toString());1076 expect(to).to.be.deep.equal(result.recipient);1077 newItemId = result.itemId;1078 });1079 return newItemId;1080}10811082export async function createItemExpectFailure(sender: IKeyringPair, collectionId: number, createMode: string, owner: string = sender.address) {1083 await usingApi(async (api) => {1084 const tx = api.tx.unique.createItem(collectionId, normalizeAccountId(owner), createMode);10851086 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;1087 const result = getCreateItemResult(events);10881089 expect(result.success).to.be.false;1090 });1091}10921093export async function setPublicAccessModeExpectSuccess(1094 sender: IKeyringPair, collectionId: number,1095 accessMode: 'Normal' | 'AllowList',1096) {1097 await usingApi(async (api) => {10981099 // Run the transaction1100 const tx = api.tx.unique.setPublicAccessMode(collectionId, accessMode);1101 const events = await submitTransactionAsync(sender, tx);1102 const result = getGenericResult(events);11031104 // Get the collection1105 const collection = await queryCollectionExpectSuccess(api, collectionId);11061107 // What to expect1108 // tslint:disable-next-line:no-unused-expression1109 expect(result.success).to.be.true;1110 expect(collection.access.toHuman()).to.be.equal(accessMode);1111 });1112}11131114export async function setPublicAccessModeExpectFail(1115 sender: IKeyringPair, collectionId: number,1116 accessMode: 'Normal' | 'AllowList',1117) {1118 await usingApi(async (api) => {11191120 // Run the transaction1121 const tx = api.tx.unique.setPublicAccessMode(collectionId, accessMode);1122 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;1123 const result = getGenericResult(events);11241125 // What to expect1126 // tslint:disable-next-line:no-unused-expression1127 expect(result.success).to.be.false;1128 });1129}11301131export async function enableAllowListExpectSuccess(sender: IKeyringPair, collectionId: number) {1132 await setPublicAccessModeExpectSuccess(sender, collectionId, 'AllowList');1133}11341135export async function enableAllowListExpectFail(sender: IKeyringPair, collectionId: number) {1136 await setPublicAccessModeExpectFail(sender, collectionId, 'AllowList');1137}11381139export async function disableAllowListExpectSuccess(sender: IKeyringPair, collectionId: number) {1140 await setPublicAccessModeExpectSuccess(sender, collectionId, 'Normal');1141}11421143export async function setMintPermissionExpectSuccess(sender: IKeyringPair, collectionId: number, enabled: boolean) {1144 await usingApi(async (api) => {11451146 // Run the transaction1147 const tx = api.tx.unique.setMintPermission(collectionId, enabled);1148 const events = await submitTransactionAsync(sender, tx);1149 const result = getGenericResult(events);1150 expect(result.success).to.be.true;11511152 // Get the collection1153 const collection = await queryCollectionExpectSuccess(api, collectionId);11541155 expect(collection.mintMode.toHuman()).to.be.equal(enabled);1156 });1157}11581159export async function enablePublicMintingExpectSuccess(sender: IKeyringPair, collectionId: number) {1160 await setMintPermissionExpectSuccess(sender, collectionId, true);1161}11621163export async function setMintPermissionExpectFailure(sender: IKeyringPair, collectionId: number, enabled: boolean) {1164 await usingApi(async (api) => {1165 // Run the transaction1166 const tx = api.tx.unique.setMintPermission(collectionId, enabled);1167 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;1168 const result = getCreateCollectionResult(events);1169 // tslint:disable-next-line:no-unused-expression1170 expect(result.success).to.be.false;1171 });1172}11731174export async function setChainLimitsExpectFailure(sender: IKeyringPair, limits: IChainLimits) {1175 await usingApi(async (api) => {1176 // Run the transaction1177 const tx = api.tx.unique.setChainLimits(limits);1178 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;1179 const result = getCreateCollectionResult(events);1180 // tslint:disable-next-line:no-unused-expression1181 expect(result.success).to.be.false;1182 });1183}11841185export async function isAllowlisted(api: ApiPromise, collectionId: number, address: string | CrossAccountId) {1186 return (await api.rpc.unique.allowed(collectionId, normalizeAccountId(address))).toJSON();1187}11881189export async function addToAllowListExpectSuccess(sender: IKeyringPair, collectionId: number, address: string | AccountId | CrossAccountId) {1190 await usingApi(async (api) => {1191 expect(await isAllowlisted(api, collectionId, normalizeAccountId(address))).to.be.false;11921193 // Run the transaction1194 const tx = api.tx.unique.addToAllowList(collectionId, normalizeAccountId(address));1195 const events = await submitTransactionAsync(sender, tx);1196 const result = getGenericResult(events);1197 expect(result.success).to.be.true;11981199 expect(await isAllowlisted(api, collectionId, normalizeAccountId(address))).to.be.true;1200 });1201}12021203export async function addToAllowListAgainExpectSuccess(sender: IKeyringPair, collectionId: number, address: string | AccountId) {1204 await usingApi(async (api) => {12051206 expect(await isAllowlisted(api, collectionId, normalizeAccountId(address))).to.be.true;12071208 // Run the transaction1209 const tx = api.tx.unique.addToAllowList(collectionId, normalizeAccountId(address));1210 const events = await submitTransactionAsync(sender, tx);1211 const result = getGenericResult(events);1212 expect(result.success).to.be.true;12131214 expect(await isAllowlisted(api, collectionId, normalizeAccountId(address))).to.be.true;1215 });1216}12171218export async function addToAllowListExpectFail(sender: IKeyringPair, collectionId: number, address: string | AccountId) {1219 await usingApi(async (api) => {12201221 // Run the transaction1222 const tx = api.tx.unique.addToAllowList(collectionId, normalizeAccountId(address));1223 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;1224 const result = getGenericResult(events);12251226 // What to expect1227 // tslint:disable-next-line:no-unused-expression1228 expect(result.success).to.be.false;1229 });1230}12311232export async function removeFromAllowListExpectSuccess(sender: IKeyringPair, collectionId: number, address: CrossAccountId) {1233 await usingApi(async (api) => {1234 // Run the transaction1235 const tx = api.tx.unique.removeFromAllowList(collectionId, normalizeAccountId(address));1236 const events = await submitTransactionAsync(sender, tx);1237 const result = getGenericResult(events);12381239 // What to expect1240 // tslint:disable-next-line:no-unused-expression1241 expect(result.success).to.be.true;1242 });1243}12441245export async function removeFromAllowListExpectFailure(sender: IKeyringPair, collectionId: number, address: CrossAccountId) {1246 await usingApi(async (api) => {1247 // Run the transaction1248 const tx = api.tx.unique.removeFromAllowList(collectionId, normalizeAccountId(address));1249 const events = await expect(submitTransactionExpectFailAsync(sender, tx)).to.be.rejected;1250 const result = getGenericResult(events);12511252 // What to expect1253 // tslint:disable-next-line:no-unused-expression1254 expect(result.success).to.be.false;1255 });1256}12571258export const getDetailedCollectionInfo = async (api: ApiPromise, collectionId: number)1259 : Promise<UpDataStructsCollection | null> => {1260 return (await api.rpc.unique.collectionById(collectionId)).unwrapOr(null);1261};12621263export const getCreatedCollectionCount = async (api: ApiPromise): Promise<number> => {1264 // set global object - collectionsCount1265 return (await api.rpc.unique.collectionStats()).created.toNumber();1266};12671268export async function queryCollectionExpectSuccess(api: ApiPromise, collectionId: number): Promise<UpDataStructsCollection> {1269 return (await api.rpc.unique.collectionById(collectionId)).unwrap();1270}12711272export async function waitNewBlocks(blocksCount = 1): Promise<void> {1273 await usingApi(async (api) => {1274 const promise = new Promise<void>(async (resolve) => {1275 const unsubscribe = await api.rpc.chain.subscribeNewHeads(() => {1276 if (blocksCount > 0) {1277 blocksCount--;1278 } else {1279 unsubscribe();1280 resolve();1281 }1282 });1283 });1284 return promise;1285 });1286}