openzeppelin-contracts/test/governance/extensions/GovernorERC721.test.js

const { BN, expectEvent } = require('@openzeppelin/test-helpers');
const { expect } = require('chai');
const Enums = require('../../helpers/enums');
const { GovernorHelper } = require('../../helpers/governance');

const Token = artifacts.require('$ERC721Votes');
const Governor = artifacts.require('$GovernorVoteMocks');
const CallReceiver = artifacts.require('CallReceiverMock');

contract('GovernorERC721', function (accounts) {
  const [owner, voter1, voter2, voter3, voter4] = accounts;

  const name = 'OZ-Governor';
  // const version = '1';
  const tokenName = 'MockNFToken';
  const tokenSymbol = 'MTKN';
  const NFT0 = new BN(0);
  const NFT1 = new BN(1);
  const NFT2 = new BN(2);
  const NFT3 = new BN(3);
  const NFT4 = new BN(4);
  const votingDelay = new BN(4);
  const votingPeriod = new BN(16);
  const value = web3.utils.toWei('1');

  beforeEach(async function () {
    this.owner = owner;
    this.token = await Token.new(tokenName, tokenSymbol, tokenName, '1');
    this.mock = await Governor.new(name, this.token.address);
    this.receiver = await CallReceiver.new();

    this.helper = new GovernorHelper(this.mock);

    await web3.eth.sendTransaction({ from: owner, to: this.mock.address, value });

    await Promise.all([NFT0, NFT1, NFT2, NFT3, NFT4].map(tokenId => this.token.$_mint(owner, tokenId)));
    await this.helper.delegate({ token: this.token, to: voter1, tokenId: NFT0 }, { from: owner });
    await this.helper.delegate({ token: this.token, to: voter2, tokenId: NFT1 }, { from: owner });
    await this.helper.delegate({ token: this.token, to: voter2, tokenId: NFT2 }, { from: owner });
    await this.helper.delegate({ token: this.token, to: voter3, tokenId: NFT3 }, { from: owner });
    await this.helper.delegate({ token: this.token, to: voter4, tokenId: NFT4 }, { from: owner });

    // default proposal
    this.proposal = this.helper.setProposal(
      [
        {
          target: this.receiver.address,
          value,
          data: this.receiver.contract.methods.mockFunction().encodeABI(),
        },
      ],
      '<proposal description>',
    );
  });

  it('deployment check', async function () {
    expect(await this.mock.name()).to.be.equal(name);
    expect(await this.mock.token()).to.be.equal(this.token.address);
    expect(await this.mock.votingDelay()).to.be.bignumber.equal(votingDelay);
    expect(await this.mock.votingPeriod()).to.be.bignumber.equal(votingPeriod);
    expect(await this.mock.quorum(0)).to.be.bignumber.equal('0');
  });

  it('voting with ERC721 token', async function () {
    await this.helper.propose();
    await this.helper.waitForSnapshot();

    expectEvent(await this.helper.vote({ support: Enums.VoteType.For }, { from: voter1 }), 'VoteCast', {
      voter: voter1,
      support: Enums.VoteType.For,
      weight: '1',
    });

    expectEvent(await this.helper.vote({ support: Enums.VoteType.For }, { from: voter2 }), 'VoteCast', {
      voter: voter2,
      support: Enums.VoteType.For,
      weight: '2',
    });

    expectEvent(await this.helper.vote({ support: Enums.VoteType.Against }, { from: voter3 }), 'VoteCast', {
      voter: voter3,
      support: Enums.VoteType.Against,
      weight: '1',
    });

    expectEvent(await this.helper.vote({ support: Enums.VoteType.Abstain }, { from: voter4 }), 'VoteCast', {
      voter: voter4,
      support: Enums.VoteType.Abstain,
      weight: '1',
    });

    await this.helper.waitForDeadline();
    await this.helper.execute();

    expect(await this.mock.hasVoted(this.proposal.id, owner)).to.be.equal(false);
    expect(await this.mock.hasVoted(this.proposal.id, voter1)).to.be.equal(true);
    expect(await this.mock.hasVoted(this.proposal.id, voter2)).to.be.equal(true);
    expect(await this.mock.hasVoted(this.proposal.id, voter3)).to.be.equal(true);
    expect(await this.mock.hasVoted(this.proposal.id, voter4)).to.be.equal(true);

    await this.mock.proposalVotes(this.proposal.id).then(results => {
      expect(results.forVotes).to.be.bignumber.equal('3');
      expect(results.againstVotes).to.be.bignumber.equal('1');
      expect(results.abstainVotes).to.be.bignumber.equal('1');
    });
  });
});