2019-10-09 15:32:15 -07:00
|
|
|
package zcash
|
|
|
|
|
|
|
|
import (
|
2019-10-12 09:42:16 -07:00
|
|
|
"net"
|
2019-10-12 19:00:20 -07:00
|
|
|
"os"
|
2019-10-12 16:18:25 -07:00
|
|
|
"sync"
|
2019-10-09 15:32:15 -07:00
|
|
|
"testing"
|
2019-10-12 19:00:20 -07:00
|
|
|
"time"
|
2019-10-09 15:32:15 -07:00
|
|
|
|
2019-10-12 09:42:16 -07:00
|
|
|
"github.com/btcsuite/btcd/peer"
|
2019-10-12 19:00:20 -07:00
|
|
|
"github.com/btcsuite/btcd/wire"
|
|
|
|
"github.com/btcsuite/btclog"
|
2019-10-09 15:32:15 -07:00
|
|
|
"github.com/gtank/coredns-zcash/zcash/network"
|
|
|
|
)
|
|
|
|
|
2019-10-12 19:00:20 -07:00
|
|
|
func TestMain(m *testing.M) {
|
|
|
|
startMockLoop()
|
|
|
|
exitCode := m.Run()
|
|
|
|
os.Exit(exitCode)
|
|
|
|
}
|
|
|
|
|
|
|
|
func startMockLoop() {
|
2019-10-12 09:42:16 -07:00
|
|
|
// Configure peer to act as a regtest node that offers no services.
|
|
|
|
config, err := newSeederPeerConfig(network.Regtest, defaultPeerConfig)
|
|
|
|
if err != nil {
|
2019-10-12 19:00:20 -07:00
|
|
|
return
|
2019-10-12 09:42:16 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
config.AllowSelfConns = true
|
|
|
|
|
2019-10-12 19:00:20 -07:00
|
|
|
backendLogger := btclog.NewBackend(os.Stdout)
|
|
|
|
mockPeerLogger := backendLogger.Logger("mockPeer")
|
|
|
|
//mockPeerLogger.SetLevel(btclog.LevelTrace)
|
|
|
|
peer.UseLogger(mockPeerLogger)
|
|
|
|
|
|
|
|
config.Listeners.OnGetAddr = func(p *peer.Peer, msg *wire.MsgGetAddr) {
|
|
|
|
cache := make([]*wire.NetAddress, 0, 1)
|
|
|
|
addr := wire.NewNetAddressTimestamp(
|
|
|
|
time.Now(),
|
|
|
|
0,
|
|
|
|
net.ParseIP("127.0.0.1"),
|
2019-10-16 19:18:30 -07:00
|
|
|
uint16(18233),
|
2019-10-12 19:00:20 -07:00
|
|
|
)
|
2019-10-16 19:18:30 -07:00
|
|
|
addr2 := wire.NewNetAddressTimestamp(
|
|
|
|
time.Now(),
|
|
|
|
0,
|
|
|
|
net.ParseIP("127.0.0.1"),
|
|
|
|
uint16(18344),
|
|
|
|
)
|
|
|
|
cache = append(cache, addr, addr2)
|
2019-10-12 19:00:20 -07:00
|
|
|
_, err := p.PushAddrMsg(cache)
|
|
|
|
if err != nil {
|
|
|
|
mockPeerLogger.Error(err)
|
|
|
|
}
|
|
|
|
}
|
2019-10-12 09:42:16 -07:00
|
|
|
|
|
|
|
listenAddr := net.JoinHostPort("127.0.0.1", config.ChainParams.DefaultPort)
|
|
|
|
listener, err := net.Listen("tcp", listenAddr)
|
|
|
|
if err != nil {
|
2019-10-12 19:00:20 -07:00
|
|
|
return
|
2019-10-12 09:42:16 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
go func() {
|
2019-10-12 19:00:20 -07:00
|
|
|
for {
|
|
|
|
conn, err := listener.Accept()
|
|
|
|
if err != nil {
|
|
|
|
return
|
|
|
|
}
|
2019-10-12 09:42:16 -07:00
|
|
|
|
2019-10-12 19:00:20 -07:00
|
|
|
mockPeer := peer.NewInboundPeer(config)
|
|
|
|
mockPeer.AssociateConnection(conn)
|
2019-10-12 09:42:16 -07:00
|
|
|
}
|
|
|
|
}()
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestOutboundPeerSync(t *testing.T) {
|
|
|
|
regSeeder, err := newTestSeeder(network.Regtest)
|
2019-10-09 15:32:15 -07:00
|
|
|
if err != nil {
|
2019-10-12 19:00:20 -07:00
|
|
|
t.Error(err)
|
|
|
|
return
|
2019-10-09 15:32:15 -07:00
|
|
|
}
|
|
|
|
|
2019-10-16 18:05:42 -07:00
|
|
|
err = regSeeder.ConnectOnDefaultPort("127.0.0.1")
|
2019-10-09 15:32:15 -07:00
|
|
|
if err != nil {
|
2019-10-12 19:00:20 -07:00
|
|
|
t.Error(err)
|
|
|
|
return
|
2019-10-12 09:42:16 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
// Can we address that peer if we want to?
|
2019-10-16 18:05:42 -07:00
|
|
|
p, err := regSeeder.GetPeer("127.0.0.1:18344")
|
2019-10-12 09:42:16 -07:00
|
|
|
if err != nil {
|
2019-10-12 19:00:20 -07:00
|
|
|
t.Error(err)
|
|
|
|
return
|
2019-10-12 09:42:16 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
if p.Connected() {
|
2019-10-16 18:05:42 -07:00
|
|
|
regSeeder.DisconnectPeer("127.0.0.1:18344")
|
2019-10-12 09:42:16 -07:00
|
|
|
} else {
|
|
|
|
t.Error("Peer never connected")
|
|
|
|
}
|
|
|
|
|
|
|
|
// Can we STILL address a flushed peer?
|
2019-10-16 18:05:42 -07:00
|
|
|
p, err = regSeeder.GetPeer("127.0.0.1:18344")
|
2019-10-12 19:00:20 -07:00
|
|
|
if err != ErrNoSuchPeer {
|
2019-10-12 09:42:16 -07:00
|
|
|
t.Error("Peer should have been cleared on disconnect")
|
2019-10-09 15:32:15 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func TestOutboundPeerAsync(t *testing.T) {
|
2019-10-12 09:42:16 -07:00
|
|
|
regSeeder, err := newTestSeeder(network.Regtest)
|
2019-10-09 15:32:15 -07:00
|
|
|
if err != nil {
|
|
|
|
t.Fatal(err)
|
|
|
|
}
|
|
|
|
|
2019-10-12 16:18:25 -07:00
|
|
|
var wg sync.WaitGroup
|
|
|
|
for i := 0; i < 4; i++ {
|
|
|
|
wg.Add(1)
|
|
|
|
go func() {
|
2019-10-16 18:05:42 -07:00
|
|
|
err := regSeeder.ConnectOnDefaultPort("127.0.0.1")
|
2019-10-12 16:18:25 -07:00
|
|
|
if err != nil && err != ErrRepeatConnection {
|
|
|
|
t.Error(err)
|
|
|
|
}
|
|
|
|
wg.Done()
|
|
|
|
}()
|
|
|
|
}
|
|
|
|
|
|
|
|
wg.Wait()
|
|
|
|
|
|
|
|
// Can we address that peer if we want to?
|
2019-10-16 18:05:42 -07:00
|
|
|
p, err := regSeeder.GetPeer("127.0.0.1:18344")
|
2019-10-12 16:18:25 -07:00
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
if p.Connected() {
|
|
|
|
// Shouldn't try to connect to a live peer again
|
2019-10-16 18:05:42 -07:00
|
|
|
err := regSeeder.ConnectOnDefaultPort("127.0.0.1")
|
2019-10-12 16:18:25 -07:00
|
|
|
if err != ErrRepeatConnection {
|
|
|
|
t.Error("should have caught repeat connection attempt")
|
2019-10-09 15:32:15 -07:00
|
|
|
}
|
2019-10-12 16:18:25 -07:00
|
|
|
} else {
|
|
|
|
t.Error("Peer never connected")
|
2019-10-09 15:32:15 -07:00
|
|
|
}
|
2019-10-12 16:18:25 -07:00
|
|
|
|
|
|
|
regSeeder.DisconnectAllPeers()
|
2019-10-09 15:32:15 -07:00
|
|
|
}
|
2019-10-12 19:00:20 -07:00
|
|
|
|
|
|
|
func TestRequestAddresses(t *testing.T) {
|
|
|
|
regSeeder, err := newTestSeeder(network.Regtest)
|
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
2019-10-16 18:05:42 -07:00
|
|
|
err = regSeeder.ConnectOnDefaultPort("127.0.0.1")
|
2019-10-12 19:00:20 -07:00
|
|
|
if err != nil {
|
|
|
|
t.Error(err)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
regSeeder.RequestAddresses()
|
2019-10-16 19:18:30 -07:00
|
|
|
regSeeder.WaitForAddresses(1)
|
|
|
|
|
|
|
|
// TODO It isn't possible to test this wait on a local mock peer without
|
|
|
|
// carving a path through absolutely all of the bad connection logic.
|
2019-10-12 19:00:20 -07:00
|
|
|
}
|