@@ -334,7 +334,7 @@ func TestProtocol_NoRewardAddr(t *testing.T) {
334
334
{
335
335
Address : identityset .Address (0 ).String (),
336
336
Votes : unit .ConvertIotxToRau (1000000 ),
337
- RewardAddress : "" ,
337
+ RewardAddress : identityset . Address ( 0 ). String () ,
338
338
},
339
339
{
340
340
Address : identityset .Address (1 ).String (),
@@ -437,28 +437,27 @@ func TestProtocol_NoRewardAddr(t *testing.T) {
437
437
require .NoError (t , err )
438
438
439
439
// Grant block reward
440
- rewardLog , err : = p .GrantBlockReward (ctx , sm )
440
+ _ , err = p .GrantBlockReward (ctx , sm )
441
441
require .NoError (t , err )
442
- require .Nil (t , rewardLog )
443
442
444
443
availableBalance , _ , err := p .AvailableBalance (ctx , sm )
445
444
require .NoError (t , err )
446
- assert .Equal (t , big .NewInt (200 ), availableBalance )
445
+ assert .Equal (t , big .NewInt (190 ), availableBalance )
447
446
unclaimedBalance , _ , err := p .UnclaimedBalance (ctx , sm , identityset .Address (0 ))
448
447
require .NoError (t , err )
449
- assert .Equal (t , big .NewInt (0 ), unclaimedBalance )
448
+ assert .Equal (t , big .NewInt (10 ), unclaimedBalance )
450
449
451
450
// Grant epoch reward
452
451
rewardLogs , err := p .GrantEpochReward (ctx , sm )
453
452
require .NoError (t , err )
454
- require .Equal (t , 2 , len (rewardLogs ))
453
+ require .Equal (t , 4 , len (rewardLogs ))
455
454
456
455
availableBalance , _ , err = p .AvailableBalance (ctx , sm )
457
456
require .NoError (t , err )
458
- assert .Equal (t , big .NewInt (145 ), availableBalance )
457
+ assert .Equal (t , big .NewInt (80 ), availableBalance )
459
458
unclaimedBalance , _ , err = p .UnclaimedBalance (ctx , sm , identityset .Address (0 ))
460
459
require .NoError (t , err )
461
- assert .Equal (t , big .NewInt (0 ), unclaimedBalance )
460
+ assert .Equal (t , big .NewInt (65 ), unclaimedBalance )
462
461
// It doesn't affect others to get reward
463
462
unclaimedBalance , _ , err = p .UnclaimedBalance (ctx , sm , identityset .Address (1 ))
464
463
require .NoError (t , err )
@@ -467,11 +466,11 @@ func TestProtocol_NoRewardAddr(t *testing.T) {
467
466
var rl rewardingpb.RewardLog
468
467
require .NoError (t , proto .Unmarshal (rewardLogs [0 ].Data , & rl ))
469
468
assert .Equal (t , rewardingpb .RewardLog_EPOCH_REWARD , rl .Type )
470
- assert .Equal (t , identityset .Address (1 ).String (), rl .Addr )
469
+ assert .Equal (t , identityset .Address (0 ).String (), rl .Addr )
471
470
assert .Equal (t , "50" , rl .Amount )
472
471
require .Equal (t , p .addr .String (), rewardLogs [1 ].Address )
473
472
require .NoError (t , proto .Unmarshal (rewardLogs [1 ].Data , & rl ))
474
- assert .Equal (t , rewardingpb .RewardLog_FOUNDATION_BONUS , rl .Type )
473
+ assert .Equal (t , rewardingpb .RewardLog_EPOCH_REWARD , rl .Type )
475
474
assert .Equal (t , identityset .Address (1 ).String (), rl .Addr )
476
- assert .Equal (t , "5 " , rl .Amount )
475
+ assert .Equal (t , "50 " , rl .Amount )
477
476
}
0 commit comments