@gnosis.pm/pm-contracts/test/javascript/utils.js

Collection of smart contracts for the Gnosis prediction market platform

const _ = require('lodash')

const PRECISION = 80
const Decimal = require('decimal.js').clone({ precision: PRECISION })

const ONE = Decimal(2).pow(64)

function isClose(a, b, relTol=1e-9, absTol=1e-18) {
    a = Decimal(a.toString()).add(1) // Adding 1 is just for the case when a and b are both near to 0
    b = Decimal(b.toString()).add(1)
    return Decimal(a).sub(b).abs().lte(
        Decimal.max(
            Decimal.max(
                a.abs(),
                b.abs()
            ).mul(relTol),
            absTol))
}

// random int in [a, b)
function randrange(a, b) {
    return Decimal.random(PRECISION).mul(Decimal(b.valueOf()).sub(a)).add(a).floor()
}

function randnums(a, b, n) {
    return _.range(n).map(() => randrange(a, b))
}

async function getParamFromTxEvent(transaction, paramName, contractFactory, eventName) {
    assert.isObject(transaction)
    let logs = transaction.logs
    if(eventName != null) {
        logs = logs.filter((l) => l.event === eventName)
    }
    assert.equal(logs.length, 1, `expected one log but got ${logs.length} logs`)
    let param = logs[0].args[paramName]
    if(contractFactory != null) {
        let contract = await contractFactory.at(param);        
        assert.isObject(contract, `getting ${paramName} failed for ${param}`)
        return contract
    } else {
        return param
    }
}

async function assertRejects(q, msg) {
    let res, catchFlag = false
    try {
        res = await q
    } catch(e) {
        catchFlag = true
    } finally {
        if(!catchFlag)
            assert.fail(res, null, msg)
    }
}

function getBlock(b) {
    return new Promise((resolve, reject) => {
        web3.eth.getBlock(b, (err, block) => {
            if(err) return reject(err)
            resolve(block)
        })
    })
}

function lmsrMarginalPrice(funding, netOutcomeTokensSold, outcomeIndex) {
    const b = Decimal(funding.toString()).div(netOutcomeTokensSold.length).ln()

    return Decimal(netOutcomeTokensSold[outcomeIndex].toString()).div(b).exp().div(
        netOutcomeTokensSold.reduce(
            (acc, tokensSold) => acc.add(Decimal(tokensSold.toString()).div(b).exp()),
            Decimal(0)
        )
    ).valueOf()
}

Object.assign(exports, {
    Decimal,
    ONE,
    isClose,
    randrange,
    randnums,
    getParamFromTxEvent,
    assertRejects,
    getBlock,
    lmsrMarginalPrice,
})