mirror of
https://github.com/ipfs/kubo.git
synced 2026-02-27 21:37:57 +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>
144 lines
2.6 KiB
Go
144 lines
2.6 KiB
Go
package conn
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"os"
|
|
"runtime"
|
|
"strconv"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
context "github.com/jbenet/go-ipfs/Godeps/_workspace/src/code.google.com/p/go.net/context"
|
|
)
|
|
|
|
func TestClose(t *testing.T) {
|
|
// t.Skip("Skipping in favor of another test")
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
c1, c2 := setupConn(t, ctx, "/ip4/127.0.0.1/tcp/5534", "/ip4/127.0.0.1/tcp/5545")
|
|
|
|
select {
|
|
case <-c1.Closed():
|
|
t.Fatal("done before close")
|
|
case <-c2.Closed():
|
|
t.Fatal("done before close")
|
|
default:
|
|
}
|
|
|
|
c1.Close()
|
|
|
|
select {
|
|
case <-c1.Closed():
|
|
default:
|
|
t.Fatal("not done after cancel")
|
|
}
|
|
|
|
c2.Close()
|
|
|
|
select {
|
|
case <-c2.Closed():
|
|
default:
|
|
t.Fatal("not done after cancel")
|
|
}
|
|
|
|
cancel() // close the listener :P
|
|
}
|
|
|
|
func TestCancel(t *testing.T) {
|
|
// t.Skip("Skipping in favor of another test")
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
c1, c2 := setupConn(t, ctx, "/ip4/127.0.0.1/tcp/5534", "/ip4/127.0.0.1/tcp/5545")
|
|
|
|
select {
|
|
case <-c1.Closed():
|
|
t.Fatal("done before close")
|
|
case <-c2.Closed():
|
|
t.Fatal("done before close")
|
|
default:
|
|
}
|
|
|
|
c1.Close()
|
|
c2.Close()
|
|
cancel() // listener
|
|
|
|
// wait to ensure other goroutines run and close things.
|
|
<-time.After(time.Microsecond * 10)
|
|
// test that cancel called Close.
|
|
|
|
select {
|
|
case <-c1.Closed():
|
|
default:
|
|
t.Fatal("not done after cancel")
|
|
}
|
|
|
|
select {
|
|
case <-c2.Closed():
|
|
default:
|
|
t.Fatal("not done after cancel")
|
|
}
|
|
|
|
}
|
|
|
|
func TestCloseLeak(t *testing.T) {
|
|
if testing.Short() {
|
|
t.SkipNow()
|
|
}
|
|
|
|
if os.Getenv("TRAVIS") == "true" {
|
|
t.Skip("this doesn't work well on travis")
|
|
}
|
|
|
|
var wg sync.WaitGroup
|
|
|
|
runPair := func(p1, p2, num int) {
|
|
a1 := strconv.Itoa(p1)
|
|
a2 := strconv.Itoa(p2)
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
c1, c2 := setupConn(t, ctx, "/ip4/127.0.0.1/tcp/"+a1, "/ip4/127.0.0.1/tcp/"+a2)
|
|
|
|
for i := 0; i < num; i++ {
|
|
b1 := []byte("beep")
|
|
c1.Out() <- b1
|
|
b2 := <-c2.In()
|
|
if !bytes.Equal(b1, b2) {
|
|
panic("bytes not equal")
|
|
}
|
|
|
|
b2 = []byte("boop")
|
|
c2.Out() <- b2
|
|
b1 = <-c1.In()
|
|
if !bytes.Equal(b1, b2) {
|
|
panic("bytes not equal")
|
|
}
|
|
|
|
<-time.After(time.Microsecond * 5)
|
|
}
|
|
|
|
c1.Close()
|
|
c2.Close()
|
|
cancel() // close the listener
|
|
wg.Done()
|
|
}
|
|
|
|
var cons = 20
|
|
var msgs = 100
|
|
fmt.Printf("Running %d connections * %d msgs.\n", cons, msgs)
|
|
for i := 0; i < cons; i++ {
|
|
wg.Add(1)
|
|
go runPair(2000+i, 2001+i, msgs)
|
|
}
|
|
|
|
fmt.Printf("Waiting...\n")
|
|
wg.Wait()
|
|
// done!
|
|
|
|
<-time.After(time.Millisecond * 150)
|
|
if runtime.NumGoroutine() > 20 {
|
|
// panic("uncomment me to debug")
|
|
t.Fatal("leaking goroutines:", runtime.NumGoroutine())
|
|
}
|
|
}
|