libgo/go/http/export_test.go

   1 // Copyright 2011 The Go Authors.  All rights reserved.
   2 // Use of this source code is governed by a BSD-style
   3 // license that can be found in the LICENSE file.
   4
   5 // Bridge package to expose http internals to tests in the http_test
   6 // package.
   7
   8 package http
   9
  10 func (t *Transport) IdleConnKeysForTesting() (keys []string) {
  11         keys = make([]string, 0)
  12         t.lk.Lock()
  13         defer t.lk.Unlock()
  14         if t.idleConn == nil {
  15                 return
  16         }
  17         for key, _ := range t.idleConn {
  18                 keys = append(keys, key)
  19         }
  20         return
  21 }
  22
  23 func (t *Transport) IdleConnCountForTesting(cacheKey string) int {
  24         t.lk.Lock()
  25         defer t.lk.Unlock()
  26         if t.idleConn == nil {
  27                 return 0
  28         }
  29         conns, ok := t.idleConn[cacheKey]
  30         if !ok {
  31                 return 0
  32         }
  33         return len(conns)
  34 }