@@ -356,8 +356,6 @@ func TestBootstrapperUnknownByzantineResponse(t *testing.T) {
356
356
357
357
require .NoError (bs .Start (context .Background (), 0 ))
358
358
359
- acceptedIDs := []ids.ID {blkID2 }
360
-
361
359
parsedBlk1 := false
362
360
vm .GetBlockF = func (_ context.Context , blkID ids.ID ) (snowman.Block , error ) {
363
361
switch blkID {
@@ -390,31 +388,29 @@ func TestBootstrapperUnknownByzantineResponse(t *testing.T) {
390
388
return nil , errUnknownBlock
391
389
}
392
390
393
- requestID := new ( uint32 )
391
+ var requestID uint32
394
392
sender .SendGetAncestorsF = func (_ context.Context , vdr ids.NodeID , reqID uint32 , blkID ids.ID ) {
395
393
require .Equal (peerID , vdr )
396
394
require .Equal (blkID1 , blkID )
397
- * requestID = reqID
395
+ requestID = reqID
398
396
}
399
397
400
398
vm .CantSetState = false
401
- require .NoError (bs .ForceAccepted (context .Background (), acceptedIDs )) // should request blk1
402
-
403
- oldReqID := * requestID
404
- require .NoError (bs .Ancestors (context .Background (), peerID , * requestID + 1 , [][]byte {blkBytes1 })) // respond with wrong request ID
405
- require .Equal (oldReqID , * requestID )
399
+ require .NoError (bs .ForceAccepted (context .Background (), []ids.ID {blkID2 })) // should request blk1
406
400
407
- require .NoError (bs .Ancestors (context .Background (), ids .BuildTestNodeID ([]byte {1 , 2 , 3 }), * requestID , [][]byte {blkBytes1 })) // respond from wrong peer
408
- require .Equal (oldReqID , * requestID )
401
+ oldReqID := requestID
402
+ require .NoError (bs .Ancestors (context .Background (), peerID , requestID , [][]byte {blkBytes0 })) // respond with wrong block
403
+ require .NotEqual (oldReqID , requestID )
409
404
410
- require .NoError (bs .Ancestors (context .Background (), peerID , * requestID , [][]byte {blkBytes0 })) // respond with wrong block
411
- require .NotEqual (oldReqID , * requestID )
405
+ require .NoError (bs .Ancestors (context .Background (), peerID , requestID , [][]byte {blkBytes1 }))
412
406
413
- require .NoError (bs .Ancestors (context .Background (), peerID , * requestID , [][]byte {blkBytes1 }))
414
- require .Equal (snow .NormalOp , config .Ctx .State .Get ().State )
407
+ require .Equal (snow .Bootstrapping , config .Ctx .State .Get ().State )
415
408
require .Equal (choices .Accepted , blk0 .Status ())
416
409
require .Equal (choices .Accepted , blk1 .Status ())
417
410
require .Equal (choices .Accepted , blk2 .Status ())
411
+
412
+ require .NoError (bs .ForceAccepted (context .Background (), []ids.ID {blkID2 }))
413
+ require .Equal (snow .NormalOp , config .Ctx .State .Get ().State )
418
414
}
419
415
420
416
// There are multiple needed blocks and Ancestors returns one at a time
@@ -554,10 +550,13 @@ func TestBootstrapperPartialFetch(t *testing.T) {
554
550
require .NoError (bs .Ancestors (context .Background (), peerID , * requestID , [][]byte {blkBytes1 })) // respond with blk1
555
551
require .Equal (blkID1 , requested )
556
552
557
- require .Equal (snow .NormalOp , config .Ctx .State .Get ().State )
553
+ require .Equal (snow .Bootstrapping , config .Ctx .State .Get ().State )
558
554
require .Equal (choices .Accepted , blk0 .Status ())
559
555
require .Equal (choices .Accepted , blk1 .Status ())
560
556
require .Equal (choices .Accepted , blk2 .Status ())
557
+
558
+ require .NoError (bs .ForceAccepted (context .Background (), acceptedIDs ))
559
+ require .Equal (snow .NormalOp , config .Ctx .State .Get ().State )
561
560
}
562
561
563
562
// There are multiple needed blocks and some validators do not have all the blocks
@@ -714,7 +713,7 @@ func TestBootstrapperEmptyResponse(t *testing.T) {
714
713
715
714
require .NoError (bs .Ancestors (context .Background (), requestedVdr , requestID , [][]byte {blkBytes1 })) // respond with blk1
716
715
717
- require .Equal (snow .NormalOp , config .Ctx .State .Get ().State )
716
+ require .Equal (snow .Bootstrapping , config .Ctx .State .Get ().State )
718
717
require .Equal (choices .Accepted , blk0 .Status ())
719
718
require .Equal (choices .Accepted , blk1 .Status ())
720
719
require .Equal (choices .Accepted , blk2 .Status ())
@@ -856,10 +855,13 @@ func TestBootstrapperAncestors(t *testing.T) {
856
855
require .NoError (bs .Ancestors (context .Background (), peerID , * requestID , [][]byte {blkBytes2 , blkBytes1 })) // respond with blk2 and blk1
857
856
require .Equal (blkID2 , requested )
858
857
859
- require .Equal (snow .NormalOp , config .Ctx .State .Get ().State )
858
+ require .Equal (snow .Bootstrapping , config .Ctx .State .Get ().State )
860
859
require .Equal (choices .Accepted , blk0 .Status ())
861
860
require .Equal (choices .Accepted , blk1 .Status ())
862
861
require .Equal (choices .Accepted , blk2 .Status ())
862
+
863
+ require .NoError (bs .ForceAccepted (context .Background (), acceptedIDs ))
864
+ require .Equal (snow .NormalOp , config .Ctx .State .Get ().State )
863
865
}
864
866
865
867
func TestBootstrapperFinalized (t * testing.T ) {
@@ -976,10 +978,13 @@ func TestBootstrapperFinalized(t *testing.T) {
976
978
977
979
require .NoError (bs .Ancestors (context .Background (), peerID , reqIDBlk2 , [][]byte {blkBytes2 , blkBytes1 }))
978
980
979
- require .Equal (snow .NormalOp , config .Ctx .State .Get ().State )
981
+ require .Equal (snow .Bootstrapping , config .Ctx .State .Get ().State )
980
982
require .Equal (choices .Accepted , blk0 .Status ())
981
983
require .Equal (choices .Accepted , blk1 .Status ())
982
984
require .Equal (choices .Accepted , blk2 .Status ())
985
+
986
+ require .NoError (bs .ForceAccepted (context .Background (), []ids.ID {blkID2 }))
987
+ require .Equal (snow .NormalOp , config .Ctx .State .Get ().State )
983
988
}
984
989
985
990
func TestRestartBootstrapping (t * testing.T ) {
@@ -1156,12 +1161,15 @@ func TestRestartBootstrapping(t *testing.T) {
1156
1161
1157
1162
require .NoError (bs .Ancestors (context .Background (), peerID , blk4RequestID , [][]byte {blkBytes4 }))
1158
1163
1159
- require .Equal (snow .NormalOp , config .Ctx .State .Get ().State )
1164
+ require .Equal (snow .Bootstrapping , config .Ctx .State .Get ().State )
1160
1165
require .Equal (choices .Accepted , blk0 .Status ())
1161
1166
require .Equal (choices .Accepted , blk1 .Status ())
1162
1167
require .Equal (choices .Accepted , blk2 .Status ())
1163
1168
require .Equal (choices .Accepted , blk3 .Status ())
1164
1169
require .Equal (choices .Accepted , blk4 .Status ())
1170
+
1171
+ require .NoError (bs .ForceAccepted (context .Background (), []ids.ID {blkID4 }))
1172
+ require .Equal (snow .NormalOp , config .Ctx .State .Get ().State )
1165
1173
}
1166
1174
1167
1175
func TestBootstrapOldBlockAfterStateSync (t * testing.T ) {
0 commit comments