| 371 | func (c *requireDeadlineNetConn) RemoteAddr() net.Addr { return c.c.RemoteAddr() } |
| 372 | |
| 373 | func TestHandshakeTimeout(t *testing.T) { |
| 374 | s := newServer(t) |
| 375 | defer s.Close() |
| 376 | |
| 377 | d := cstDialer |
| 378 | d.NetDial = func(n, a string) (net.Conn, error) { |
| 379 | c, err := net.Dial(n, a) |
| 380 | return &requireDeadlineNetConn{c: c, t: t}, err |
| 381 | } |
| 382 | ws, _, err := d.Dial(s.URL, nil) |
| 383 | if err != nil { |
| 384 | t.Fatal("Dial:", err) |
| 385 | } |
| 386 | ws.Close() |
| 387 | } |
| 388 | |
| 389 | func TestHandshakeTimeoutInContext(t *testing.T) { |
| 390 | s := newServer(t) |