83 lines
2.5 KiB
Go
83 lines
2.5 KiB
Go
package server
|
|
|
|
import (
|
|
"log"
|
|
"os/exec"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
// TestAddPeer tests the ability to add peers
|
|
func TestUDPPing(t *testing.T) {
|
|
args := makeDefaultArgs()
|
|
args.StartUDP = false
|
|
|
|
tests := []struct {
|
|
name string
|
|
wantIP string
|
|
wantCountry string
|
|
wantProtocolVersion int
|
|
wantHeightMin int
|
|
wantFlags byte
|
|
} {
|
|
{
|
|
name: "Correctly parse information from production server.",
|
|
wantIP: "SETME",
|
|
wantCountry: "US",
|
|
wantProtocolVersion: 1,
|
|
wantHeightMin: 1060000,
|
|
wantFlags: 1,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T){
|
|
|
|
toAddr := "spv16.lbry.com"
|
|
toPort := "50001"
|
|
|
|
pong, err := UDPPing(toAddr, toPort)
|
|
gotCountry := pong.DecodeCountry()
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
|
|
res, err := exec.Command("dig", "@resolver4.opendns.com", "myip.opendns.com", "+short").Output()
|
|
|
|
if err != nil {
|
|
log.Println(err)
|
|
}
|
|
|
|
digIP := strings.TrimSpace(string(res))
|
|
udpIP := pong.DecodeAddress().String()
|
|
tt.wantIP = digIP
|
|
|
|
log.Println("Height:", pong.DecodeHeight())
|
|
log.Printf("Flags: %x\n", pong.DecodeFlags())
|
|
log.Println("ProtocolVersion:", pong.DecodeProtocolVersion())
|
|
log.Printf("Tip: %x\n", pong.DecodeTip())
|
|
|
|
gotHeight := pong.DecodeHeight()
|
|
gotProtocolVersion := pong.DecodeProtocolVersion()
|
|
gotFlags := pong.DecodeFlags()
|
|
gotIP := udpIP
|
|
|
|
if gotIP != tt.wantIP {
|
|
t.Errorf("ip: got: '%s', want: '%s'\n", gotIP, tt.wantIP)
|
|
}
|
|
if gotCountry != tt.wantCountry {
|
|
t.Errorf("country: got: '%s', want: '%s'\n", gotCountry, tt.wantCountry)
|
|
}
|
|
if gotHeight < tt.wantHeightMin {
|
|
t.Errorf("height: got: %d, want >=: %d\n", gotHeight, tt.wantHeightMin)
|
|
}
|
|
if gotProtocolVersion != tt.wantProtocolVersion {
|
|
t.Errorf("protocolVersion: got: %d, want: %d\n", gotProtocolVersion, tt.wantProtocolVersion)
|
|
}
|
|
if gotFlags != tt.wantFlags {
|
|
t.Errorf("flags: got: %d, want: %d\n", gotFlags, tt.wantFlags)
|
|
}
|
|
})
|
|
}
|
|
|
|
}
|