mirror of
https://github.com/golang/go
synced 2024-11-05 16:56:16 -07:00
c92aeb7438
Replace the String method with a Format method so we can use it for extra formats. Add some tests to make sure it is all correct Change-Id: I39f361ffba036fad99c93f8c0944164f7cf199ec Reviewed-on: https://go-review.googlesource.com/c/tools/+/227486 Run-TryBot: Ian Cottrell <iancottrell@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org> Reviewed-by: Robert Findley <rfindley@google.com>
60 lines
1.2 KiB
Go
60 lines
1.2 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"
|
|
)
|
|
|
|
func TestIdleTimeout(t *testing.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)
|
|
}
|
|
}
|