123456import chai from "chai";7import chaiAsPromised from 'chai-as-promised';8import { submitTransactionAsync } from "../substrate/substrate-api";9import fs from "fs";10import { Abi, BlueprintPromise as Blueprint, CodePromise, ContractPromise as Contract } from "@polkadot/api-contract";11import { IKeyringPair } from "@polkadot/types/types";12import { ApiPromise, Keyring } from "@polkadot/api";1314chai.use(chaiAsPromised);15const expect = chai.expect;16import { BigNumber } from 'bignumber.js';17import { findUnusedAddress, getGenericResult } from '../util/helpers';1819const value = 0;20const gasLimit = '200000000000';21const endowment = '100000000000000000';2223function deployBlueprint(alice: IKeyringPair, code: CodePromise): Promise<Blueprint> {24 return new Promise<Blueprint>(async (resolve, reject) => {25 const unsub = await code26 .createBlueprint()27 .signAndSend(alice, (result) => {28 if (result.status.isInBlock || result.status.isFinalized) {29 30 resolve(result.blueprint);31 unsub();32 }33 })34 });35}3637function deployContract(alice: IKeyringPair, blueprint: Blueprint) : Promise<any> {38 return new Promise<any>(async (resolve, reject) => {39 const initValue = true;4041 const unsub = await blueprint.tx42 .new(endowment, gasLimit, initValue)43 .signAndSend(alice, (result) => {44 if (result.status.isInBlock || result.status.isFinalized) {45 unsub();46 resolve(result);47 }48 }); 49 });50}5152async function prepareDeployer(api: ApiPromise) {53 54 const deployer = await findUnusedAddress(api);5556 57 const keyring = new Keyring({ type: 'sr25519' });58 const alice = keyring.addFromUri(`//Alice`);59 let amount = new BigNumber(endowment);60 amount = amount.plus(100e15);61 const tx = api.tx.balances.transfer(deployer.address, amount.toFixed());62 await submitTransactionAsync(alice, tx);6364 return deployer;65}6667export async function deployFlipper(api: ApiPromise): Promise<[Contract, IKeyringPair]> {68 const metadata = JSON.parse(fs.readFileSync('./src/flipper/metadata.json').toString('utf-8'));69 const abi = new Abi(metadata);7071 const deployer = await prepareDeployer(api);7273 const wasm = fs.readFileSync('./src/flipper/flipper.wasm');7475 const code = new CodePromise(api, abi, wasm);7677 const blueprint = await deployBlueprint(deployer, code);78 const contract = (await deployContract(deployer, blueprint))['contract'] as Contract;7980 const initialGetResponse = await getFlipValue(contract, deployer);81 expect(initialGetResponse).to.be.true;8283 return [contract, deployer];84}8586export async function getFlipValue(contract: Contract, deployer: IKeyringPair) {87 const result = await contract.query.get(deployer.address, value, gasLimit);8889 if(!result.result.isOk) {90 throw `Failed to get flipper value`;91 }92 return (result.result.asOk.data[0] == 0x00) ? false : true;93}9495export async function toggleFlipValueExpectSuccess(sender: IKeyringPair, contract: Contract) {96 const tx = contract.tx.flip(value, gasLimit);97 const events = await submitTransactionAsync(sender, tx);98 const result = getGenericResult(events);99100 expect(result.success).to.be.true;101}102103function instantiateTransferContract(alice: IKeyringPair, blueprint: Blueprint) : Promise<any> {104 return new Promise<any>(async (resolve, reject) => {105 const unsub = await blueprint.tx106 .default(endowment, gasLimit)107 .signAndSend(alice, (result) => {108 if (result.status.isInBlock || result.status.isFinalized) {109 unsub();110 resolve(result);111 }112 }); 113 });114}115116export async function deployTransferContract(api: ApiPromise): Promise<[Contract, IKeyringPair]> {117 const metadata = JSON.parse(fs.readFileSync('./src/transfer_contract/metadata.json').toString('utf-8'));118 const abi = new Abi(metadata);119120 const deployer = await prepareDeployer(api);121122 const wasm = fs.readFileSync('./src/transfer_contract/nft_transfer.wasm');123124 const code = new CodePromise(api, abi, wasm);125126 const blueprint = await deployBlueprint(deployer, code);127 const contract = (await instantiateTransferContract(deployer, blueprint))['contract'] as Contract;128129 return [contract, deployer];130}