mirror of
https://github.com/libp2p/go-libp2p.git
synced 2025-09-27 04:26:41 +08:00
f645cb8c04
This commit makes all network tests use ZeroLocalTCPAddress as the initial peer address, and then relies on net.ListenAddresses() This should get rid of the tcp addr clash problems.
68 lines
1.4 KiB
Go
68 lines
1.4 KiB
Go
package swarm
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
peer "github.com/jbenet/go-ipfs/p2p/peer"
|
|
|
|
context "github.com/jbenet/go-ipfs/Godeps/_workspace/src/code.google.com/p/go.net/context"
|
|
ma "github.com/jbenet/go-ipfs/Godeps/_workspace/src/github.com/jbenet/go-multiaddr"
|
|
)
|
|
|
|
func TestSimultOpen(t *testing.T) {
|
|
// t.Skip("skipping for another test")
|
|
|
|
ctx := context.Background()
|
|
swarms := makeSwarms(ctx, t, 2)
|
|
|
|
// connect everyone
|
|
{
|
|
var wg sync.WaitGroup
|
|
connect := func(s *Swarm, dst peer.ID, addr ma.Multiaddr) {
|
|
// copy for other peer
|
|
log.Debugf("TestSimultOpen: connecting: %s --> %s (%s)", s.local, dst, addr)
|
|
s.peers.AddAddress(dst, addr)
|
|
if _, err := s.Dial(ctx, dst); err != nil {
|
|
t.Fatal("error swarm dialing to peer", err)
|
|
}
|
|
wg.Done()
|
|
}
|
|
|
|
log.Info("Connecting swarms simultaneously.")
|
|
wg.Add(2)
|
|
go connect(swarms[0], swarms[1].local, swarms[1].ListenAddresses()[0])
|
|
go connect(swarms[1], swarms[0].local, swarms[0].ListenAddresses()[0])
|
|
wg.Wait()
|
|
}
|
|
|
|
for _, s := range swarms {
|
|
s.Close()
|
|
}
|
|
}
|
|
|
|
func TestSimultOpenMany(t *testing.T) {
|
|
// t.Skip("very very slow")
|
|
|
|
addrs := 20
|
|
SubtestSwarm(t, addrs, 10)
|
|
}
|
|
|
|
func TestSimultOpenFewStress(t *testing.T) {
|
|
if testing.Short() {
|
|
t.SkipNow()
|
|
}
|
|
// t.Skip("skipping for another test")
|
|
|
|
msgs := 40
|
|
swarms := 2
|
|
rounds := 10
|
|
// rounds := 100
|
|
|
|
for i := 0; i < rounds; i++ {
|
|
SubtestSwarm(t, swarms, msgs)
|
|
<-time.After(10 * time.Millisecond)
|
|
}
|
|
}
|