go-pkg/cmux
2
0
Fork 0
mirror of https://github.com/soheilhy/cmux.git synced 2024-09-20 02:55:46 +08:00
Code Issues Projects Releases Wiki Activity

let the kernel assign ports

Browse Source
This commit is contained in:
Tamir Duberstein 2016-02-20 15:43:18 -05:00 committed by Soheil Hassas Yeganeh
parent 4f1b8abafd
commit ecc37f82cd
1 changed files with 23 additions and 39 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
cmux_test.go
View File

@ -14,20 +14,16 @@ const (
rpcVal = 1234 rpcVal = 1234
) )
var testPort = 5125 func testListener(t *testing.T) (net.Listener, func()) {
l, err := net.Listen("tcp", ":0")
func testAddr() string {
testPort++
return fmt.Sprintf("127.0.0.1:%d", testPort)
}
func testListener(t *testing.T) (net.Listener, string) {
addr := testAddr()
l, err := net.Listen("tcp", addr)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
return l, addr return l, func() {
if err := l.Close(); err != nil {
t.Error(err)
}
}
} }
type testHTTP1Handler struct{} type testHTTP1Handler struct{}
@ -40,13 +36,13 @@ func runTestHTTPServer(t *testing.T, l net.Listener) {
s := &http.Server{ s := &http.Server{
Handler: &testHTTP1Handler{}, Handler: &testHTTP1Handler{},
} }
if err := s.Serve(l); err != nil { if err := s.Serve(l); err != nil && err != ErrListenerClosed {
t.Log(err) t.Log(err)
} }
} }
func runTestHTTP1Client(t *testing.T, addr string) { func runTestHTTP1Client(t *testing.T, addr net.Addr) {
r, err := http.Get("http://" + addr) r, err := http.Get("http://" + addr.String())
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
@ -88,8 +84,8 @@ func runTestRPCServer(t *testing.T, l net.Listener) {
} }
} }
func runTestRPCClient(t *testing.T, addr string) { func runTestRPCClient(t *testing.T, addr net.Addr) {
c, err := rpc.Dial("tcp", addr) c, err := rpc.Dial(addr.Network(), addr.String())
if err != nil { if err != nil {
t.Error(err) t.Error(err)
return return
@ -107,12 +103,8 @@ func runTestRPCClient(t *testing.T, addr string) {
} }
func TestAny(t *testing.T) { func TestAny(t *testing.T) {
l, addr := testListener(t) l, cleanup := testListener(t)
defer func() { defer cleanup()
if err := l.Close(); err != nil {
t.Log(err)
}
}()
muxl := New(l) muxl := New(l)
httpl := muxl.Match(Any()) httpl := muxl.Match(Any())
@ -124,11 +116,10 @@ func TestAny(t *testing.T) {
} }
}() }()
r, err := http.Get("http://" + addr) r, err := http.Get("http://" + l.Addr().String())
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
defer func() { defer func() {
if err := r.Body.Close(); err != nil { if err := r.Body.Close(); err != nil {
t.Log(err) t.Log(err)
@ -144,12 +135,8 @@ func TestAny(t *testing.T) {
} }
func TestHTTPGoRPC(t *testing.T) { func TestHTTPGoRPC(t *testing.T) {
l, addr := testListener(t) l, cleanup := testListener(t)
defer func() { defer cleanup()
if err := l.Close(); err != nil {
t.Log(err)
}
}()
muxl := New(l) muxl := New(l)
httpl := muxl.Match(HTTP2(), HTTP1Fast()) httpl := muxl.Match(HTTP2(), HTTP1Fast())
@ -163,17 +150,13 @@ func TestHTTPGoRPC(t *testing.T) {
} }
}() }()
runTestHTTP1Client(t, addr) runTestHTTP1Client(t, l.Addr())
runTestRPCClient(t, addr) runTestRPCClient(t, l.Addr())
} }
func TestErrorHandler(t *testing.T) { func TestErrorHandler(t *testing.T) {
l, addr := testListener(t) l, cleanup := testListener(t)
defer func() { defer cleanup()
if err := l.Close(); err != nil {
t.Log(err)
}
}()
muxl := New(l) muxl := New(l)
httpl := muxl.Match(HTTP2(), HTTP1Fast()) httpl := muxl.Match(HTTP2(), HTTP1Fast())
@ -197,7 +180,8 @@ func TestErrorHandler(t *testing.T) {
return true return true
}) })
c, err := rpc.Dial("tcp", addr) addr := l.Addr()
c, err := rpc.Dial(addr.Network(), addr.String())
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }