mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-24 11:57:44 +08:00
ed4a8eb782
vanilla:
21.57 real 45.14 user 8.51 sys
short:
14.40 real 31.13 user 5.56 sys
License: MIT
Signed-off-by: Brian Tiger Chow <brian@perfmode.com>
82 lines
1.5 KiB
Go
82 lines
1.5 KiB
Go
package swarm
|
|
|
|
import (
|
|
"fmt"
|
|
"sync"
|
|
"testing"
|
|
|
|
peer "github.com/jbenet/go-ipfs/peer"
|
|
|
|
context "github.com/jbenet/go-ipfs/Godeps/_workspace/src/code.google.com/p/go.net/context"
|
|
)
|
|
|
|
func TestSimultOpen(t *testing.T) {
|
|
if testing.Short() {
|
|
t.SkipNow()
|
|
}
|
|
|
|
addrs := []string{
|
|
"/ip4/127.0.0.1/tcp/1244",
|
|
"/ip4/127.0.0.1/tcp/1245",
|
|
}
|
|
|
|
ctx := context.Background()
|
|
swarms, _ := makeSwarms(ctx, t, addrs)
|
|
|
|
// connect everyone
|
|
{
|
|
var wg sync.WaitGroup
|
|
connect := func(s *Swarm, dst peer.Peer) {
|
|
// copy for other peer
|
|
cp := peer.WithID(dst.ID())
|
|
cp.AddAddress(dst.Addresses()[0])
|
|
|
|
if _, err := s.Dial(cp); 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)
|
|
go connect(swarms[1], swarms[0].local)
|
|
wg.Wait()
|
|
}
|
|
|
|
for _, s := range swarms {
|
|
s.Close()
|
|
}
|
|
}
|
|
|
|
func TestSimultOpenMany(t *testing.T) {
|
|
t.Skip("laggy")
|
|
|
|
many := 500
|
|
addrs := []string{}
|
|
for i := 2200; i < (2200 + many); i++ {
|
|
s := fmt.Sprintf("/ip4/127.0.0.1/tcp/%d", i)
|
|
addrs = append(addrs, s)
|
|
}
|
|
|
|
SubtestSwarm(t, addrs, 10)
|
|
}
|
|
|
|
func TestSimultOpenFewStress(t *testing.T) {
|
|
if testing.Short() {
|
|
t.SkipNow()
|
|
}
|
|
// t.Skip("skipping for another test")
|
|
|
|
num := 10
|
|
// num := 100
|
|
for i := 0; i < num; i++ {
|
|
addrs := []string{
|
|
fmt.Sprintf("/ip4/127.0.0.1/tcp/%d", 1900+i),
|
|
fmt.Sprintf("/ip4/127.0.0.1/tcp/%d", 2900+i),
|
|
}
|
|
|
|
SubtestSwarm(t, addrs, 10)
|
|
}
|
|
}
|