add test for unstaking with zero

airswap · Jan 25, 2024 · 927e820 · 927e820
1 parent ba862d3
commit 927e820
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/source/staking/test/Staking.js b/source/staking/test/Staking.js
@@ -332,6 +332,15 @@ describe('Staking Unit', () => {
       expect(userStake.balance).to.equal(0)
     })
 
+    it('fails attempting to unstake zero', async () => {
+      await token.mock.transferFrom.returns(true)
+      await token.mock.transfer.returns(true)
+      await staking.connect(account1).stake('100')
+      await expect(staking.connect(account1).unstake('0'))
+        .to.be.revertedWith('AmountInvalid')
+        .withArgs(0)
+    })
+
     it('successful partial unstaking', async () => {
       await token.mock.transferFrom.returns(true)
       await token.mock.transfer.returns(true)