mirror of https://github.com/libp2p/go-libp2p.git
Marten Seemann
1 year ago
committed by
GitHub
GitHub
8 changed files with 21 additions and 104 deletions
@ -1,59 +0,0 @@ |
|||
package network |
|||
|
|||
import ( |
|||
"fmt" |
|||
"math" |
|||
"math/rand" |
|||
"testing" |
|||
|
|||
ma "github.com/multiformats/go-multiaddr" |
|||
|
|||
"github.com/stretchr/testify/require" |
|||
) |
|||
|
|||
func TestDedupAddrs(t *testing.T) { |
|||
tcpAddr := ma.StringCast("/ip4/127.0.0.1/tcp/1234") |
|||
quicAddr := ma.StringCast("/ip4/127.0.0.1/udp/1234/quic-v1") |
|||
wsAddr := ma.StringCast("/ip4/127.0.0.1/tcp/1234/ws") |
|||
|
|||
type testcase struct { |
|||
in, out []ma.Multiaddr |
|||
} |
|||
|
|||
for i, tc := range []testcase{ |
|||
{in: nil, out: nil}, |
|||
{in: []ma.Multiaddr{tcpAddr}, out: []ma.Multiaddr{tcpAddr}}, |
|||
{in: []ma.Multiaddr{tcpAddr, tcpAddr, tcpAddr}, out: []ma.Multiaddr{tcpAddr}}, |
|||
{in: []ma.Multiaddr{tcpAddr, quicAddr, tcpAddr}, out: []ma.Multiaddr{tcpAddr, quicAddr}}, |
|||
{in: []ma.Multiaddr{tcpAddr, quicAddr, wsAddr}, out: []ma.Multiaddr{tcpAddr, quicAddr, wsAddr}}, |
|||
} { |
|||
tc := tc |
|||
t.Run(fmt.Sprintf("test %d", i), func(t *testing.T) { |
|||
deduped := DedupAddrs(tc.in) |
|||
for _, a := range tc.out { |
|||
require.Contains(t, deduped, a) |
|||
} |
|||
}) |
|||
} |
|||
} |
|||
|
|||
func BenchmarkDedupAddrs(b *testing.B) { |
|||
b.ReportAllocs() |
|||
var addrs []ma.Multiaddr |
|||
r := rand.New(rand.NewSource(1234)) |
|||
for i := 0; i < 100; i++ { |
|||
tcpAddr := ma.StringCast(fmt.Sprintf("/ip4/127.0.0.1/tcp/%d", r.Intn(math.MaxUint16))) |
|||
quicAddr := ma.StringCast(fmt.Sprintf("/ip4/127.0.0.1/udp/%d/quic-v1", r.Intn(math.MaxUint16))) |
|||
wsAddr := ma.StringCast(fmt.Sprintf("/ip4/127.0.0.1/tcp/%d/ws", r.Intn(math.MaxUint16))) |
|||
addrs = append(addrs, tcpAddr, tcpAddr, quicAddr, quicAddr, wsAddr) |
|||
} |
|||
for _, sz := range []int{10, 20, 30, 50, 100} { |
|||
b.Run(fmt.Sprintf("%d", sz), func(b *testing.B) { |
|||
items := make([]ma.Multiaddr, sz) |
|||
for i := 0; i < b.N; i++ { |
|||
copy(items, addrs[:sz]) |
|||
DedupAddrs(items) |
|||
} |
|||
}) |
|||
} |
|||
} |
|||
Loading…
Reference in new issue