go-libp2p-peerstore/test/utils.go

75 lines
1.4 KiB
Go
Raw Normal View History

package test
2018-06-05 14:18:06 +08:00
import (
2018-07-25 07:48:16 +08:00
"context"
2018-06-05 14:18:06 +08:00
"fmt"
"testing"
2018-09-11 21:34:26 +08:00
peer "github.com/libp2p/go-libp2p-peer"
2018-09-01 01:49:07 +08:00
pt "github.com/libp2p/go-libp2p-peer/test"
2018-09-02 19:10:55 +08:00
ma "github.com/multiformats/go-multiaddr"
2018-06-05 14:18:06 +08:00
)
2018-11-28 08:13:13 +08:00
func Multiaddr(m string) ma.Multiaddr {
2018-09-04 18:34:49 +08:00
maddr, err := ma.NewMultiaddr(m)
if err != nil {
panic(err)
}
return maddr
}
2018-06-05 14:18:06 +08:00
type peerpair struct {
2018-09-01 01:49:07 +08:00
ID peer.ID
Addr []ma.Multiaddr
2018-06-05 14:18:06 +08:00
}
2018-11-28 08:13:13 +08:00
func RandomPeer(b *testing.B, addrCount int) *peerpair {
var (
pid peer.ID
err error
addrs = make([]ma.Multiaddr, addrCount)
aFmt = "/ip4/127.0.0.1/tcp/%d/ipfs/%s"
)
2018-06-05 14:18:06 +08:00
b.Helper()
2018-09-01 01:49:07 +08:00
if pid, err = pt.RandPeerID(); err != nil {
2018-06-05 14:18:06 +08:00
b.Fatal(err)
}
for i := 0; i < addrCount; i++ {
if addrs[i], err = ma.NewMultiaddr(fmt.Sprintf(aFmt, i, pid.Pretty())); err != nil {
b.Fatal(err)
}
2018-06-05 14:18:06 +08:00
}
return &peerpair{pid, addrs}
2018-06-05 14:18:06 +08:00
}
2018-11-28 08:13:13 +08:00
func AddressProducer(ctx context.Context, b *testing.B, addrs chan *peerpair, addrsPerPeer int) {
b.Helper()
2018-06-05 14:18:06 +08:00
defer close(addrs)
for {
2018-11-28 08:13:13 +08:00
p := RandomPeer(b, addrsPerPeer)
2018-07-25 07:48:16 +08:00
select {
case addrs <- p:
case <-ctx.Done():
return
2018-06-05 14:18:06 +08:00
}
}
}
2018-11-28 08:13:13 +08:00
func GenerateAddrs(count int) []ma.Multiaddr {
var addrs = make([]ma.Multiaddr, count)
for i := 0; i < count; i++ {
addrs[i] = Multiaddr(fmt.Sprintf("/ip4/1.1.1.%d/tcp/1111", i))
}
return addrs
}
func GeneratePeerIDs(count int) []peer.ID {
var ids = make([]peer.ID, count)
for i := 0; i < count; i++ {
ids[i], _ = pt.RandPeerID()
}
return ids
}