@@ -143,8 +143,13 @@ describe("AcceleratingDistributor: Admin Functions", async function () {
143
143
} ) ;
144
144
145
145
it ( "Permissioning on staking-related methods" , async function ( ) {
146
- await expect ( distributor . connect ( owner ) . stake ( lpToken1 . address , 0 ) ) . to . be . revertedWith ( "stakedToken not enabled" ) ;
147
- await expect ( distributor . connect ( owner ) . unstake ( lpToken1 . address , 0 ) ) . to . be . revertedWith (
146
+ await lpToken1 . mint ( owner . address , toWei ( 69 ) ) ;
147
+ await lpToken1 . connect ( owner ) . approve ( distributor . address , toWei ( 69 ) ) ;
148
+
149
+ await expect ( distributor . connect ( owner ) . stake ( lpToken1 . address , toWei ( 1 ) ) ) . to . be . revertedWith (
150
+ "stakedToken not enabled"
151
+ ) ;
152
+ await expect ( distributor . connect ( owner ) . unstake ( lpToken1 . address , toWei ( 1 ) ) ) . to . be . revertedWith (
148
153
"stakedToken not initialized"
149
154
) ;
150
155
await expect ( distributor . connect ( owner ) . withdrawReward ( lpToken1 . address ) ) . to . be . revertedWith (
@@ -160,11 +165,13 @@ describe("AcceleratingDistributor: Admin Functions", async function () {
160
165
secondsToMaxMultiplier
161
166
) ;
162
167
163
- await expect ( distributor . connect ( owner ) . stake ( lpToken1 . address , 0 ) ) . to . not . be . reverted ;
164
- await expect ( distributor . connect ( owner ) . unstake ( lpToken1 . address , 0 ) ) . to . not . be . reverted ;
168
+ await expect ( distributor . connect ( owner ) . stake ( lpToken1 . address , toWei ( 2 ) ) ) . to . not . be . reverted ;
169
+ await expect ( distributor . connect ( owner ) . unstake ( lpToken1 . address , toWei ( 1 ) ) ) . to . not . be . reverted ;
165
170
await expect ( distributor . connect ( owner ) . withdrawReward ( lpToken1 . address ) ) . to . not . be . reverted ;
166
171
await expect ( distributor . connect ( owner ) . exit ( lpToken1 . address ) ) . to . not . be . reverted ;
167
172
173
+ // Balance => non-zero before disabling, to verify that unstake/withdraw/exit is still possible.
174
+ await expect ( distributor . connect ( owner ) . stake ( lpToken1 . address , toWei ( 2 ) ) ) . to . not . be . reverted ;
168
175
await distributor . configureStakingToken (
169
176
lpToken1 . address ,
170
177
false ,
@@ -173,8 +180,10 @@ describe("AcceleratingDistributor: Admin Functions", async function () {
173
180
secondsToMaxMultiplier
174
181
) ;
175
182
176
- await expect ( distributor . connect ( owner ) . stake ( lpToken1 . address , 0 ) ) . to . be . revertedWith ( "stakedToken not enabled" ) ;
177
- await expect ( distributor . connect ( owner ) . unstake ( lpToken1 . address , 0 ) ) . to . not . be . reverted ;
183
+ await expect ( distributor . connect ( owner ) . stake ( lpToken1 . address , toWei ( 1 ) ) ) . to . be . revertedWith (
184
+ "stakedToken not enabled"
185
+ ) ;
186
+ await expect ( distributor . connect ( owner ) . unstake ( lpToken1 . address , toWei ( 1 ) ) ) . to . not . be . reverted ;
178
187
await expect ( distributor . connect ( owner ) . withdrawReward ( lpToken1 . address ) ) . to . not . be . reverted ;
179
188
await expect ( distributor . connect ( owner ) . exit ( lpToken1 . address ) ) . to . not . be . reverted ;
180
189
} ) ;
0 commit comments