diff options
Diffstat (limited to 'src/pkg/http/export_test.go')
-rw-r--r-- | src/pkg/http/export_test.go | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/src/pkg/http/export_test.go b/src/pkg/http/export_test.go new file mode 100644 index 000000000..47c687760 --- /dev/null +++ b/src/pkg/http/export_test.go @@ -0,0 +1,34 @@ +// 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) +} |