mirror of
https://github.com/fluencelabs/tendermint
synced 2025-04-25 23:02:16 +00:00
162 lines
3.2 KiB
Go
162 lines
3.2 KiB
Go
|
package p2p
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"io/ioutil"
|
||
|
"math/rand"
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
func createTempFileName(prefix string) string {
|
||
|
f, err := ioutil.TempFile("", prefix)
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
fname := f.Name()
|
||
|
err = f.Close()
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
return fname
|
||
|
}
|
||
|
|
||
|
func TestEmpty(t *testing.T) {
|
||
|
fname := createTempFileName("addrbook_test")
|
||
|
// t.Logf("New tempfile name: %v", fname)
|
||
|
|
||
|
// Save an empty book & load it
|
||
|
book := NewAddrBook(fname)
|
||
|
book.saveToFile(fname)
|
||
|
|
||
|
book = NewAddrBook(fname)
|
||
|
book.loadFromFile(fname)
|
||
|
|
||
|
if book.Size() != 0 {
|
||
|
t.Errorf("Expected 0 addresses, found %v", book.Size())
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func randIPv4Address() *NetAddress {
|
||
|
for {
|
||
|
ip := fmt.Sprintf("%v.%v.%v.%v",
|
||
|
rand.Intn(254)+1,
|
||
|
rand.Intn(255),
|
||
|
rand.Intn(255),
|
||
|
rand.Intn(255),
|
||
|
)
|
||
|
port := rand.Intn(65535-1) + 1
|
||
|
addr := NewNetAddressString(fmt.Sprintf("%v:%v", ip, port))
|
||
|
if addr.Routable() {
|
||
|
return addr
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestSaveAddresses(t *testing.T) {
|
||
|
fname := createTempFileName("addrbook_test")
|
||
|
//t.Logf("New tempfile name: %v", fname)
|
||
|
|
||
|
// Create some random addresses
|
||
|
randAddrs := []struct {
|
||
|
addr *NetAddress
|
||
|
src *NetAddress
|
||
|
}{}
|
||
|
for i := 0; i < 100; i++ {
|
||
|
addr := randIPv4Address()
|
||
|
src := randIPv4Address()
|
||
|
randAddrs = append(randAddrs, struct {
|
||
|
addr *NetAddress
|
||
|
src *NetAddress
|
||
|
}{
|
||
|
addr: addr,
|
||
|
src: src,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
// Create the book & populate & save
|
||
|
book := NewAddrBook(fname)
|
||
|
for _, addrSrc := range randAddrs {
|
||
|
book.AddAddress(addrSrc.addr, addrSrc.src)
|
||
|
}
|
||
|
if book.Size() != 100 {
|
||
|
t.Errorf("Expected 100 addresses, found %v", book.Size())
|
||
|
}
|
||
|
book.saveToFile(fname)
|
||
|
|
||
|
// Reload the book
|
||
|
book = NewAddrBook(fname)
|
||
|
book.loadFromFile(fname)
|
||
|
|
||
|
// Test ...
|
||
|
|
||
|
if book.Size() != 100 {
|
||
|
t.Errorf("Expected 100 addresses, found %v", book.Size())
|
||
|
}
|
||
|
|
||
|
for _, addrSrc := range randAddrs {
|
||
|
addr := addrSrc.addr
|
||
|
src := addrSrc.src
|
||
|
ka := book.addrLookup[addr.String()]
|
||
|
if ka == nil {
|
||
|
t.Fatalf("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")
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestPromoteToOld(t *testing.T) {
|
||
|
fname := createTempFileName("addrbook_test")
|
||
|
t.Logf("New tempfile name: %v", fname)
|
||
|
|
||
|
// Create some random addresses
|
||
|
randAddrs := []struct {
|
||
|
addr *NetAddress
|
||
|
src *NetAddress
|
||
|
}{}
|
||
|
for i := 0; i < 100; i++ {
|
||
|
addr := randIPv4Address()
|
||
|
src := randIPv4Address()
|
||
|
randAddrs = append(randAddrs, struct {
|
||
|
addr *NetAddress
|
||
|
src *NetAddress
|
||
|
}{
|
||
|
addr: addr,
|
||
|
src: src,
|
||
|
})
|
||
|
}
|
||
|
|
||
|
// Create the book & populate & save
|
||
|
book := NewAddrBook(fname)
|
||
|
for _, addrSrc := range randAddrs {
|
||
|
book.AddAddress(addrSrc.addr, addrSrc.src)
|
||
|
}
|
||
|
// Attempt all addresses.
|
||
|
for _, addrSrc := range randAddrs {
|
||
|
book.MarkAttempt(addrSrc.addr)
|
||
|
}
|
||
|
// Promote half of them
|
||
|
for i, addrSrc := range randAddrs {
|
||
|
if i%2 == 0 {
|
||
|
book.MarkGood(addrSrc.addr)
|
||
|
}
|
||
|
}
|
||
|
book.saveToFile(fname)
|
||
|
|
||
|
// Reload the book
|
||
|
book = NewAddrBook(fname)
|
||
|
book.loadFromFile(fname)
|
||
|
|
||
|
// Test ...
|
||
|
|
||
|
if book.Size() != 100 {
|
||
|
t.Errorf("Expected 100 addresses, found %v", book.Size())
|
||
|
}
|
||
|
|
||
|
// TODO: do more testing :)
|
||
|
|
||
|
selection := book.GetSelection()
|
||
|
t.Logf("selection: %v", selection)
|
||
|
}
|