import * as chai from 'chai'; import { AnyRevertError, RevertError, StringRevertError } from '../src/revert_error'; import { chaiSetup } from './utils/chai_setup'; chaiSetup.configure(); const expect = chai.expect; // tslint:disable: max-classes-per-file class DescendantRevertError extends StringRevertError { public constructor(message: string) { super(message); } } class CustomRevertError extends RevertError { public constructor(message?: string) { super('CustomRevertError(string message)', { message }); } } RevertError.registerType(CustomRevertError); describe('RevertError', () => { describe('equality', () => { const message = 'foo'; it('should equate two identical RevertErrors', () => { const revert1 = new StringRevertError(message); const revert2 = new StringRevertError(message); expect(revert1.equals(revert2)).to.be.true(); }); it('should equate two RevertErrors with missing fields', () => { const revert1 = new StringRevertError(message); const revert2 = new StringRevertError(); expect(revert1.equals(revert2)).to.be.true(); }); it('should equate AnyRevertError with a real RevertError', () => { const revert1 = new StringRevertError(message); const revert2 = new AnyRevertError(); expect(revert1.equals(revert2)).to.be.true(); }); it('should not equate a the same RevertError type with different values', () => { const revert1 = new StringRevertError(message); const revert2 = new StringRevertError(`${message}1`); expect(revert1.equals(revert2)).to.be.false(); }); it('should not equate different RevertError types', () => { const revert1 = new StringRevertError(message); const revert2 = new DescendantRevertError(message); expect(revert1.equals(revert2)).to.be.false(); }); }); describe('registering', () => { it('should throw when registering an already registered signature', () => { class CustomRevertError2 extends RevertError { public constructor() { super(new CustomRevertError().signature, {}); } } expect(() => RevertError.registerType(CustomRevertError2)).to.throw(); }); }); describe('decoding', () => { // tslint:disable: prefer-template custom-no-magic-numbers const message = 'foobar'; const encoded = '0x08c379a0' + '0000000000000000000000000000000000000000000000000000000000000020' + '0000000000000000000000000000000000000000000000000000000000000006' + Buffer.from(message).toString('hex'); it('should decode an ABI encoded revert error', () => { const expected = new StringRevertError(message); const decoded = RevertError.decode(encoded); expect(decoded.equals(expected)).to.be.true(); }); it('should fail to decode an ABI encoded revert error with an unknown selector', () => { const _encoded = encoded.substr(0, 2) + '00' + encoded.substr(4); const decode = () => RevertError.decode(_encoded); expect(decode).to.be.throw(); }); it('should fail to decode a malformed ABI encoded revert error', () => { const _encoded = encoded.substr(0, encoded.substr.length - 1); const decode = () => RevertError.decode(_encoded); expect(decode).to.be.throw(); }); }); });