@@ -29,7 +29,6 @@ import (
29
29
"github.com/ava-labs/avalanchego/utils/math/meter"
30
30
"github.com/ava-labs/avalanchego/utils/resource"
31
31
"github.com/ava-labs/avalanchego/utils/set"
32
- "github.com/ava-labs/avalanchego/utils/timer/mockable"
33
32
"github.com/ava-labs/avalanchego/version"
34
33
)
35
34
@@ -591,14 +590,9 @@ func TestShouldDisconnect(t *testing.T) {
591
590
expectedShouldDisconnect : false ,
592
591
},
593
592
{
594
- name : "past durango without a signature" ,
593
+ name : "peer without signature" ,
595
594
initialPeer : & peer {
596
595
Config : & Config {
597
- Clock : func () mockable.Clock {
598
- clk := mockable.Clock {}
599
- clk .Set (mockable .MaxTime )
600
- return clk
601
- }(),
602
596
Log : logging.NoLog {},
603
597
VersionCompatibility : version .GetCompatibility (constants .UnitTestID ),
604
598
Validators : func () validators.Manager {
@@ -619,11 +613,6 @@ func TestShouldDisconnect(t *testing.T) {
619
613
},
620
614
expectedPeer : & peer {
621
615
Config : & Config {
622
- Clock : func () mockable.Clock {
623
- clk := mockable.Clock {}
624
- clk .Set (mockable .MaxTime )
625
- return clk
626
- }(),
627
616
Log : logging.NoLog {},
628
617
VersionCompatibility : version .GetCompatibility (constants .UnitTestID ),
629
618
Validators : func () validators.Manager {
@@ -645,68 +634,9 @@ func TestShouldDisconnect(t *testing.T) {
645
634
expectedShouldDisconnect : true ,
646
635
},
647
636
{
648
- name : "pre durango without a signature" ,
637
+ name : "peer with invalid signature" ,
649
638
initialPeer : & peer {
650
639
Config : & Config {
651
- Clock : func () mockable.Clock {
652
- clk := mockable.Clock {}
653
- clk .Set (time.Time {})
654
- return clk
655
- }(),
656
- Log : logging.NoLog {},
657
- VersionCompatibility : version .GetCompatibility (constants .UnitTestID ),
658
- Validators : func () validators.Manager {
659
- vdrs := validators .NewManager ()
660
- require .NoError (t , vdrs .AddStaker (
661
- constants .PrimaryNetworkID ,
662
- peerID ,
663
- bls .PublicFromSecretKey (blsKey ),
664
- txID ,
665
- 1 ,
666
- ))
667
- return vdrs
668
- }(),
669
- },
670
- id : peerID ,
671
- version : version .CurrentApp ,
672
- ip : & SignedIP {},
673
- },
674
- expectedPeer : & peer {
675
- Config : & Config {
676
- Clock : func () mockable.Clock {
677
- clk := mockable.Clock {}
678
- clk .Set (time.Time {})
679
- return clk
680
- }(),
681
- Log : logging.NoLog {},
682
- VersionCompatibility : version .GetCompatibility (constants .UnitTestID ),
683
- Validators : func () validators.Manager {
684
- vdrs := validators .NewManager ()
685
- require .NoError (t , vdrs .AddStaker (
686
- constants .PrimaryNetworkID ,
687
- peerID ,
688
- bls .PublicFromSecretKey (blsKey ),
689
- txID ,
690
- 1 ,
691
- ))
692
- return vdrs
693
- }(),
694
- },
695
- id : peerID ,
696
- version : version .CurrentApp ,
697
- ip : & SignedIP {},
698
- },
699
- expectedShouldDisconnect : false ,
700
- },
701
- {
702
- name : "pre durango with an invalid signature" ,
703
- initialPeer : & peer {
704
- Config : & Config {
705
- Clock : func () mockable.Clock {
706
- clk := mockable.Clock {}
707
- clk .Set (time.Time {})
708
- return clk
709
- }(),
710
640
Log : logging.NoLog {},
711
641
VersionCompatibility : version .GetCompatibility (constants .UnitTestID ),
712
642
Validators : func () validators.Manager {
@@ -729,11 +659,6 @@ func TestShouldDisconnect(t *testing.T) {
729
659
},
730
660
expectedPeer : & peer {
731
661
Config : & Config {
732
- Clock : func () mockable.Clock {
733
- clk := mockable.Clock {}
734
- clk .Set (time.Time {})
735
- return clk
736
- }(),
737
662
Log : logging.NoLog {},
738
663
VersionCompatibility : version .GetCompatibility (constants .UnitTestID ),
739
664
Validators : func () validators.Manager {
@@ -757,14 +682,9 @@ func TestShouldDisconnect(t *testing.T) {
757
682
expectedShouldDisconnect : true ,
758
683
},
759
684
{
760
- name : "pre durango with a valid signature" ,
685
+ name : "peer with valid signature" ,
761
686
initialPeer : & peer {
762
687
Config : & Config {
763
- Clock : func () mockable.Clock {
764
- clk := mockable.Clock {}
765
- clk .Set (time.Time {})
766
- return clk
767
- }(),
768
688
Log : logging.NoLog {},
769
689
VersionCompatibility : version .GetCompatibility (constants .UnitTestID ),
770
690
Validators : func () validators.Manager {
@@ -787,11 +707,6 @@ func TestShouldDisconnect(t *testing.T) {
787
707
},
788
708
expectedPeer : & peer {
789
709
Config : & Config {
790
- Clock : func () mockable.Clock {
791
- clk := mockable.Clock {}
792
- clk .Set (time.Time {})
793
- return clk
794
- }(),
795
710
Log : logging.NoLog {},
796
711
VersionCompatibility : version .GetCompatibility (constants .UnitTestID ),
797
712
Validators : func () validators.Manager {
0 commit comments