123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927 |
- const { expectEvent, constants, time } = require('@openzeppelin/test-helpers');
- const { expectRevertCustomError } = require('../../helpers/customError');
- const { AccessMode } = require('../../helpers/enums');
- const { selector } = require('../../helpers/methods');
- const { clockFromReceipt } = require('../../helpers/time');
- const AccessManager = artifacts.require('$AccessManager');
- const AccessManagedTarget = artifacts.require('$AccessManagedTarget');
- const GROUPS = {
- ADMIN: web3.utils.toBN(0),
- SOME_ADMIN: web3.utils.toBN(17),
- SOME: web3.utils.toBN(42),
- PUBLIC: constants.MAX_UINT256,
- };
- Object.assign(GROUPS, Object.fromEntries(Object.entries(GROUPS).map(([key, value]) => [value, key])));
- const executeDelay = web3.utils.toBN(10);
- const grantDelay = web3.utils.toBN(10);
- const MAX_UINT = n => web3.utils.toBN(1).shln(n).subn(1);
- const split = delay => ({
- oldValue: web3.utils.toBN(delay).shrn(0).and(MAX_UINT(32)).toString(),
- newValue: web3.utils.toBN(delay).shrn(32).and(MAX_UINT(32)).toString(),
- effect: web3.utils.toBN(delay).shrn(64).and(MAX_UINT(48)).toString(),
- });
- contract('AccessManager', function (accounts) {
- const [admin, manager, member, user, other] = accounts;
- beforeEach(async function () {
- this.manager = await AccessManager.new(admin);
- this.target = await AccessManagedTarget.new(this.manager.address);
- // add member to group
- await this.manager.$_setGroupAdmin(GROUPS.SOME, GROUPS.SOME_ADMIN);
- await this.manager.$_setGroupGuardian(GROUPS.SOME, GROUPS.SOME_ADMIN);
- await this.manager.$_grantGroup(GROUPS.SOME_ADMIN, manager, 0, 0);
- await this.manager.$_grantGroup(GROUPS.SOME, member, 0, 0);
- // helpers for indirect calls
- this.call = [this.target.address, selector('fnRestricted()')];
- this.opId = web3.utils.keccak256(
- web3.eth.abi.encodeParameters(['address', 'address', 'bytes'], [user, ...this.call]),
- );
- this.schedule = (opts = {}) => this.manager.schedule(...this.call, { from: user, ...opts });
- this.relay = (opts = {}) => this.manager.relay(...this.call, { from: user, ...opts });
- this.cancel = (opts = {}) => this.manager.cancel(user, ...this.call, { from: user, ...opts });
- });
- it('groups are correctly initialized', async function () {
- // group admin
- expect(await this.manager.getGroupAdmin(GROUPS.ADMIN)).to.be.bignumber.equal(GROUPS.ADMIN);
- expect(await this.manager.getGroupAdmin(GROUPS.SOME_ADMIN)).to.be.bignumber.equal(GROUPS.ADMIN);
- expect(await this.manager.getGroupAdmin(GROUPS.SOME)).to.be.bignumber.equal(GROUPS.SOME_ADMIN);
- expect(await this.manager.getGroupAdmin(GROUPS.PUBLIC)).to.be.bignumber.equal(GROUPS.ADMIN);
- // group guardian
- expect(await this.manager.getGroupGuardian(GROUPS.ADMIN)).to.be.bignumber.equal(GROUPS.ADMIN);
- expect(await this.manager.getGroupGuardian(GROUPS.SOME_ADMIN)).to.be.bignumber.equal(GROUPS.ADMIN);
- expect(await this.manager.getGroupGuardian(GROUPS.SOME)).to.be.bignumber.equal(GROUPS.SOME_ADMIN);
- expect(await this.manager.getGroupGuardian(GROUPS.PUBLIC)).to.be.bignumber.equal(GROUPS.ADMIN);
- // group members
- expect(await this.manager.hasGroup(GROUPS.ADMIN, admin)).to.be.equal(true);
- expect(await this.manager.hasGroup(GROUPS.ADMIN, manager)).to.be.equal(false);
- expect(await this.manager.hasGroup(GROUPS.ADMIN, member)).to.be.equal(false);
- expect(await this.manager.hasGroup(GROUPS.ADMIN, user)).to.be.equal(false);
- expect(await this.manager.hasGroup(GROUPS.SOME_ADMIN, admin)).to.be.equal(false);
- expect(await this.manager.hasGroup(GROUPS.SOME_ADMIN, manager)).to.be.equal(true);
- expect(await this.manager.hasGroup(GROUPS.SOME_ADMIN, member)).to.be.equal(false);
- expect(await this.manager.hasGroup(GROUPS.SOME_ADMIN, user)).to.be.equal(false);
- expect(await this.manager.hasGroup(GROUPS.SOME, admin)).to.be.equal(false);
- expect(await this.manager.hasGroup(GROUPS.SOME, manager)).to.be.equal(false);
- expect(await this.manager.hasGroup(GROUPS.SOME, member)).to.be.equal(true);
- expect(await this.manager.hasGroup(GROUPS.SOME, user)).to.be.equal(false);
- expect(await this.manager.hasGroup(GROUPS.PUBLIC, admin)).to.be.equal(true);
- expect(await this.manager.hasGroup(GROUPS.PUBLIC, manager)).to.be.equal(true);
- expect(await this.manager.hasGroup(GROUPS.PUBLIC, member)).to.be.equal(true);
- expect(await this.manager.hasGroup(GROUPS.PUBLIC, user)).to.be.equal(true);
- });
- describe('Groups management', function () {
- describe('label group', function () {
- it('admin can emit a label event', async function () {
- expectEvent(await this.manager.labelGroup(GROUPS.SOME, 'Some label', { from: admin }), 'GroupLabel', {
- groupId: GROUPS.SOME,
- label: 'Some label',
- });
- });
- it('admin can re-emit a label event', async function () {
- await this.manager.labelGroup(GROUPS.SOME, 'Some label', { from: admin });
- expectEvent(await this.manager.labelGroup(GROUPS.SOME, 'Updated label', { from: admin }), 'GroupLabel', {
- groupId: GROUPS.SOME,
- label: 'Updated label',
- });
- });
- it('emitting a label is restricted', async function () {
- await expectRevertCustomError(
- this.manager.labelGroup(GROUPS.SOME, 'Invalid label', { from: other }),
- 'AccessControlUnauthorizedAccount',
- [other, GROUPS.ADMIN],
- );
- });
- });
- describe('grand group', function () {
- describe('without a grant delay', function () {
- it('without an execute delay', async function () {
- expect(await this.manager.hasGroup(GROUPS.SOME, user)).to.be.false;
- const { receipt } = await this.manager.grantGroup(GROUPS.SOME, user, 0, { from: manager });
- const timestamp = await clockFromReceipt.timestamp(receipt).then(web3.utils.toBN);
- expectEvent(receipt, 'GroupGranted', { groupId: GROUPS.SOME, account: user, since: timestamp, delay: '0' });
- expect(await this.manager.hasGroup(GROUPS.SOME, user)).to.be.true;
- const { delay, since } = await this.manager.getAccess(GROUPS.SOME, user);
- expect(delay).to.be.bignumber.equal('0');
- expect(since).to.be.bignumber.equal(timestamp);
- });
- it('with an execute delay', async function () {
- expect(await this.manager.hasGroup(GROUPS.SOME, user)).to.be.false;
- const { receipt } = await this.manager.grantGroup(GROUPS.SOME, user, executeDelay, { from: manager });
- const timestamp = await clockFromReceipt.timestamp(receipt).then(web3.utils.toBN);
- expectEvent(receipt, 'GroupGranted', {
- groupId: GROUPS.SOME,
- account: user,
- since: timestamp,
- delay: executeDelay,
- });
- expect(await this.manager.hasGroup(GROUPS.SOME, user)).to.be.true;
- const { delay, since } = await this.manager.getAccess(GROUPS.SOME, user);
- expect(delay).to.be.bignumber.equal(executeDelay);
- expect(since).to.be.bignumber.equal(timestamp);
- });
- it('to a user that is already in the group', async function () {
- expect(await this.manager.hasGroup(GROUPS.SOME, member)).to.be.true;
- await expectRevertCustomError(
- this.manager.grantGroup(GROUPS.SOME, member, 0, { from: manager }),
- 'AccessManagerAcountAlreadyInGroup',
- [GROUPS.SOME, member],
- );
- });
- it('to a user that is scheduled for joining the group', async function () {
- await this.manager.$_grantGroup(GROUPS.SOME, user, 10, 0); // grant delay 10
- expect(await this.manager.hasGroup(GROUPS.SOME, user)).to.be.false;
- await expectRevertCustomError(
- this.manager.grantGroup(GROUPS.SOME, user, 0, { from: manager }),
- 'AccessManagerAcountAlreadyInGroup',
- [GROUPS.SOME, user],
- );
- });
- it('grant group is restricted', async function () {
- await expectRevertCustomError(
- this.manager.grantGroup(GROUPS.SOME, user, 0, { from: other }),
- 'AccessControlUnauthorizedAccount',
- [other, GROUPS.SOME_ADMIN],
- );
- });
- });
- describe('with a grant delay', function () {
- beforeEach(async function () {
- await this.manager.$_setGrantDelay(GROUPS.SOME, grantDelay);
- });
- it('granted group is not active immediatly', async function () {
- const { receipt } = await this.manager.grantGroup(GROUPS.SOME, user, 0, { from: manager });
- const timestamp = await clockFromReceipt.timestamp(receipt).then(web3.utils.toBN);
- expectEvent(receipt, 'GroupGranted', {
- groupId: GROUPS.SOME,
- account: user,
- since: timestamp.add(grantDelay),
- delay: '0',
- });
- expect(await this.manager.hasGroup(GROUPS.SOME, user)).to.be.false;
- const { delay, since } = await this.manager.getAccess(GROUPS.SOME, user);
- expect(delay).to.be.bignumber.equal('0');
- expect(since).to.be.bignumber.equal(timestamp.add(grantDelay));
- });
- it('granted group is active after the delay', async function () {
- const { receipt } = await this.manager.grantGroup(GROUPS.SOME, user, 0, { from: manager });
- const timestamp = await clockFromReceipt.timestamp(receipt).then(web3.utils.toBN);
- expectEvent(receipt, 'GroupGranted', {
- groupId: GROUPS.SOME,
- account: user,
- since: timestamp.add(grantDelay),
- delay: '0',
- });
- await time.increase(grantDelay);
- expect(await this.manager.hasGroup(GROUPS.SOME, user)).to.be.true;
- const { delay, since } = await this.manager.getAccess(GROUPS.SOME, user);
- expect(delay).to.be.bignumber.equal('0');
- expect(since).to.be.bignumber.equal(timestamp.add(grantDelay));
- });
- });
- });
- describe('revoke group', function () {
- it('from a user that is already in the group', async function () {
- expect(await this.manager.hasGroup(GROUPS.SOME, member)).to.be.true;
- const { receipt } = await this.manager.revokeGroup(GROUPS.SOME, member, { from: manager });
- expectEvent(receipt, 'GroupRevoked', { groupId: GROUPS.SOME, account: member });
- expect(await this.manager.hasGroup(GROUPS.SOME, member)).to.be.false;
- const { delay, since } = await this.manager.getAccess(GROUPS.SOME, user);
- expect(delay).to.be.bignumber.equal('0');
- expect(since).to.be.bignumber.equal('0');
- });
- it('from a user that is scheduled for joining the group', async function () {
- await this.manager.$_grantGroup(GROUPS.SOME, user, 10, 0); // grant delay 10
- expect(await this.manager.hasGroup(GROUPS.SOME, user)).to.be.false;
- const { receipt } = await this.manager.revokeGroup(GROUPS.SOME, user, { from: manager });
- expectEvent(receipt, 'GroupRevoked', { groupId: GROUPS.SOME, account: user });
- expect(await this.manager.hasGroup(GROUPS.SOME, user)).to.be.false;
- const { delay, since } = await this.manager.getAccess(GROUPS.SOME, user);
- expect(delay).to.be.bignumber.equal('0');
- expect(since).to.be.bignumber.equal('0');
- });
- it('from a user that is not in the group', async function () {
- expect(await this.manager.hasGroup(GROUPS.SOME, user)).to.be.false;
- await expectRevertCustomError(
- this.manager.revokeGroup(GROUPS.SOME, user, { from: manager }),
- 'AccessManagerAcountNotInGroup',
- [GROUPS.SOME, user],
- );
- });
- it('revoke group is restricted', async function () {
- await expectRevertCustomError(
- this.manager.revokeGroup(GROUPS.SOME, member, { from: other }),
- 'AccessControlUnauthorizedAccount',
- [other, GROUPS.SOME_ADMIN],
- );
- });
- });
- describe('renounce group', function () {
- it('for a user that is already in the group', async function () {
- expect(await this.manager.hasGroup(GROUPS.SOME, member)).to.be.true;
- const { receipt } = await this.manager.renounceGroup(GROUPS.SOME, member, { from: member });
- expectEvent(receipt, 'GroupRevoked', { groupId: GROUPS.SOME, account: member });
- expect(await this.manager.hasGroup(GROUPS.SOME, member)).to.be.false;
- const { delay, since } = await this.manager.getAccess(GROUPS.SOME, member);
- expect(delay).to.be.bignumber.equal('0');
- expect(since).to.be.bignumber.equal('0');
- });
- it('for a user that is schedule for joining the group', async function () {
- await this.manager.$_grantGroup(GROUPS.SOME, user, 10, 0); // grant delay 10
- expect(await this.manager.hasGroup(GROUPS.SOME, user)).to.be.false;
- const { receipt } = await this.manager.renounceGroup(GROUPS.SOME, user, { from: user });
- expectEvent(receipt, 'GroupRevoked', { groupId: GROUPS.SOME, account: user });
- expect(await this.manager.hasGroup(GROUPS.SOME, user)).to.be.false;
- const { delay, since } = await this.manager.getAccess(GROUPS.SOME, user);
- expect(delay).to.be.bignumber.equal('0');
- expect(since).to.be.bignumber.equal('0');
- });
- it('for a user that is not in the group', async function () {
- await expectRevertCustomError(
- this.manager.renounceGroup(GROUPS.SOME, user, { from: user }),
- 'AccessManagerAcountNotInGroup',
- [GROUPS.SOME, user],
- );
- });
- it('bad user confirmation', async function () {
- await expectRevertCustomError(
- this.manager.renounceGroup(GROUPS.SOME, member, { from: user }),
- 'AccessManagerBadConfirmation',
- [],
- );
- });
- });
- describe('change group admin', function () {
- it("admin can set any group's admin", async function () {
- expect(await this.manager.getGroupAdmin(GROUPS.SOME)).to.be.bignumber.equal(GROUPS.SOME_ADMIN);
- const { receipt } = await this.manager.setGroupAdmin(GROUPS.SOME, GROUPS.ADMIN, { from: admin });
- expectEvent(receipt, 'GroupAdminChanged', { groupId: GROUPS.SOME, admin: GROUPS.ADMIN });
- expect(await this.manager.getGroupAdmin(GROUPS.SOME)).to.be.bignumber.equal(GROUPS.ADMIN);
- });
- it("seeting a group's admin is restricted", async function () {
- await expectRevertCustomError(
- this.manager.setGroupAdmin(GROUPS.SOME, GROUPS.SOME, { from: manager }),
- 'AccessControlUnauthorizedAccount',
- [manager, GROUPS.ADMIN],
- );
- });
- });
- describe('change group guardian', function () {
- it("admin can set any group's admin", async function () {
- expect(await this.manager.getGroupGuardian(GROUPS.SOME)).to.be.bignumber.equal(GROUPS.SOME_ADMIN);
- const { receipt } = await this.manager.setGroupGuardian(GROUPS.SOME, GROUPS.ADMIN, { from: admin });
- expectEvent(receipt, 'GroupGuardianChanged', { groupId: GROUPS.SOME, guardian: GROUPS.ADMIN });
- expect(await this.manager.getGroupGuardian(GROUPS.SOME)).to.be.bignumber.equal(GROUPS.ADMIN);
- });
- it("setting a group's admin is restricted", async function () {
- await expectRevertCustomError(
- this.manager.setGroupGuardian(GROUPS.SOME, GROUPS.SOME, { from: other }),
- 'AccessControlUnauthorizedAccount',
- [other, GROUPS.ADMIN],
- );
- });
- });
- describe('change execution delay', function () {
- it('increassing the delay has immediate effect', async function () {
- const oldDelay = web3.utils.toBN(10);
- const newDelay = web3.utils.toBN(100);
- const { receipt: receipt1 } = await this.manager.$_setExecuteDelay(GROUPS.SOME, member, oldDelay);
- const timestamp1 = await clockFromReceipt.timestamp(receipt1).then(web3.utils.toBN);
- const delayBefore = await this.manager.getAccess(GROUPS.SOME, member).then(([, delay]) => split(delay));
- expect(delayBefore.oldValue).to.be.bignumber.equal('0');
- expect(delayBefore.newValue).to.be.bignumber.equal(oldDelay);
- expect(delayBefore.effect).to.be.bignumber.equal(timestamp1);
- const { receipt: receipt2 } = await this.manager.setExecuteDelay(GROUPS.SOME, member, newDelay, {
- from: manager,
- });
- const timestamp2 = await clockFromReceipt.timestamp(receipt2).then(web3.utils.toBN);
- expectEvent(receipt2, 'GroupExecutionDelayUpdate', {
- groupId: GROUPS.SOME,
- account: member,
- delay: newDelay,
- from: timestamp2,
- });
- // immediate effect
- const delayAfter = await this.manager.getAccess(GROUPS.SOME, member).then(([, delay]) => split(delay));
- expect(delayAfter.oldValue).to.be.bignumber.equal(oldDelay);
- expect(delayAfter.newValue).to.be.bignumber.equal(newDelay);
- expect(delayAfter.effect).to.be.bignumber.equal(timestamp2);
- });
- it('decreassing the delay takes time', async function () {
- const oldDelay = web3.utils.toBN(100);
- const newDelay = web3.utils.toBN(10);
- const { receipt: receipt1 } = await this.manager.$_setExecuteDelay(GROUPS.SOME, member, oldDelay);
- const timestamp1 = await clockFromReceipt.timestamp(receipt1).then(web3.utils.toBN);
- const delayBefore = await this.manager.getAccess(GROUPS.SOME, member).then(([, delay]) => split(delay));
- expect(delayBefore.oldValue).to.be.bignumber.equal('0');
- expect(delayBefore.newValue).to.be.bignumber.equal(oldDelay);
- expect(delayBefore.effect).to.be.bignumber.equal(timestamp1);
- const { receipt: receipt2 } = await this.manager.setExecuteDelay(GROUPS.SOME, member, newDelay, {
- from: manager,
- });
- const timestamp2 = await clockFromReceipt.timestamp(receipt2).then(web3.utils.toBN);
- expectEvent(receipt2, 'GroupExecutionDelayUpdate', {
- groupId: GROUPS.SOME,
- account: member,
- delay: newDelay,
- from: timestamp2.add(oldDelay).sub(newDelay),
- });
- // delayed effect
- const delayAfter = await this.manager.getAccess(GROUPS.SOME, member).then(([, delay]) => split(delay));
- expect(delayAfter.oldValue).to.be.bignumber.equal(oldDelay);
- expect(delayAfter.newValue).to.be.bignumber.equal(newDelay);
- expect(delayAfter.effect).to.be.bignumber.equal(timestamp2.add(oldDelay).sub(newDelay));
- });
- it('cannot set the delay of a non member', async function () {
- await expectRevertCustomError(
- this.manager.setExecuteDelay(GROUPS.SOME, other, executeDelay, { from: manager }),
- 'AccessManagerAcountNotInGroup',
- [GROUPS.SOME, other],
- );
- });
- it('can set a user execution delay during the grant delay', async function () {
- await this.manager.$_grantGroup(GROUPS.SOME, other, 10, 0);
- const { receipt } = await this.manager.setExecuteDelay(GROUPS.SOME, other, executeDelay, { from: manager });
- const timestamp = await clockFromReceipt.timestamp(receipt).then(web3.utils.toBN);
- expectEvent(receipt, 'GroupExecutionDelayUpdate', {
- groupId: GROUPS.SOME,
- account: other,
- delay: executeDelay,
- from: timestamp,
- });
- });
- it('changing the execution delay is restricted', async function () {
- await expectRevertCustomError(
- this.manager.setExecuteDelay(GROUPS.SOME, member, executeDelay, { from: other }),
- 'AccessControlUnauthorizedAccount',
- [GROUPS.SOME_ADMIN, other],
- );
- });
- });
- describe('change grant delay', function () {
- it('increassing the delay has immediate effect', async function () {
- const oldDelay = web3.utils.toBN(10);
- const newDelay = web3.utils.toBN(100);
- await this.manager.$_setGrantDelay(GROUPS.SOME, oldDelay);
- expect(await this.manager.getGroupGrantDelay(GROUPS.SOME)).to.be.bignumber.equal(oldDelay);
- const { receipt } = await this.manager.setGrantDelay(GROUPS.SOME, newDelay, { from: admin });
- const timestamp = await clockFromReceipt.timestamp(receipt).then(web3.utils.toBN);
- expectEvent(receipt, 'GroupGrantDelayChanged', { groupId: GROUPS.SOME, delay: newDelay, from: timestamp });
- expect(await this.manager.getGroupGrantDelay(GROUPS.SOME)).to.be.bignumber.equal(newDelay);
- });
- it('increassing the delay has delay effect', async function () {
- const oldDelay = web3.utils.toBN(100);
- const newDelay = web3.utils.toBN(10);
- await this.manager.$_setGrantDelay(GROUPS.SOME, oldDelay);
- expect(await this.manager.getGroupGrantDelay(GROUPS.SOME)).to.be.bignumber.equal(oldDelay);
- const { receipt } = await this.manager.setGrantDelay(GROUPS.SOME, newDelay, { from: admin });
- const timestamp = await clockFromReceipt.timestamp(receipt).then(web3.utils.toBN);
- expectEvent(receipt, 'GroupGrantDelayChanged', {
- groupId: GROUPS.SOME,
- delay: newDelay,
- from: timestamp.add(oldDelay).sub(newDelay),
- });
- expect(await this.manager.getGroupGrantDelay(GROUPS.SOME)).to.be.bignumber.equal(oldDelay);
- await time.increase(oldDelay.sub(newDelay));
- expect(await this.manager.getGroupGrantDelay(GROUPS.SOME)).to.be.bignumber.equal(newDelay);
- });
- it('changing the grant delay is restricted', async function () {
- await expectRevertCustomError(
- this.manager.setGrantDelay(GROUPS.SOME, grantDelay, { from: other }),
- 'AccessControlUnauthorizedAccount',
- [GROUPS.ADMIN, other],
- );
- });
- });
- });
- describe('Mode management', function () {
- for (const [modeName, mode] of Object.entries(AccessMode)) {
- describe(`setContractMode${modeName}`, function () {
- it('set the mode and emits an event', async function () {
- // set the target to another mode, so we can check the effects
- await this.manager.$_setContractMode(
- this.target.address,
- Object.values(AccessMode).find(m => m != mode),
- );
- expect(await this.manager.getContractMode(this.target.address)).to.not.be.bignumber.equal(mode);
- expectEvent(
- await this.manager[`setContractMode${modeName}`](this.target.address, { from: admin }),
- 'AccessModeUpdated',
- { target: this.target.address, mode },
- );
- expect(await this.manager.getContractMode(this.target.address)).to.be.bignumber.equal(mode);
- });
- it('is restricted', async function () {
- await expectRevertCustomError(
- this.manager[`setContractMode${modeName}`](this.target.address, { from: other }),
- 'AccessControlUnauthorizedAccount',
- [other, GROUPS.ADMIN],
- );
- });
- });
- }
- });
- describe('Change function permissions', function () {
- const sigs = ['someFunction()', 'someOtherFunction(uint256)', 'oneMoreFunction(address,uint8)'].map(selector);
- it('admin can set function allowed group', async function () {
- for (const sig of sigs) {
- expect(await this.manager.getFunctionAllowedGroup(this.target.address, sig)).to.be.bignumber.equal(
- GROUPS.ADMIN,
- );
- }
- const { receipt: receipt1 } = await this.manager.setFunctionAllowedGroup(this.target.address, sigs, GROUPS.SOME, {
- from: admin,
- });
- for (const sig of sigs) {
- expectEvent(receipt1, 'FunctionAllowedGroupUpdated', {
- target: this.target.address,
- selector: sig,
- groupId: GROUPS.SOME,
- });
- expect(await this.manager.getFunctionAllowedGroup(this.target.address, sig)).to.be.bignumber.equal(GROUPS.SOME);
- }
- const { receipt: receipt2 } = await this.manager.setFunctionAllowedGroup(
- this.target.address,
- [sigs[1]],
- GROUPS.SOME_ADMIN,
- { from: admin },
- );
- expectEvent(receipt2, 'FunctionAllowedGroupUpdated', {
- target: this.target.address,
- selector: sigs[1],
- groupId: GROUPS.SOME_ADMIN,
- });
- for (const sig of sigs) {
- expect(await this.manager.getFunctionAllowedGroup(this.target.address, sig)).to.be.bignumber.equal(
- sig == sigs[1] ? GROUPS.SOME_ADMIN : GROUPS.SOME,
- );
- }
- });
- it('changing function permissions is restricted', async function () {
- await expectRevertCustomError(
- this.manager.setFunctionAllowedGroup(this.target.address, sigs, GROUPS.SOME, { from: other }),
- 'AccessControlUnauthorizedAccount',
- [other, GROUPS.ADMIN],
- );
- });
- });
- describe('Calling restricted & unrestricted functions', function () {
- const product = (...arrays) => arrays.reduce((a, b) => a.flatMap(ai => b.map(bi => [ai, bi].flat())));
- for (const [callerOpt, targetOpt] of product(
- [
- { groups: [] },
- { groups: [GROUPS.SOME] },
- { groups: [GROUPS.SOME], delay: executeDelay },
- { groups: [GROUPS.SOME, GROUPS.PUBLIC], delay: executeDelay },
- ],
- [
- { mode: AccessMode.Open },
- { mode: AccessMode.Closed },
- { mode: AccessMode.Custom, group: GROUPS.ADMIN },
- { mode: AccessMode.Custom, group: GROUPS.SOME },
- { mode: AccessMode.Custom, group: GROUPS.PUBLIC },
- ],
- )) {
- const public =
- targetOpt.mode == AccessMode.Open || (targetOpt.mode == AccessMode.Custom && targetOpt.group == GROUPS.PUBLIC);
- // can we call with a delay ?
- const indirectSuccess =
- public || (targetOpt.mode == AccessMode.Custom && callerOpt.groups?.includes(targetOpt.group));
- // can we call without a delay ?
- const directSuccess =
- public ||
- (targetOpt.mode == AccessMode.Custom && callerOpt.groups?.includes(targetOpt.group) && !callerOpt.delay);
- const description = [
- 'Caller in groups',
- '[' + (callerOpt.groups ?? []).map(groupId => GROUPS[groupId]).join(', ') + ']',
- callerOpt.delay ? 'with a delay' : 'without a delay',
- '+',
- 'contract in mode',
- Object.keys(AccessMode)[targetOpt.mode.toNumber()],
- targetOpt.mode == AccessMode.Custom ? `(${GROUPS[targetOpt.group]})` : '',
- ].join(' ');
- describe(description, function () {
- beforeEach(async function () {
- // setup
- await Promise.all([
- this.manager.$_setContractMode(this.target.address, targetOpt.mode),
- targetOpt.group &&
- this.manager.$_setFunctionAllowedGroup(this.target.address, selector('fnRestricted()'), targetOpt.group),
- targetOpt.group &&
- this.manager.$_setFunctionAllowedGroup(
- this.target.address,
- selector('fnUnrestricted()'),
- targetOpt.group,
- ),
- ...(callerOpt.groups ?? [])
- .filter(groupId => groupId != GROUPS.PUBLIC)
- .map(groupId => this.manager.$_grantGroup(groupId, user, 0, callerOpt.delay ?? 0)),
- ]);
- // post setup checks
- expect(await this.manager.getContractMode(this.target.address)).to.be.bignumber.equal(targetOpt.mode);
- if (targetOpt.group) {
- expect(
- await this.manager.getFunctionAllowedGroup(this.target.address, selector('fnRestricted()')),
- ).to.be.bignumber.equal(targetOpt.group);
- expect(
- await this.manager.getFunctionAllowedGroup(this.target.address, selector('fnUnrestricted()')),
- ).to.be.bignumber.equal(targetOpt.group);
- }
- for (const groupId of callerOpt.groups ?? []) {
- const access = await this.manager.getAccess(groupId, user);
- if (groupId == GROUPS.PUBLIC) {
- expect(access.since).to.be.bignumber.eq('0');
- expect(access.delay).to.be.bignumber.eq('0');
- } else {
- expect(access.since).to.be.bignumber.gt('0');
- expect(access.delay).to.be.bignumber.eq(String(callerOpt.delay ?? 0));
- }
- }
- });
- it('canCall', async function () {
- const result = await this.manager.canCall(user, this.target.address, selector('fnRestricted()'));
- expect(result[0]).to.be.equal(directSuccess);
- expect(result[1]).to.be.bignumber.equal(!directSuccess && indirectSuccess ? callerOpt.delay ?? '0' : '0');
- });
- it('Calling a non restricted function never revert', async function () {
- expectEvent(await this.target.fnUnrestricted({ from: user }), 'CalledUnrestricted', {
- caller: user,
- });
- });
- it(`Calling a restricted function directly should ${directSuccess ? 'succeed' : 'revert'}`, async function () {
- const promise = this.target.fnRestricted({ from: user });
- if (directSuccess) {
- expectEvent(await promise, 'CalledRestricted', { caller: user });
- } else {
- await expectRevertCustomError(promise, 'AccessManagedUnauthorized', [user]);
- }
- });
- it('Calling indirectly: only relay', async function () {
- // relay without schedule
- if (directSuccess) {
- const { receipt, tx } = await this.relay();
- expectEvent.notEmitted(receipt, 'Executed', { operationId: this.opId });
- expectEvent.inTransaction(tx, this.target, 'Calledrestricted', { caller: this.manager.address });
- } else if (indirectSuccess) {
- await expectRevertCustomError(this.relay(), 'AccessManagerNotScheduled', [this.opId]);
- } else {
- await expectRevertCustomError(this.relay(), 'AccessManagerUnauthorizedCall', [user, ...this.call]);
- }
- });
- it('Calling indirectly: schedule and relay', async function () {
- if (directSuccess || indirectSuccess) {
- const { receipt } = await this.schedule();
- const timestamp = await clockFromReceipt.timestamp(receipt).then(web3.utils.toBN);
- expectEvent(receipt, 'Scheduled', {
- operationId: this.opId,
- caller: user,
- target: this.call[0],
- data: this.call[1],
- });
- // if can call directly, delay should be 0. Otherwize, the delay should be applied
- expect(await this.manager.getSchedule(this.opId)).to.be.bignumber.equal(
- timestamp.add(directSuccess ? web3.utils.toBN(0) : callerOpt.delay),
- );
- // execute without wait
- if (directSuccess) {
- const { receipt, tx } = await this.relay();
- expectEvent(receipt, 'Executed', { operationId: this.opId });
- expectEvent.inTransaction(tx, this.target, 'Calledrestricted', { caller: this.manager.address });
- expect(await this.manager.getSchedule(this.opId)).to.be.bignumber.equal('0');
- } else if (indirectSuccess) {
- await expectRevertCustomError(this.relay(), 'AccessManagerNotReady', [this.opId]);
- } else {
- await expectRevertCustomError(this.relay(), 'AccessManagerUnauthorizedCall', [user, ...this.call]);
- }
- } else {
- await expectRevertCustomError(this.schedule(), 'AccessManagerUnauthorizedCall', [user, ...this.call]);
- }
- });
- it('Calling indirectly: schedule wait and relay', async function () {
- if (directSuccess || indirectSuccess) {
- const { receipt } = await this.schedule();
- const timestamp = await clockFromReceipt.timestamp(receipt).then(web3.utils.toBN);
- expectEvent(receipt, 'Scheduled', {
- operationId: this.opId,
- caller: user,
- target: this.call[0],
- data: this.call[1],
- });
- // if can call directly, delay should be 0. Otherwize, the delay should be applied
- expect(await this.manager.getSchedule(this.opId)).to.be.bignumber.equal(
- timestamp.add(directSuccess ? web3.utils.toBN(0) : callerOpt.delay),
- );
- // wait
- await time.increase(callerOpt.delay ?? 0);
- // execute without wait
- if (directSuccess || indirectSuccess) {
- const { receipt, tx } = await this.relay();
- expectEvent(receipt, 'Executed', { operationId: this.opId });
- expectEvent.inTransaction(tx, this.target, 'Calledrestricted', { caller: this.manager.address });
- expect(await this.manager.getSchedule(this.opId)).to.be.bignumber.equal('0');
- } else {
- await expectRevertCustomError(this.relay(), 'AccessManagerUnauthorizedCall', [user, ...this.call]);
- }
- } else {
- await expectRevertCustomError(this.schedule(), 'AccessManagerUnauthorizedCall', [user, ...this.call]);
- }
- });
- });
- }
- });
- describe('Indirect execution corner-cases', async function () {
- beforeEach(async function () {
- await this.manager.$_setFunctionAllowedGroup(...this.call, GROUPS.SOME);
- await this.manager.$_grantGroup(GROUPS.SOME, user, 0, executeDelay);
- });
- it('Checking canCall when caller is the manager depend on the _relayIdentifier', async function () {
- expect(await this.manager.getContractMode(this.target.address)).to.be.bignumber.equal(AccessMode.Custom);
- const result = await this.manager.canCall(this.manager.address, this.target.address, '0x00000000');
- expect(result[0]).to.be.false;
- expect(result[1]).to.be.bignumber.equal('0');
- });
- it('Cannot execute earlier', async function () {
- const { receipt } = await this.schedule();
- const timestamp = await clockFromReceipt.timestamp(receipt).then(web3.utils.toBN);
- expect(await this.manager.getSchedule(this.opId)).to.be.bignumber.equal(timestamp.add(executeDelay));
- // we need to set the clock 2 seconds before the value, because the increaseTo "consumes" the timestamp
- // and the next transaction will be one after that (see check bellow)
- await time.increaseTo(timestamp.add(executeDelay).subn(2));
- // too early
- await expectRevertCustomError(this.relay(), 'AccessManagerNotReady', [this.opId]);
- // the revert happened one second before the execution delay expired
- expect(await time.latest()).to.be.bignumber.equal(timestamp.add(executeDelay).subn(1));
- // ok
- await this.relay();
- // the success happened when the delay was reached (earliest possible)
- expect(await time.latest()).to.be.bignumber.equal(timestamp.add(executeDelay));
- });
- it('Cannot schedule an already scheduled operation', async function () {
- const { receipt } = await this.schedule();
- expectEvent(receipt, 'Scheduled', {
- operationId: this.opId,
- caller: user,
- target: this.call[0],
- data: this.call[1],
- });
- await expectRevertCustomError(this.schedule(), 'AccessManagerAlreadyScheduled', [this.opId]);
- });
- it('Cannot cancel an operation that is not scheduled', async function () {
- await expectRevertCustomError(this.cancel(), 'AccessManagerNotScheduled', [this.opId]);
- });
- it('Cannot cancel an operation that is not already relayed', async function () {
- await this.schedule();
- await time.increase(executeDelay);
- await this.relay();
- await expectRevertCustomError(this.cancel(), 'AccessManagerNotScheduled', [this.opId]);
- });
- it('Scheduler can cancel', async function () {
- await this.schedule();
- expect(await this.manager.getSchedule(this.opId)).to.not.be.bignumber.equal('0');
- expectEvent(await this.cancel({ from: manager }), 'Canceled', { operationId: this.opId });
- expect(await this.manager.getSchedule(this.opId)).to.be.bignumber.equal('0');
- });
- it('Guardian can cancel', async function () {
- await this.schedule();
- expect(await this.manager.getSchedule(this.opId)).to.not.be.bignumber.equal('0');
- expectEvent(await this.cancel({ from: manager }), 'Canceled', { operationId: this.opId });
- expect(await this.manager.getSchedule(this.opId)).to.be.bignumber.equal('0');
- });
- it('Cancel is restricted', async function () {
- await this.schedule();
- expect(await this.manager.getSchedule(this.opId)).to.not.be.bignumber.equal('0');
- await expectRevertCustomError(this.cancel({ from: other }), 'AccessManagerCannotCancel', [
- other,
- user,
- ...this.call,
- ]);
- expect(await this.manager.getSchedule(this.opId)).to.not.be.bignumber.equal('0');
- });
- it('Can re-schedule after execution', async function () {
- await this.schedule();
- await time.increase(executeDelay);
- await this.relay();
- // reschedule
- const { receipt } = await this.schedule();
- expectEvent(receipt, 'Scheduled', {
- operationId: this.opId,
- caller: user,
- target: this.call[0],
- data: this.call[1],
- });
- });
- it('Can re-schedule after cancel', async function () {
- await this.schedule();
- await this.cancel();
- // reschedule
- const { receipt } = await this.schedule();
- expectEvent(receipt, 'Scheduled', {
- operationId: this.opId,
- caller: user,
- target: this.call[0],
- data: this.call[1],
- });
- });
- });
- describe('authority update', function () {
- beforeEach(async function () {
- this.newManager = await AccessManager.new(admin);
- });
- it('admin can change authority', async function () {
- expect(await this.target.authority()).to.be.equal(this.manager.address);
- const { tx } = await this.manager.updateAuthority(this.target.address, this.newManager.address, { from: admin });
- expectEvent.inTransaction(tx, this.target, 'AuthorityUpdated', { authority: this.newManager.address });
- expect(await this.target.authority()).to.be.equal(this.newManager.address);
- });
- it('cannot set an address without code as the authority', async function () {
- await expectRevertCustomError(
- this.manager.updateAuthority(this.target.address, user, { from: admin }),
- 'AccessManagedInvalidAuthority',
- [user],
- );
- });
- it('updateAuthority is restricted on manager', async function () {
- await expectRevertCustomError(
- this.manager.updateAuthority(this.target.address, this.newManager.address, { from: other }),
- 'AccessControlUnauthorizedAccount',
- [other, GROUPS.ADMIN],
- );
- });
- it('setAuthority is restricted on AccessManaged', async function () {
- await expectRevertCustomError(
- this.target.setAuthority(this.newManager.address, { from: admin }),
- 'AccessManagedUnauthorized',
- [admin],
- );
- });
- });
- });
|