Extend Checkpoints with new sizes and lookup mechanisms (#3589)

This commit is contained in:
Hadrien Croubois
2022-08-30 21:32:12 +02:00
committed by GitHub
parent 4b16e88747
commit 71aaca2d9d
20 changed files with 1189 additions and 153 deletions

View File

@ -2,7 +2,9 @@ require('@openzeppelin/test-helpers');
const { expect } = require('chai');
const ArraysImpl = artifacts.require('ArraysImpl');
const AddressArraysMock = artifacts.require('AddressArraysMock');
const Bytes32ArraysMock = artifacts.require('Bytes32ArraysMock');
const Uint256ArraysMock = artifacts.require('Uint256ArraysMock');
contract('Arrays', function (accounts) {
describe('findUpperBound', function () {
@ -10,7 +12,7 @@ contract('Arrays', function (accounts) {
const EVEN_ELEMENTS_ARRAY = [11, 12, 13, 14, 15, 16, 17, 18, 19, 20];
beforeEach(async function () {
this.arrays = await ArraysImpl.new(EVEN_ELEMENTS_ARRAY);
this.arrays = await Uint256ArraysMock.new(EVEN_ELEMENTS_ARRAY);
});
it('returns correct index for the basic case', async function () {
@ -38,7 +40,7 @@ contract('Arrays', function (accounts) {
const ODD_ELEMENTS_ARRAY = [11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21];
beforeEach(async function () {
this.arrays = await ArraysImpl.new(ODD_ELEMENTS_ARRAY);
this.arrays = await Uint256ArraysMock.new(ODD_ELEMENTS_ARRAY);
});
it('returns correct index for the basic case', async function () {
@ -66,7 +68,7 @@ contract('Arrays', function (accounts) {
const WITH_GAP_ARRAY = [11, 12, 13, 14, 15, 20, 21, 22, 23, 24];
beforeEach(async function () {
this.arrays = await ArraysImpl.new(WITH_GAP_ARRAY);
this.arrays = await Uint256ArraysMock.new(WITH_GAP_ARRAY);
});
it('returns index of first element in next filled range', async function () {
@ -76,7 +78,7 @@ contract('Arrays', function (accounts) {
context('Empty array', function () {
beforeEach(async function () {
this.arrays = await ArraysImpl.new([]);
this.arrays = await Uint256ArraysMock.new([]);
});
it('always returns 0 for empty array', async function () {
@ -84,4 +86,20 @@ contract('Arrays', function (accounts) {
});
});
});
describe('unsafeAccess', function () {
for (const { type, artifact, elements } of [
{ type: 'address', artifact: AddressArraysMock, elements: Array(10).fill().map(() => web3.utils.randomHex(20)) },
{ type: 'bytes32', artifact: Bytes32ArraysMock, elements: Array(10).fill().map(() => web3.utils.randomHex(32)) },
{ type: 'uint256', artifact: Uint256ArraysMock, elements: Array(10).fill().map(() => web3.utils.randomHex(32)) },
]) {
it(type, async function () {
const contract = await artifact.new(elements);
for (const i in elements) {
expect(await contract.unsafeAccess(i)).to.be.bignumber.equal(elements[i]);
}
});
}
});
});