poanetwork
/
gecko
mirror of https://github.com/poanetwork/gecko.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge branch 'master' into test-ids-aliaser

This commit is contained in:
Stephen Buttolph 2020-04-06 14:23:51 -04:00 committed by GitHub
parent cf69425775 5d9c7e52d2
commit 7c69cf16bf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 25 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
snow/validators/set_test.go
View File

@ -10,6 +10,31 @@ import (
"github.com/ava-labs/gecko/ids"
)
func TestSetSet(t *testing.T) {
vdr0 := NewValidator(ids.ShortEmpty, 1)
vdr1_0 := NewValidator(ids.NewShortID([20]byte{0xFF}), 1)
// Should replace vdr1_0, because later additions replace earlier ones
vdr1_1 := NewValidator(ids.NewShortID([20]byte{0xFF}), math.MaxInt64-1)
// Should be discarded, because it has a weight of 0
vdr2 := NewValidator(ids.NewShortID([20]byte{0xAA}), 0)
s := NewSet()
s.Set([]Validator{vdr0, vdr1_0, vdr1_1, vdr2})
if !s.Contains(vdr0.ID()) {
t.Fatal("Should have contained vdr0", vdr0.ID())
}
if !s.Contains(vdr1_0.ID()) {
t.Fatal("Should have contained vdr1", vdr1_0.ID())
}
if sampled := s.Sample(1); !sampled[0].ID().Equals(vdr1_0.ID()) {
t.Fatal("Should have sampled vdr1")
}
if len := s.Len(); len != 2 {
t.Fatalf("Got size %d, expected 2", len)
}
}
func TestSamplerSample(t *testing.T) {
vdr0 := GenerateRandomValidator(1)
vdr1 := GenerateRandomValidator(math.MaxInt64 - 1)