50 lines
1.8 KiB
TypeScript
50 lines
1.8 KiB
TypeScript
|
import { expect } from 'chai'
|
||
|
import { TimedPollVoteMsg } from '../../src/models/TimedPollVoteMsg'
|
||
|
import { MockProvider } from 'ethereum-waffle'
|
||
|
import { BigNumber, utils } from 'ethers'
|
||
|
|
||
|
describe('TimedPollVoteMsg', () => {
|
||
|
const provider = new MockProvider()
|
||
|
const [alice] = provider.getWallets()
|
||
|
const pollId = '0x14c336ef626274f156d094fc1d7ffad2bbc83cccc9817598dd55e42a86b56b72'
|
||
|
it('success', async () => {
|
||
|
const poll = await TimedPollVoteMsg.create(alice, pollId, 0)
|
||
|
|
||
|
expect(poll).to.not.be.undefined
|
||
|
expect(poll.voter).to.eq(alice.address)
|
||
|
expect(poll.answer).to.eq(0)
|
||
|
expect(poll.id).to.be.eq(pollId)
|
||
|
expect(poll.tokenAmount).to.be.undefined
|
||
|
|
||
|
const msg: (string | number | BigNumber)[] = [poll.id, poll.voter, poll.timestamp, poll.answer]
|
||
|
const types = ['bytes32', 'address', 'uint256', 'uint64']
|
||
|
const packedData = utils.arrayify(utils.solidityPack(types, msg))
|
||
|
|
||
|
const verifiedAddress = utils.verifyMessage(packedData, poll.signature)
|
||
|
expect(verifiedAddress).to.eq(alice.address)
|
||
|
})
|
||
|
|
||
|
it('success token amount', async () => {
|
||
|
const poll = await TimedPollVoteMsg.create(alice, pollId, 1, BigNumber.from(100))
|
||
|
|
||
|
expect(poll).to.not.be.undefined
|
||
|
expect(poll.voter).to.eq(alice.address)
|
||
|
expect(poll.answer).to.eq(1)
|
||
|
expect(poll.id).to.be.eq(pollId)
|
||
|
expect(poll.tokenAmount).to.deep.eq(BigNumber.from(100))
|
||
|
|
||
|
const msg: (string | number | BigNumber | undefined)[] = [
|
||
|
poll.id,
|
||
|
poll.voter,
|
||
|
poll.timestamp,
|
||
|
poll.answer,
|
||
|
poll?.tokenAmount,
|
||
|
]
|
||
|
const types = ['bytes32', 'address', 'uint256', 'uint64', 'uint256']
|
||
|
const packedData = utils.arrayify(utils.solidityPack(types, msg))
|
||
|
|
||
|
const verifiedAddress = utils.verifyMessage(packedData, poll.signature)
|
||
|
expect(verifiedAddress).to.eq(alice.address)
|
||
|
})
|
||
|
})
|