def test_maxsize_callback(self):
q = Queue(maxsize=2)
self.assertEqual(q.maxsize, 2)
yield self. async .assertEqual(q.put('hello'), None)
yield self. async .assertEqual(q.put('ciao'), None)
result = maybe_async(q.put('bla'))
self.assertEqual(q.qsize(), 2)
item = yield q.get()
self.assertEqual(item, 'hello')
self.assertEqual(q.qsize(), 2)
def test_async_get(self):
q = Queue()
self.assertEqual(q.qsize(), 0)
item = maybe_async(q.get())
self.assertIsInstance(item, Deferred)
result = yield q.put('Hello')
self.assertEqual(result, None)
self.assertTrue(item.done())
self.assertEqual(item.result, 'Hello')
self.assertEqual(q.qsize(), 0)
def test_maxsize(self):
q = Queue(maxsize=2)
self.assertEqual(q.maxsize, 2)
yield self. async .assertEqual(q.put('hello'), None)
yield self. async .assertEqual(q.put('ciao'), None)
self.assertEqual(q.qsize(), 2)
self.assertTrue(q.full())
start = default_timer()
yield self. async .assertRaises(Full, q.put, 'ciao', timeout=0.5)
self.assertTrue(default_timer() - start >= 0.5)
def test_timeout(self):
q = Queue()
self.assertEqual(q.qsize(), 0)
start = default_timer()
try:
item = yield q.get(timeout=0.5)
except Empty:
pass
self.assertTrue(default_timer() - start >= 0.5)
self.assertEqual(q.qsize(), 0)
def test_get_nowait(self):
q = Queue()
self.assertRaises(Empty, q.get_nowait)
self.assertRaises(Empty, q.get, wait=False)
r = q.put('ciao')
self.assertIsInstance(r, Deferred)
self.assertTrue(r.done())
self.assertEqual(r.result, None)
self.assertEqual(q.get_nowait(), 'ciao')
self.assertRaises(Empty, q.get_nowait)
def test_sync(self):
q = Queue()
self.assertEqual(q.qsize(), 0)
result = yield q.put('hello')
self.assertEqual(result, None)
self.assertEqual(q.qsize(), 1)
item = yield q.get()
self.assertEqual(item, 'hello')
self.assertEqual(q.qsize(), 0)
result = yield q.put('ciao')
self.assertEqual(result, None)
self.assertEqual(q.qsize(), 1)
def test_put_timeout(self):
q = Queue(maxsize=2)
self.assertEqual(q.maxsize, 2)
yield self. async .assertEqual(q.put('hello'), None)
yield self. async .assertEqual(q.put('ciao'), None)
yield self. async .assertRaises(Full, q.put, 'bla1', timeout=0.5)
result = maybe_async(q.put('bla2'))
self.assertEqual(q.qsize(), 2)
item = yield q.get()
self.assertEqual(item, 'hello')
self.assertEqual(q.qsize(), 2)
item = yield q.get()
self.assertEqual(item, 'ciao')
item = yield q.get()
self.assertEqual(item, 'bla2')
self.assertEqual(q.qsize(), 0)
def test_get_nowait_with_putter(self):
q = Queue(maxsize=2)
d = q.put('first')
self.assertTrue(d.done())
d = q.put('second')
self.assertTrue(d.done())
d = q.put('third')
self.assertFalse(d.done())
d2 = q.put('fourth')
self.assertFalse(d2.done())
# We have a putter waiting for a free slot
self.assertEqual(q.get_nowait(), 'first')
self.assertTrue(d.done())
self.assertFalse(d2.done())
self.assertEqual(q.get_nowait(), 'second')
self.assertTrue(d2.done())
self.assertEqual(q.get_nowait(), 'third')
self.assertEqual(q.get_nowait(), 'fourth')
def __init__(self):
self.queue = Queue()
def _init(self):
self._tasks = {}
self.queue = Queue()
def test_event_loop(self):
q1 = Queue(event_loop=get_request_loop())
q2 = Queue(event_loop=get_event_loop())
q3 = Queue()
self.assertEqual(q2.event_loop, q3.event_loop)
self.assertNotEqual(q1.event_loop, q3.event_loop)
def test_put_nowait(self):
q = Queue(maxsize=2)
self.assertEqual(q.put_nowait('first'), None)
self.assertEqual(q.put_nowait('second'), None)
# the queue is now full
self.assertRaises(Full, q.put_nowait, 'third')
def __init__(self, environ, start_response, response):
self.environ = environ
self.start_response = start_response
self.queue = Queue()
response.on_finished.add_errback(self.error)
self.setup(response)
