test timeout override

This commit is contained in:
aisch 2015-11-23 12:06:40 -08:00
parent 146bbd1cbb
commit 701e2f97f4
1 changed files with 22 additions and 7 deletions

29
test.py
View File

@ -100,6 +100,15 @@ def queued_pool(params):
)
@pytest.fixture
def empty_queued_pool(request, queued_pool):
queued = [queued_pool.acquire() for _ in range(queued_pool.max_size)]
request.addfinalizer(lambda: [cxn.release() for cxn in queued])
overflow = [queued_pool.acquire() for _ in range(queued_pool.max_overflow)]
request.addfinalizer(lambda: [cxn.release() for cxn in overflow])
return queued_pool
def test_use_it():
params = pika.URLParameters(
'amqp://guest:guest@localhost:5672/?'
@ -181,11 +190,17 @@ class TestQueuedPool(object):
cxn.release()
assert fairy.cxn.is_closed
def test_timeout(self, request, queued_pool):
queued = [queued_pool.acquire() for _ in range(queued_pool.max_size)]
request.addfinalizer(lambda: [cxn.release() for cxn in queued])
overflow = [queued_pool.acquire() for _ in range(queued_pool.max_overflow)]
request.addfinalizer(lambda: [cxn.release() for cxn in overflow])
queued_pool.timeout = 1
def test_timeout(self, empty_queued_pool):
empty_queued_pool.timeout = 2
st = time.time()
with pytest.raises(pika_pool.Timeout):
queued_pool.acquire()
empty_queued_pool.acquire()
elapsed = time.time() - st
assert elapsed < 2.5
def test_timeout_override(self, empty_queued_pool):
st = time.time()
with pytest.raises(pika_pool.Timeout):
empty_queued_pool.acquire(timeout=1)
elapsed = time.time() - st
assert elapsed < 1.5