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

Add NoncesKeyed variant (#5272)

Browse Source
This commit is contained in:
Hadrien Croubois
2024-10-23 09:16:10 +02:00
committed by GitHub
parent 205f59e9b6
commit 2fa4d103fe
7 changed files with 242 additions and 62 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
test/utils/Nonces.test.js
View File

@ -1,13 +1,10 @@
const { ethers } = require('hardhat');
const { expect } = require('chai');
const { loadFixture } = require('@nomicfoundation/hardhat-network-helpers');
const { shouldBehaveLikeNonces } = require('./Nonces.behavior');
async function fixture() {
const [sender, other] = await ethers.getSigners();
const mock = await ethers.deployContract('$Nonces');
return { sender, other, mock };
return { mock };
}
describe('Nonces', function () {
@ -15,61 +12,5 @@ describe('Nonces', function () {
Object.assign(this, await loadFixture(fixture));
});
it('gets a nonce', async function () {
expect(await this.mock.nonces(this.sender)).to.equal(0n);
});
describe('_useNonce', function () {
it('increments a nonce', async function () {
expect(await this.mock.nonces(this.sender)).to.equal(0n);
await expect(await this.mock.$_useNonce(this.sender))
.to.emit(this.mock, 'return$_useNonce')
.withArgs(0n);
expect(await this.mock.nonces(this.sender)).to.equal(1n);
});
it("increments only sender's nonce", async function () {
expect(await this.mock.nonces(this.sender)).to.equal(0n);
expect(await this.mock.nonces(this.other)).to.equal(0n);
await this.mock.$_useNonce(this.sender);
expect(await this.mock.nonces(this.sender)).to.equal(1n);
expect(await this.mock.nonces(this.other)).to.equal(0n);
});
});
describe('_useCheckedNonce', function () {
it('increments a nonce', async function () {
const currentNonce = await this.mock.nonces(this.sender);
expect(currentNonce).to.equal(0n);
await this.mock.$_useCheckedNonce(this.sender, currentNonce);
expect(await this.mock.nonces(this.sender)).to.equal(1n);
});
it("increments only sender's nonce", async function () {
const currentNonce = await this.mock.nonces(this.sender);
expect(currentNonce).to.equal(0n);
expect(await this.mock.nonces(this.other)).to.equal(0n);
await this.mock.$_useCheckedNonce(this.sender, currentNonce);
expect(await this.mock.nonces(this.sender)).to.equal(1n);
expect(await this.mock.nonces(this.other)).to.equal(0n);
});
it('reverts when nonce is not the expected', async function () {
const currentNonce = await this.mock.nonces(this.sender);
await expect(this.mock.$_useCheckedNonce(this.sender, currentNonce + 1n))
.to.be.revertedWithCustomError(this.mock, 'InvalidAccountNonce')
.withArgs(this.sender, currentNonce);
});
});
shouldBehaveLikeNonces();
});