openzeppelin-contracts/test/utils/cryptography/SignatureChecker.test.js

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);
    });
  });
});