42 lines
864 B
Go
42 lines
864 B
Go
// Copyright 2011 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.
|
|
|
|
// Bridge package to expose http internals to tests in the http_test
|
|
// package.
|
|
|
|
package http
|
|
|
|
func (t *Transport) IdleConnKeysForTesting() (keys []string) {
|
|
keys = make([]string, 0)
|
|
t.lk.Lock()
|
|
defer t.lk.Unlock()
|
|
if t.idleConn == nil {
|
|
return
|
|
}
|
|
for key := range t.idleConn {
|
|
keys = append(keys, key)
|
|
}
|
|
return
|
|
}
|
|
|
|
func (t *Transport) IdleConnCountForTesting(cacheKey string) int {
|
|
t.lk.Lock()
|
|
defer t.lk.Unlock()
|
|
if t.idleConn == nil {
|
|
return 0
|
|
}
|
|
conns, ok := t.idleConn[cacheKey]
|
|
if !ok {
|
|
return 0
|
|
}
|
|
return len(conns)
|
|
}
|
|
|
|
func NewTestTimeoutHandler(handler Handler, ch <-chan int64) Handler {
|
|
f := func() <-chan int64 {
|
|
return ch
|
|
}
|
|
return &timeoutHandler{handler, f, ""}
|
|
}
|