`d83ab4fde4e8`

source

tests/src/util/playgrounds/unique.dev.ts8.0 KiBsource history
1// Copyright 2019-2022 Unique Network (Gibraltar) Ltd.2// SPDX-License-Identifier: Apache-2.034import {mnemonicGenerate} from '@polkadot/util-crypto';5import {UniqueHelper} from './unique';6import {ApiPromise, WsProvider} from '@polkadot/api';7import * as defs from '../../interfaces/definitions';8import {IKeyringPair} from '@polkadot/types/types';91011export class SilentLogger {12  log(_msg: any, _level: any): void { }13  level = {14    ERROR: 'ERROR' as const,15    WARNING: 'WARNING' as const,16    INFO: 'INFO' as const,17  };18}192021export class SilentConsole {22  // TODO: Remove, this is temporary: Filter unneeded API output23  // (Jaco promised it will be removed in the next version)24  consoleErr: any;25  consoleLog: any;26  consoleWarn: any;2728  constructor() {29    this.consoleErr = console.error;30    this.consoleLog = console.log;31    this.consoleWarn = console.warn;32  }3334  enable() {  35    const outFn = (printer: any) => (...args: any[]) => {36      for (const arg of args) {37        if (typeof arg !== 'string')38          continue;39        if (arg.includes('1000:: Normal connection closure') || arg.includes('Not decorating unknown runtime apis: UniqueApi/2, RmrkApi/1') || arg.includes('RPC methods not decorated:') || arg === 'Normal connection closure')40          return;41      }42      printer(...args);43    };44  45    console.error = outFn(this.consoleErr.bind(console));46    console.log = outFn(this.consoleLog.bind(console));47    console.warn = outFn(this.consoleWarn.bind(console));48  }4950  disable() {51    console.error = this.consoleErr;52    console.log = this.consoleLog;53    console.warn = this.consoleWarn;54  }55}565758export class DevUniqueHelper extends UniqueHelper {59  /**60   * Arrange methods for tests61   */62  arrange: ArrangeGroup;6364  constructor(logger: { log: (msg: any, level: any) => void, level: any }) {65    super(logger);66    this.arrange = new ArrangeGroup(this);67  }6869  async connect(wsEndpoint: string, _listeners?: any): Promise<void> {70    const wsProvider = new WsProvider(wsEndpoint);71    this.api = new ApiPromise({72      provider: wsProvider,73      signedExtensions: {74        ContractHelpers: {75          extrinsic: {},76          payload: {},77        },78        FakeTransactionFinalizer: {79          extrinsic: {},80          payload: {},81        },82      },83      rpc: {84        unique: defs.unique.rpc,85        appPromotion: defs.appPromotion.rpc,86        rmrk: defs.rmrk.rpc,87        eth: {88          feeHistory: {89            description: 'Dummy',90            params: [],91            type: 'u8',92          },93          maxPriorityFeePerGas: {94            description: 'Dummy',95            params: [],96            type: 'u8',97          },98        },99      },100    });101    await this.api.isReadyOrError;102    this.network = await UniqueHelper.detectNetwork(this.api);103  }104}105106class ArrangeGroup {107  helper: UniqueHelper;108109  constructor(helper: UniqueHelper) {110    this.helper = helper;111  }112113  /**114   * Generates accounts with the specified UNQ token balance 115   * @param balances balances for generated accounts. Each balance will be multiplied by the token nominal.116   * @param donor donor account for balances117   * @returns array of newly created accounts118   * @example const [acc1, acc2, acc3] = await createAccounts([0n, 10n, 20n], donor); 119   */120  createAccounts = async (balances: bigint[], donor: IKeyringPair): Promise<IKeyringPair[]> => {121    let nonce = await this.helper.chain.getNonce(donor.address);122    const tokenNominal = this.helper.balance.getOneTokenNominal();123    const transactions = [];124    const accounts: IKeyringPair[] = [];125    for (const balance of balances) {126      const recepient = this.helper.util.fromSeed(mnemonicGenerate());127      accounts.push(recepient);128      if (balance !== 0n) {129        const tx = this.helper.constructApiCall('api.tx.balances.transfer', [{Id: recepient.address}, balance * tokenNominal]);130        transactions.push(this.helper.signTransaction(donor, tx, 'account generation', {nonce}));131        nonce++;132      }133    }134135    await Promise.all(transactions).catch(_e => {});136    137    //#region TODO remove this region, when nonce problem will be solved138    const checkBalances = async () => {139      let isSuccess = true;140      for (let i = 0; i < balances.length; i++) {141        const balance = await this.helper.balance.getSubstrate(accounts[i].address);142        if (balance !== balances[i] * tokenNominal) {143          isSuccess = false;144          break;145        }146      }147      return isSuccess;148    };149150    let accountsCreated = false;151    // checkBalances retry up to 5 blocks152    for (let index = 0; index < 5; index++) {153      accountsCreated = await checkBalances();154      if(accountsCreated) break;155      await this.waitNewBlocks(1);156    }157158    if (!accountsCreated) throw Error('Accounts generation failed');159    //#endregion160161    return accounts;162  };163164  // TODO combine this method and createAccounts into one165  createCrowd = async (accountsToCreate: number, withBalance: bigint, donor: IKeyringPair): Promise<IKeyringPair[]> => {166    let transactions: any = [];167    const accounts: IKeyringPair[] = [];168169    const createAsManyAsCan = async () => {170      let nonce = await this.helper.chain.getNonce(donor.address);171      const tokenNominal = this.helper.balance.getOneTokenNominal();172      for (let i = 0; i < accountsToCreate; i++) {173        if (i === 500) { // if there are too many accounts to create174          await Promise.allSettled(transactions); // wait while first 500 (should be 100 for devnode) tx will be settled 175          transactions = []; //176          nonce = await this.helper.chain.getNonce(donor.address); // update nonce 177        }178        const recepient = this.helper.util.fromSeed(mnemonicGenerate());179        accounts.push(recepient);180        if (withBalance !== 0n) {181          const tx = this.helper.constructApiCall('api.tx.balances.transfer', [{Id: recepient.address}, withBalance * tokenNominal]);182          transactions.push(this.helper.signTransaction(donor, tx, 'account generation', {nonce}));183          nonce++;184        }185      }186      187      const fullfilledAccounts = [];188      await Promise.allSettled(transactions);189      for (const account of accounts) {190        const accountBalance = await this.helper.balance.getSubstrate(account.address);191        if (accountBalance === withBalance * tokenNominal) {192          fullfilledAccounts.push(account);193        }194      }195      return fullfilledAccounts;196    };197198    199    const crowd: IKeyringPair[] = [];200    // do up to 5 retries201    for (let index = 0; index < 5 && accountsToCreate !== 0; index++) {202      const asManyAsCan = await createAsManyAsCan();203      crowd.push(...asManyAsCan);204      accountsToCreate -= asManyAsCan.length;205    }206207    if (accountsToCreate !== 0) throw Error(`Crowd generation failed: ${accountsToCreate} accounts left`);208209    return crowd;210  };211212  isDevNode = async () => {213    const block1 = await this.helper.api?.rpc.chain.getBlock(await this.helper.api?.rpc.chain.getBlockHash(1));214    const block2 = await this.helper.api?.rpc.chain.getBlock(await this.helper.api?.rpc.chain.getBlockHash(2));215    const findCreationDate = async (block: any) => {216      const humanBlock = block.toHuman();217      let date;218      humanBlock.block.extrinsics.forEach((ext: any) => {219        if(ext.method.section === 'timestamp') {220          date = Number(ext.method.args.now.replaceAll(',', ''));221        }222      });223      return date;224    };225    const block1date = await findCreationDate(block1);226    const block2date = await findCreationDate(block2);227    if(block2date! - block1date! < 9000) return true;228  };229230  /**231   * Wait for specified bnumber of blocks232   * @param blocksCount number of blocks to wait233   * @returns 234   */235  async waitNewBlocks(blocksCount = 1): Promise<void> {236    // eslint-disable-next-line no-async-promise-executor237    const promise = new Promise<void>(async (resolve) => {238      const unsubscribe = await this.helper.api!.rpc.chain.subscribeNewHeads(() => {239        if (blocksCount > 0) {240          blocksCount--;241        } else {242          unsubscribe();243          resolve();244        }245      });246    });247    return promise;248  }249}