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

const { toEthSignedMessageHash } = require('../../helpers/sign');

const { expect } = require('chai');

const SignatureChecker = artifacts.require('$SignatureChecker');
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 SignatureChecker.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 () {
    for (const signature of ['isValidERC1271SignatureNow', 'isValidSignatureNow']) {
      context(signature, function () {
        it('with matching signer and signature', async function () {
          expect(
            await this.signaturechecker[`$${signature}`](
              this.wallet.address,
              toEthSignedMessageHash(TEST_MESSAGE),
              this.signature,
            ),
          ).to.equal(true);
        });

        it('with invalid signer', async function () {
          expect(
            await this.signaturechecker[`$${signature}`](
              this.signaturechecker.address,
              toEthSignedMessageHash(TEST_MESSAGE),
              this.signature,
            ),
          ).to.equal(false);
        });

        it('with invalid signature', async function () {
          expect(
            await this.signaturechecker[`$${signature}`](
              this.wallet.address,
              toEthSignedMessageHash(WRONG_MESSAGE),
              this.signature,
            ),
          ).to.equal(false);
        });

        it('with malicious wallet', async function () {
          expect(
            await this.signaturechecker[`$${signature}`](
              this.malicious.address,
              toEthSignedMessageHash(TEST_MESSAGE),
              this.signature,
            ),
          ).to.equal(false);
        });
      });
    }
  });
});