From 30add1b64b28cde27556fb56cc07bf40f8fa42a6 Mon Sep 17 00:00:00 2001 From: Ethan Buchman Date: Tue, 29 Jan 2019 17:07:51 -0800 Subject: [PATCH] failing test --- p2p/pex/addrbook_test.go | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/p2p/pex/addrbook_test.go b/p2p/pex/addrbook_test.go index ade02d490..b173816d2 100644 --- a/p2p/pex/addrbook_test.go +++ b/p2p/pex/addrbook_test.go @@ -36,6 +36,28 @@ func deleteTempFile(fname string) { } } +func TestAddrBookGetSelectionWithMarkedGood(t *testing.T) { + fname := createTempFileName("addrbook_test") + defer deleteTempFile(fname) + + // 0 addresses + book := NewAddrBook(fname, true) + book.SetLogger(log.TestingLogger()) + assert.Zero(t, book.Size()) + + N := 10 + randAddrs := randNetAddressPairs(t, N) + for _, addr := range randAddrs { + book.AddAddress(addr.addr, addr.src) + } + for _, addr := range randAddrs[:1] { + book.MarkGood(addr.addr) + } + + addrs := book.GetSelectionWithBias(10) + assert.NotNil(t, addrs, "expected an address") +} + func TestAddrBookPickAddress(t *testing.T) { fname := createTempFileName("addrbook_test") defer deleteTempFile(fname)