Method TestClose

net2/http2/simple_pool_test.go:282–340

(c *C)

}

func (s *SimplePoolSuite) TestClose(c *C) {
	server, addr := test_utils.SetupTestServer(false)
	defer server.Close()

	// do single request
	params := ConnectionParams{
		MaxIdle: 1,
	}
	pool := NewSimplePool(addr, params)
	pool.closeWait = 100 * time.Millisecond

	// do 10 requests concurrently
	origMaxProcs := runtime.GOMAXPROCS(runtime.NumCPU())
	defer func() { runtime.GOMAXPROCS(origMaxProcs) }()

	const count = 10
	finished := make(chan bool)
	for i := 0; i < count; i++ {
		go func() {
			req, err := http.NewRequest("GET", "/", nil)
			c.Assert(err, IsNil)

			_, err = pool.Do(req)
			c.Assert(err, IsNil)
			finished <- true
		}()
	}

	for i := 0; i < count; i++ {
		select {
		case <-finished:
			// cool

		case <-time.After(5 * time.Second):
			c.FailNow()
		}
	}

	c.Assert(pool.conns.NumAlive() > 0, IsTrue)

	pool.Close()

	failCount := 0
	for ; failCount < 100; failCount++ {
		time.Sleep(10 * time.Millisecond)
		if pool.conns.NumAlive() == 0 {
			break
		}
	}

	c.Assert(failCount < 100, IsTrue)

	req, err := http.NewRequest("GET", "/connection_refused", nil)
	c.Assert(err, IsNil)
	_, err = pool.Do(req)
	c.Assert(err, NotNil)
	_, ok := err.(DialError)
	c.Assert(ok, IsTrue)