status-im
/
dappconnect-voting-sdk
mirror of https://github.com/status-im/dappconnect-voting-sdk.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

Stop voting with 0 tokens (#65)

This commit is contained in:
Szymon Szlachtowicz 2021-09-14 09:04:39 +02:00 committed by GitHub
parent 26cb1823d9
commit 8575e7de55
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
packages/contracts/contracts/VotingContract.sol
View File

@ -79,6 +79,7 @@ contract VotingContract {
string calldata description,
uint256 voteAmount
) public {
require(voteAmount > 0, 'token amount must not be 0');
require(token.balanceOf(msg.sender) >= voteAmount, 'not enough token');
VotingRoom memory newVotingRoom;
newVotingRoom.startBlock = block.number;

20
packages/contracts/test/1.votingContract.test.ts
View File

@ -81,12 +81,12 @@ const getSignedMessages = async (
return { messages, signedMessages }
}
async function fixture([alice, firstAddress, secondAddress]: any[], provider: JsonRpcProvider) {
async function fixture([alice, firstAddress, secondAddress, noTokensAddress]: any[], provider: JsonRpcProvider) {
const erc20 = await deployContract(alice, ERC20Mock, ['MSNT', 'Mock SNT', alice.address, 100000])
await erc20.transfer(firstAddress.address, 10000)
await erc20.transfer(secondAddress.address, 10000)
const contract = await deployContract(alice, VotingContract, [erc20.address, 1000])
return { contract, alice, firstAddress, secondAddress, provider }
return { contract, alice, firstAddress, secondAddress, provider, noTokensAddress }
}
before(async function () {
@ -116,6 +116,22 @@ describe('Contract', () => {
contract.initializeVotingRoom('test', 'short desc', BigNumber.from(10000000000000))
).to.be.revertedWith('not enough token')
})
it('no tokens address', async () => {
const { contract, noTokensAddress } = await loadFixture(fixture)
const noTokensContract = contract.connect(noTokensAddress)
await expect(
noTokensContract.initializeVotingRoom('test', 'short desc', BigNumber.from(10))
).to.be.revertedWith('not enough token')
})
it("can't start voting with 0 tokens", async () => {
const { contract, noTokensAddress } = await loadFixture(fixture)
const noTokensContract = contract.connect(noTokensAddress)
await expect(noTokensContract.initializeVotingRoom('test', 'short desc', BigNumber.from(0))).to.be.revertedWith(
'token amount must not be 0'
)
})
})
it('gets', async () => {