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

Revert on useless operations.

Browse Source
This commit is contained in:
Alberto Cuesta Cañada
2020-01-23 20:43:57 +00:00
parent dfdf794e3c
commit 41cb935930
2 changed files with 22 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
contracts/utils/EnumerableSet.sol
View File

@ -33,9 +33,8 @@ library EnumerableSet {
function add(AddressSet storage set, address value)
internal
{
if (!contains(set, value)){
set.index[value] = set.values.push(value);
}
require(!contains(set, value), "EnumerableSet: value already in set");
set.index[value] = set.values.push(value);
}
/**
@ -44,11 +43,10 @@ library EnumerableSet {
function remove(AddressSet storage set, address value)
internal
{
if (contains(set, value)) {
set.values[set.index[value] - 1] = set.values[set.values.length - 1];
set.values.pop();
delete set.index[value];
}
require(contains(set, value), "EnumerableSet: value not in set");
set.values[set.index[value] - 1] = set.values[set.values.length - 1];
set.values.pop();
delete set.index[value];
}
/**

17
test/utils/EnumerableSet.test.js
View File

@ -1,5 +1,5 @@
const { contract } = require('@openzeppelin/test-environment');
const { expectRevert } = require('@openzeppelin/test-helpers');
const { expect } = require('chai');
const EnumerableSetMock = contract.fromArtifact('EnumerableSetMock');
@ -31,12 +31,27 @@ describe('EnumerableSet', function () {
expect(await this.set.testContains(c)).to.equal(false);
});
it('can\'t add same value twice.', async function () {
await this.set.testAdd(a);
await expectRevert(
this.set.testAdd(a),
'EnumerableSet: value already in set',
);
});
it('removes values.', async function () {
await this.set.testAdd(a);
await this.set.testRemove(a);
expect(await this.set.testContains(a)).to.equal(false);
});
it('can\'t remove values that are not in the set.', async function () {
await expectRevert(
this.set.testRemove(a),
'EnumerableSet: value not in set',
);
});
it('enumerates values as an empty array', async function () {
expect(await this.set.testEnumerate()).to.eql([]);
});