packages/contracts/src/utils/assertions.ts


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32

import * as chai from 'chai';
import * as _ from 'lodash';

import { constants } from './constants';

const expect = chai.expect;

// throws if the given promise does not reject with one of two expected error
// messages.
export function expectRevertOrAlwaysFailingTransaction<T>(p: Promise<T>): PromiseLike<void> {
    return expect(p)
        .to.be.rejected()
        .then(e => {
            expect(e).to.satisfy(
                (err: Error) =>
                    _.includes(err.message, constants.REVERT) ||
                    _.includes(err.message, constants.ALWAYS_FAILING_TRANSACTION),
            );
        });
}

export function expectInsufficientFunds<T>(p: Promise<T>): PromiseLike<void> {
    return expect(p)
        .to.be.rejected()
        .then(e => {
            expect(e).to.satisfy(
                (err: Error) =>
                    _.includes(err.message, 'insufficient funds') ||
                    _.includes(err.message, "sender doesn't have enough funds"),
            );
        });
}