NoBey/openzeppelin-contracts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ERC4626 (#3171)

Browse Source
This commit is contained in:
Hadrien Croubois
2022-06-02 10:03:03 +02:00
committed by GitHub
parent 62f2c0531b
commit 5e00787199
10 changed files with 1317 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

99
test/utils/math/Math.test.js
View File

@ -1,12 +1,15 @@
const { BN, constants } = require('@openzeppelin/test-helpers');
const { BN, constants, expectRevert } = require('@openzeppelin/test-helpers');
const { expect } = require('chai');
const { MAX_UINT256 } = constants;
const { Rounding } = require('../../helpers/enums.js');
const MathMock = artifacts.require('MathMock');
contract('Math', function (accounts) {
const min = new BN('1234');
const max = new BN('5678');
const MAX_UINT256_SUB1 = MAX_UINT256.sub(new BN('1'));
const MAX_UINT256_SUB2 = MAX_UINT256.sub(new BN('2'));
beforeEach(async function () {
this.math = await MathMock.new();
@ -85,4 +88,98 @@ contract('Math', function (accounts) {
expect(await this.math.ceilDiv(MAX_UINT256, b)).to.be.bignumber.equal(MAX_UINT256);
});
});
describe('muldiv', function () {
it('divide by 0', async function () {
await expectRevert.unspecified(this.math.mulDiv(1, 1, 0, Rounding.Down));
});
describe('does round down', async function () {
it('small values', async function () {
expect(await this.math.mulDiv('3', '4', '5', Rounding.Down)).to.be.bignumber.equal('2');
expect(await this.math.mulDiv('3', '5', '5', Rounding.Down)).to.be.bignumber.equal('3');
});
it('large values', async function () {
expect(await this.math.mulDiv(
new BN('42'),
MAX_UINT256_SUB1,
MAX_UINT256,
Rounding.Down,
)).to.be.bignumber.equal(new BN('41'));
expect(await this.math.mulDiv(
new BN('17'),
MAX_UINT256,
MAX_UINT256,
Rounding.Down,
)).to.be.bignumber.equal(new BN('17'));
expect(await this.math.mulDiv(
MAX_UINT256_SUB1,
MAX_UINT256_SUB1,
MAX_UINT256,
Rounding.Down,
)).to.be.bignumber.equal(MAX_UINT256_SUB2);
expect(await this.math.mulDiv(
MAX_UINT256,
MAX_UINT256_SUB1,
MAX_UINT256,
Rounding.Down,
)).to.be.bignumber.equal(MAX_UINT256_SUB1);
expect(await this.math.mulDiv(
MAX_UINT256,
MAX_UINT256,
MAX_UINT256,
Rounding.Down,
)).to.be.bignumber.equal(MAX_UINT256);
});
});
describe('does round up', async function () {
it('small values', async function () {
expect(await this.math.mulDiv('3', '4', '5', Rounding.Up)).to.be.bignumber.equal('3');
expect(await this.math.mulDiv('3', '5', '5', Rounding.Up)).to.be.bignumber.equal('3');
});
it('large values', async function () {
expect(await this.math.mulDiv(
new BN('42'),
MAX_UINT256_SUB1,
MAX_UINT256,
Rounding.Up,
)).to.be.bignumber.equal(new BN('42'));
expect(await this.math.mulDiv(
new BN('17'),
MAX_UINT256,
MAX_UINT256,
Rounding.Up,
)).to.be.bignumber.equal(new BN('17'));
expect(await this.math.mulDiv(
MAX_UINT256_SUB1,
MAX_UINT256_SUB1,
MAX_UINT256,
Rounding.Up,
)).to.be.bignumber.equal(MAX_UINT256_SUB1);
expect(await this.math.mulDiv(
MAX_UINT256,
MAX_UINT256_SUB1,
MAX_UINT256,
Rounding.Up,
)).to.be.bignumber.equal(MAX_UINT256_SUB1);
expect(await this.math.mulDiv(
MAX_UINT256,
MAX_UINT256,
MAX_UINT256,
Rounding.Up,
)).to.be.bignumber.equal(MAX_UINT256);
});
});
});
});