Files
openzeppelin-contracts/test/utils/structs/EnumerableMap/helpers.js
Gaspar Dip ca755ce799 Add AddressToUintMap (#3150)
* add AddressToUintMap

* Update contracts/utils/structs/EnumerableMap.sol

Co-authored-by: Francisco Giordano <frangio.1@gmail.com>

* address comments

* lint code

* merge mocks into a single file

* add PR link to changelog entry

Co-authored-by: Francisco Giordano <frangio.1@gmail.com>
Co-authored-by: Hadrien Croubois <hadrien.croubois@gmail.com>
2022-02-01 17:37:32 +01:00

32 lines
1.0 KiB
JavaScript

const zip = require('lodash.zip');
const toStringArray = (array) => array.map((i) => i.toString());
async function expectMembersMatch (map, keys, values) {
const stringKeys = toStringArray(keys);
const stringValues = toStringArray(values);
expect(keys.length).to.equal(values.length);
await Promise.all(keys.map(async (key) => expect(await map.contains(key)).to.equal(true)));
expect(await map.length()).to.bignumber.equal(keys.length.toString());
expect(toStringArray(await Promise.all(keys.map((key) => map.get(key))))).to.have.same.members(stringValues);
// to compare key-value pairs, we zip keys and values
// we also stringify pairs because this helper is used for multiple types of maps
expect(
await Promise.all(
[...Array(keys.length).keys()].map(async (index) => {
const { key, value } = await map.at(index);
return [key.toString(), value.toString()];
}),
),
).to.have.same.deep.members(zip(stringKeys, stringValues));
}
module.exports = {
expectMembersMatch,
};