82 lines
2.6 KiB
JavaScript
82 lines
2.6 KiB
JavaScript
const { toEthSignedMessageHash } = require('../../helpers/sign');
|
|
|
|
const { expect } = require('chai');
|
|
|
|
const SignatureCheckerMock = artifacts.require('SignatureCheckerMock');
|
|
const ERC1271WalletMock = artifacts.require('ERC1271WalletMock');
|
|
const ERC1271MaliciousMock = artifacts.require('ERC1271MaliciousMock');
|
|
|
|
const TEST_MESSAGE = web3.utils.sha3('OpenZeppelin');
|
|
const WRONG_MESSAGE = web3.utils.sha3('Nope');
|
|
|
|
contract('SignatureChecker (ERC1271)', function (accounts) {
|
|
const [signer, other] = accounts;
|
|
|
|
before('deploying', async function () {
|
|
this.signaturechecker = await SignatureCheckerMock.new();
|
|
this.wallet = await ERC1271WalletMock.new(signer);
|
|
this.malicious = await ERC1271MaliciousMock.new();
|
|
this.signature = await web3.eth.sign(TEST_MESSAGE, signer);
|
|
});
|
|
|
|
context('EOA account', function () {
|
|
it('with matching signer and signature', async function () {
|
|
expect(await this.signaturechecker.isValidSignatureNow(
|
|
signer,
|
|
toEthSignedMessageHash(TEST_MESSAGE),
|
|
this.signature,
|
|
)).to.equal(true);
|
|
});
|
|
|
|
it('with invalid signer', async function () {
|
|
expect(await this.signaturechecker.isValidSignatureNow(
|
|
other,
|
|
toEthSignedMessageHash(TEST_MESSAGE),
|
|
this.signature,
|
|
)).to.equal(false);
|
|
});
|
|
|
|
it('with invalid signature', async function () {
|
|
expect(await this.signaturechecker.isValidSignatureNow(
|
|
signer,
|
|
toEthSignedMessageHash(WRONG_MESSAGE),
|
|
this.signature,
|
|
)).to.equal(false);
|
|
});
|
|
});
|
|
|
|
context('ERC1271 wallet', function () {
|
|
it('with matching signer and signature', async function () {
|
|
expect(await this.signaturechecker.isValidSignatureNow(
|
|
this.wallet.address,
|
|
toEthSignedMessageHash(TEST_MESSAGE),
|
|
this.signature,
|
|
)).to.equal(true);
|
|
});
|
|
|
|
it('with invalid signer', async function () {
|
|
expect(await this.signaturechecker.isValidSignatureNow(
|
|
this.signaturechecker.address,
|
|
toEthSignedMessageHash(TEST_MESSAGE),
|
|
this.signature,
|
|
)).to.equal(false);
|
|
});
|
|
|
|
it('with invalid signature', async function () {
|
|
expect(await this.signaturechecker.isValidSignatureNow(
|
|
this.wallet.address,
|
|
toEthSignedMessageHash(WRONG_MESSAGE),
|
|
this.signature,
|
|
)).to.equal(false);
|
|
});
|
|
|
|
it('with malicious wallet', async function () {
|
|
expect(await this.signaturechecker.isValidSignatureNow(
|
|
this.malicious.address,
|
|
toEthSignedMessageHash(TEST_MESSAGE),
|
|
this.signature,
|
|
)).to.equal(false);
|
|
});
|
|
});
|
|
});
|