const { expectRevert, BN } = require('@openzeppelin/test-helpers'); const { expect } = require('chai'); const { getChainId } = require('../../helpers/chainid'); const { shouldBehaveLikeVotes } = require('./Votes.behavior'); const Votes = artifacts.require('$VotesMock'); contract('Votes', function (accounts) { const [account1, account2, account3] = accounts; beforeEach(async function () { this.name = 'My Vote'; this.votes = await Votes.new(this.name, '1'); }); it('starts with zero votes', async function () { expect(await this.votes.getTotalSupply()).to.be.bignumber.equal('0'); }); describe('performs voting operations', function () { beforeEach(async function () { this.tx1 = await this.votes.$_mint(account1, 1); this.tx2 = await this.votes.$_mint(account2, 1); this.tx3 = await this.votes.$_mint(account3, 1); }); it('reverts if block number >= current block', async function () { await expectRevert(this.votes.getPastTotalSupply(this.tx3.receipt.blockNumber + 1), 'Votes: block not yet mined'); }); it('delegates', async function () { await this.votes.delegate(account3, account2); expect(await this.votes.delegates(account3)).to.be.equal(account2); }); it('returns total amount of votes', async function () { expect(await this.votes.getTotalSupply()).to.be.bignumber.equal('3'); }); }); describe('performs voting workflow', function () { beforeEach(async function () { this.chainId = await getChainId(); this.account1 = account1; this.account2 = account2; this.account1Delegatee = account2; this.NFT0 = new BN('10000000000000000000000000'); this.NFT1 = new BN('10'); this.NFT2 = new BN('20'); this.NFT3 = new BN('30'); }); shouldBehaveLikeVotes(); }); });