@@ -255,14 +255,10 @@ fn commission_split(commission: u8, on: u64) -> (u64, u64, bool) {
255255#[ cfg( test) ]
256256mod tests {
257257 use {
258- self :: points:: null_tracer,
259- super :: * ,
260- solana_program:: stake:: state:: Delegation ,
261- solana_pubkey:: Pubkey ,
262- solana_sdk:: native_token:: sol_to_lamports,
263- solana_vote:: alpenglow:: { accounting:: EpochCredit , state:: VoteState as AlpenglowVoteState } ,
264- solana_vote_program:: vote_state:: VoteState ,
265- test_case:: test_case,
258+ self :: points:: null_tracer, super :: * , solana_program:: stake:: state:: Delegation ,
259+ solana_pubkey:: Pubkey , solana_sdk:: native_token:: sol_to_lamports,
260+ solana_vote:: alpenglow:: state:: VoteState as AlpenglowVoteState ,
261+ solana_vote_program:: vote_state:: VoteState , test_case:: test_case,
266262 } ;
267263
268264 fn new_stake (
@@ -277,21 +273,15 @@ mod tests {
277273 }
278274 }
279275
280- #[ test_case( true ; "alpenglow" ) ]
281- #[ test_case( false ; "towerbft" ) ]
282- fn test_stake_state_redeem_rewards ( is_alpenglow : bool ) {
276+ #[ test]
277+ fn test_stake_state_redeem_rewards ( ) {
283278 let mut vote_state = VoteState :: default ( ) ;
284- let mut alpenglow_vote_state = AlpenglowVoteState :: default ( ) ;
285279 // assume stake.stake() is right
286280 // bootstrap means fully-vested stake at epoch 0
287281 let stake_lamports = 1 ;
288282 let mut stake = new_stake ( stake_lamports, & Pubkey :: default ( ) , 0 , u64:: MAX ) ;
289283
290- let vote_account = if is_alpenglow {
291- VoteAccount :: new_from_alpenglow_vote_state ( & alpenglow_vote_state)
292- } else {
293- VoteAccount :: new_from_vote_state ( & vote_state)
294- } ;
284+ let vote_account = { VoteAccount :: new_from_vote_state ( & vote_state) } ;
295285 // this one can't collect now, credits_observed == vote_state.credits()
296286 assert_eq ! (
297287 None ,
@@ -310,10 +300,7 @@ mod tests {
310300 ) ;
311301
312302 // put 2 credits in at epoch 0
313- let vote_account = if is_alpenglow {
314- alpenglow_vote_state. set_epoch_credits ( EpochCredit :: new ( 0 , 2 , 0 ) ) ;
315- VoteAccount :: new_from_alpenglow_vote_state ( & alpenglow_vote_state)
316- } else {
303+ let vote_account = {
317304 vote_state. increment_credits ( 0 , 1 ) ;
318305 vote_state. increment_credits ( 0 , 1 ) ;
319306 VoteAccount :: new_from_vote_state ( & vote_state)
@@ -343,21 +330,15 @@ mod tests {
343330 assert_eq ! ( stake. credits_observed, 2 ) ;
344331 }
345332
346- //TODO(wen): the alpenglow test cases don't work yet because we only keep credits for one Epoch
347- #[ test_case( false ; "towerbft" ) ]
348- fn test_stake_state_calculate_rewards ( is_alpenglow : bool ) {
333+ #[ test]
334+ fn test_stake_state_calculate_rewards ( ) {
349335 let mut vote_state = VoteState :: default ( ) ;
350- let mut alpenglow_vote_state = AlpenglowVoteState :: default ( ) ;
351336
352337 // assume stake.stake() is right
353338 // bootstrap means fully-vested stake at epoch 0
354339 let mut stake = new_stake ( 1 , & Pubkey :: default ( ) , 0 , u64:: MAX ) ;
355340
356- let vote_account = if is_alpenglow {
357- VoteAccount :: new_from_alpenglow_vote_state ( & alpenglow_vote_state)
358- } else {
359- VoteAccount :: new_from_vote_state ( & vote_state)
360- } ;
341+ let vote_account = VoteAccount :: new_from_vote_state ( & vote_state) ;
361342
362343 // this one can't collect now, credits_observed == vote_state.credits()
363344 assert_eq ! (
@@ -377,10 +358,7 @@ mod tests {
377358 ) ;
378359
379360 // put 2 credits in at epoch 0
380- let vote_account = if is_alpenglow {
381- alpenglow_vote_state. set_epoch_credits ( EpochCredit :: new ( 0 , 2 , 0 ) ) ;
382- VoteAccount :: new_from_alpenglow_vote_state ( & alpenglow_vote_state)
383- } else {
361+ let vote_account = {
384362 vote_state. increment_credits ( 0 , 1 ) ;
385363 vote_state. increment_credits ( 0 , 1 ) ;
386364 VoteAccount :: new_from_vote_state ( & vote_state)
@@ -430,10 +408,7 @@ mod tests {
430408 ) ;
431409
432410 // put 1 credit in epoch 1
433- let vote_account = if is_alpenglow {
434- alpenglow_vote_state. set_epoch_credits ( EpochCredit :: new ( 1 , 1 , 2 ) ) ;
435- VoteAccount :: new_from_alpenglow_vote_state ( & alpenglow_vote_state)
436- } else {
411+ let vote_account = {
437412 vote_state. increment_credits ( 1 , 1 ) ;
438413 VoteAccount :: new_from_vote_state ( & vote_state)
439414 } ;
@@ -461,10 +436,7 @@ mod tests {
461436 ) ;
462437
463438 // put 1 credit in epoch 2
464- let vote_account = if is_alpenglow {
465- alpenglow_vote_state. set_epoch_credits ( EpochCredit :: new ( 2 , 1 , 1 ) ) ;
466- VoteAccount :: new_from_alpenglow_vote_state ( & alpenglow_vote_state)
467- } else {
439+ let vote_account = {
468440 vote_state. increment_credits ( 2 , 1 ) ;
469441 VoteAccount :: new_from_vote_state ( & vote_state)
470442 } ;
@@ -516,10 +488,7 @@ mod tests {
516488
517489 // same as above, but is a really small commission out of 32 bits,
518490 // verify that None comes back on small redemptions where no one gets paid
519- let vote_account = if is_alpenglow {
520- alpenglow_vote_state. set_commission ( 1 ) ;
521- VoteAccount :: new_from_alpenglow_vote_state ( & alpenglow_vote_state)
522- } else {
491+ let vote_account = {
523492 vote_state. commission = 1 ;
524493 VoteAccount :: new_from_vote_state ( & vote_state)
525494 } ;
@@ -538,10 +507,7 @@ mod tests {
538507 None ,
539508 )
540509 ) ;
541- let vote_account = if is_alpenglow {
542- alpenglow_vote_state. set_commission ( 99 ) ;
543- VoteAccount :: new_from_alpenglow_vote_state ( & alpenglow_vote_state)
544- } else {
510+ let vote_account = {
545511 vote_state. commission = 99 ;
546512 VoteAccount :: new_from_vote_state ( & vote_state)
547513 } ;
@@ -658,10 +624,7 @@ mod tests {
658624 ) ;
659625
660626 // get rewards and credits observed when not the activation epoch
661- let vote_account = if is_alpenglow {
662- alpenglow_vote_state. set_commission ( 0 ) ;
663- VoteAccount :: new_from_alpenglow_vote_state ( & alpenglow_vote_state)
664- } else {
627+ let vote_account = {
665628 vote_state. commission = 0 ;
666629 VoteAccount :: new_from_vote_state ( & vote_state)
667630 } ;
0 commit comments