mirrors/tendermint
1
0
Fork 0
mirror of https://github.com/tendermint/tendermint.git synced 2026-01-09 14:43:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix addrbook tests (#4526)

Browse Source 
Co-authored-by: Anton Kaliaev <anton.kalyaev@gmail.com>
Co-authored-by: Erik Grinaker <erik@grinaker.org>
This commit is contained in:
mergify[bot]
2020-03-05 09:29:26 +00:00
committed by GitHub
parent 4c8e3c8145
commit 632b53d653
1 changed files with 11 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
p2p/pex/addrbook_test.go
View File

@@ -60,15 +60,15 @@ func TestAddrBookSaveLoad(t *testing.T) {
defer deleteTempFile(fname)
// 0 addresses
book := NewAddrBook(fname, true).(*addrBook)
book := NewAddrBook(fname, true)
book.SetLogger(log.TestingLogger())
book.saveToFile(fname)
book.Save()
book = NewAddrBook(fname, true).(*addrBook)
book = NewAddrBook(fname, true)
book.SetLogger(log.TestingLogger())
book.loadFromFile(fname)
book.Start()
assert.Zero(t, book.Size())
assert.True(t, book.Empty())
// 100 addresses
randAddrs := randNetAddressPairs(t, 100)
@@ -78,11 +78,11 @@ func TestAddrBookSaveLoad(t *testing.T) {
}
assert.Equal(t, 100, book.Size())
book.saveToFile(fname)
book.Save()
book = NewAddrBook(fname, true).(*addrBook)
book = NewAddrBook(fname, true)
book.SetLogger(log.TestingLogger())
book.loadFromFile(fname)
book.Start()
assert.Equal(t, 100, book.Size())
}
@@ -93,19 +93,15 @@ func TestAddrBookLookup(t *testing.T) {
randAddrs := randNetAddressPairs(t, 100)
book := NewAddrBook(fname, true).(*addrBook)
book := NewAddrBook(fname, true)
book.SetLogger(log.TestingLogger())
for _, addrSrc := range randAddrs {
addr := addrSrc.addr
src := addrSrc.src
book.AddAddress(addr, src)
ka := book.addrLookup[addr.ID]
assert.NotNil(t, ka, "Expected to find KnownAddress %v but wasn't there.", addr)
if !(ka.Addr.Equals(addr) && ka.Src.Equals(src)) {
t.Fatalf("KnownAddress doesn't match addr & src")
}
ka := book.HasAddress(addr)
assert.True(t, ka, "Expected to find KnownAddress %v but wasn't there.", addr)
}
}