mirror of
https://github.com/golang/go
synced 2024-09-30 16:28:32 -06:00
dd5f9d1033
This adds a package for dealing with stacks in tests. The only function at this time is NoLeak which verifies that a test does not leak any goroutines, and prints a stack summary if it does. Change-Id: I284b846f422334b745bef0e96d84138295120a62 Reviewed-on: https://go-review.googlesource.com/c/tools/+/232681 Run-TryBot: Ian Cottrell <iancottrell@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Robert Findley <rfindley@google.com>
63 lines
1.3 KiB
Go
63 lines
1.3 KiB
Go
// Copyright 2020 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package jsonrpc2
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
"golang.org/x/tools/internal/stack/stacktest"
|
|
)
|
|
|
|
func TestIdleTimeout(t *testing.T) {
|
|
stacktest.NoLeak(t)
|
|
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
|
defer cancel()
|
|
|
|
ln, err := net.Listen("tcp", "localhost:0")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer ln.Close()
|
|
|
|
connect := func() net.Conn {
|
|
conn, err := net.DialTimeout("tcp", ln.Addr().String(), 5*time.Second)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return conn
|
|
}
|
|
|
|
server := HandlerServer(MethodNotFound)
|
|
// connTimer := &fakeTimer{c: make(chan time.Time, 1)}
|
|
var (
|
|
runErr error
|
|
wg sync.WaitGroup
|
|
)
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
runErr = Serve(ctx, ln, server, 100*time.Millisecond)
|
|
}()
|
|
|
|
// Exercise some connection/disconnection patterns, and then assert that when
|
|
// our timer fires, the server exits.
|
|
conn1 := connect()
|
|
conn2 := connect()
|
|
conn1.Close()
|
|
conn2.Close()
|
|
conn3 := connect()
|
|
conn3.Close()
|
|
|
|
wg.Wait()
|
|
|
|
if runErr != ErrIdleTimeout {
|
|
t.Errorf("run() returned error %v, want %v", runErr, ErrIdleTimeout)
|
|
}
|
|
}
|