mirror of
https://github.com/libp2p/go-libp2p-peerstore.git
synced 2024-12-28 23:50:12 +08:00
150 lines
3.1 KiB
Go
150 lines
3.1 KiB
Go
package peerstore
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/libp2p/go-libp2p-peer"
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
)
|
|
|
|
func mustAddr(t *testing.T, s string) ma.Multiaddr {
|
|
addr, err := ma.NewMultiaddr(s)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
return addr
|
|
}
|
|
|
|
func TestPeerInfoMarshal(t *testing.T) {
|
|
a := mustAddr(t, "/ip4/1.2.3.4/tcp/4536")
|
|
b := mustAddr(t, "/ip4/1.2.3.8/udp/7777")
|
|
id, err := peer.IDB58Decode("QmaCpDMGvV2BGHeYERUEnRQAwe3N8SzbUtfsmvsqQLuvuJ")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
pi := &PeerInfo{
|
|
ID: id,
|
|
Addrs: []ma.Multiaddr{a, b},
|
|
}
|
|
|
|
data, err := pi.MarshalJSON()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
pi2 := new(PeerInfo)
|
|
if err := pi2.UnmarshalJSON(data); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if pi2.ID != pi.ID {
|
|
t.Fatal("ids didnt match after marshal")
|
|
}
|
|
|
|
if !pi.Addrs[0].Equal(pi2.Addrs[0]) {
|
|
t.Fatal("wrong addrs")
|
|
}
|
|
|
|
if !pi.Addrs[1].Equal(pi2.Addrs[1]) {
|
|
t.Fatal("wrong addrs")
|
|
}
|
|
|
|
lgbl := pi2.Loggable()
|
|
if lgbl["peerID"] != id.Pretty() {
|
|
t.Fatal("loggables gave wrong peerID output")
|
|
}
|
|
}
|
|
|
|
func TestP2pAddrParsing(t *testing.T) {
|
|
id, err := peer.IDB58Decode("QmaCpDMGvV2BGHeYERUEnRQAwe3N8SzbUtfsmvsqQLuvuJ")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
addr := ma.StringCast("/ip4/1.2.3.4/tcp/4536")
|
|
p2paddr := ma.Join(addr, ma.StringCast("/ipfs/"+peer.IDB58Encode(id)))
|
|
|
|
pinfo, err := InfoFromP2pAddr(p2paddr)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if pinfo.ID != id {
|
|
t.Fatalf("expected PeerID [%s], got [%s]", id, pinfo.ID)
|
|
}
|
|
|
|
if len(pinfo.Addrs) != 1 {
|
|
t.Fatalf("expected 1 addr, got %d", len(pinfo.Addrs))
|
|
}
|
|
|
|
if !addr.Equal(pinfo.Addrs[0]) {
|
|
t.Fatalf("expected addr [%s], got [%s]", addr, pinfo.Addrs[0])
|
|
}
|
|
|
|
addr = ma.StringCast("/ipfs/" + peer.IDB58Encode(id))
|
|
pinfo, err = InfoFromP2pAddr(addr)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if pinfo.ID != id {
|
|
t.Fatalf("expected PeerID [%s], got [%s]", id, pinfo.ID)
|
|
}
|
|
|
|
if len(pinfo.Addrs) > 0 {
|
|
t.Fatalf("expected 0 addrs, got %d", len(pinfo.Addrs))
|
|
}
|
|
|
|
addr = ma.StringCast("/ip4/1.2.3.4/tcp/4536")
|
|
pinfo, err = InfoFromP2pAddr(addr)
|
|
if err == nil {
|
|
t.Fatalf("expected error, got none")
|
|
}
|
|
|
|
addr = ma.StringCast("/ip4/1.2.3.4/tcp/4536/http")
|
|
pinfo, err = InfoFromP2pAddr(addr)
|
|
if err == nil {
|
|
t.Fatalf("expected error, got none")
|
|
}
|
|
}
|
|
|
|
func TestP2pAddrConstruction(t *testing.T) {
|
|
id, err := peer.IDB58Decode("QmaCpDMGvV2BGHeYERUEnRQAwe3N8SzbUtfsmvsqQLuvuJ")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
addr := ma.StringCast("/ip4/1.2.3.4/tcp/4536")
|
|
p2paddr := ma.Join(addr, ma.StringCast("/ipfs/"+peer.IDB58Encode(id)))
|
|
|
|
pi := &PeerInfo{ID: id, Addrs: []ma.Multiaddr{addr}}
|
|
p2paddrs, err := InfoToP2pAddrs(pi)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if len(p2paddrs) != 1 {
|
|
t.Fatalf("expected 1 addr, got %d", len(p2paddrs))
|
|
}
|
|
|
|
if !p2paddr.Equal(p2paddrs[0]) {
|
|
t.Fatalf("expected [%s], got [%s]", p2paddr, p2paddrs[0])
|
|
}
|
|
|
|
pi = &PeerInfo{ID: id}
|
|
p2paddrs, err = InfoToP2pAddrs(pi)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if len(p2paddrs) > 0 {
|
|
t.Fatalf("expected 0 addrs, got %d", len(p2paddrs))
|
|
}
|
|
|
|
pi = &PeerInfo{Addrs: []ma.Multiaddr{ma.StringCast("/ip4/1.2.3.4/tcp/4536")}}
|
|
_, err = InfoToP2pAddrs(pi)
|
|
if err == nil {
|
|
t.Fatalf("expected error, got none")
|
|
}
|
|
}
|