Avoid ResourceWarnings in Python 3.
This commit is contained in:
parent
edc23501f4
commit
5eb13b26fa
@ -1018,7 +1018,7 @@ class _TestMaxOpenSockets(_TestPoolingBase):
|
||||
def test_wait_queue_timeout(self):
|
||||
wait_queue_timeout = 2 # Seconds
|
||||
pool = self.get_pool_with_wait_queue_timeout(wait_queue_timeout)
|
||||
pool.get_socket()
|
||||
sock_info = pool.get_socket()
|
||||
start = time.time()
|
||||
self.assertRaises(ConnectionFailure, pool.get_socket)
|
||||
duration = time.time() - start
|
||||
@ -1027,6 +1027,8 @@ class _TestMaxOpenSockets(_TestPoolingBase):
|
||||
"Waited %.2f seconds for a socket, expected %f" % (
|
||||
duration, wait_queue_timeout))
|
||||
|
||||
sock_info.close()
|
||||
|
||||
def test_blocking(self):
|
||||
# Verify get_socket() with no wait_queue_timeout blocks forever.
|
||||
pool = self.get_pool_with_wait_queue_timeout(None)
|
||||
@ -1046,6 +1048,7 @@ class _TestMaxOpenSockets(_TestPoolingBase):
|
||||
|
||||
self.assertEqual(t.state, 'sock')
|
||||
self.assertEqual(t.sock, s1)
|
||||
s1.close()
|
||||
|
||||
|
||||
class _TestWaitQueueMultiple(_TestPoolingBase):
|
||||
@ -1064,8 +1067,8 @@ class _TestWaitQueueMultiple(_TestPoolingBase):
|
||||
pool = self.get_pool_with_wait_queue_multiple(3)
|
||||
|
||||
# Reach max_size sockets.
|
||||
pool.get_socket()
|
||||
pool.get_socket()
|
||||
socket_info_0 = pool.get_socket()
|
||||
socket_info_1 = pool.get_socket()
|
||||
|
||||
# Reach max_size * wait_queue_multiple waiters.
|
||||
threads = []
|
||||
@ -1079,6 +1082,8 @@ class _TestWaitQueueMultiple(_TestPoolingBase):
|
||||
self.assertEqual(t.state, 'get_socket')
|
||||
|
||||
self.assertRaises(ExceededMaxWaiters, pool.get_socket)
|
||||
socket_info_0.close()
|
||||
socket_info_1.close()
|
||||
|
||||
def test_wait_queue_multiple_unset(self):
|
||||
pool = self.get_pool_with_wait_queue_multiple(None)
|
||||
@ -1095,6 +1100,9 @@ class _TestWaitQueueMultiple(_TestPoolingBase):
|
||||
for t in threads:
|
||||
self.assertEqual(t.state, 'get_socket')
|
||||
|
||||
for socket_info in socks:
|
||||
socket_info.close()
|
||||
|
||||
|
||||
class _TestPoolSocketSharing(_TestPoolingBase):
|
||||
"""Directly test that two simultaneous operations don't share a socket. To
|
||||
|
||||
Loading…
Reference in New Issue
Block a user