You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
5 years ago
|
import gevent
|
||
|
|
import gevent.testing as greentest
|
||
|
|
from gevent.lock import Semaphore
|
||
|
|
|
||
|
|
|
||
|
|
class Testiwait(greentest.TestCase):
|
||
|
|
|
||
|
|
def test_noiter(self):
|
||
|
|
# Test that gevent.iwait returns objects which can be iterated upon
|
||
|
|
# without additional calls to iter()
|
||
|
|
|
||
|
|
sem1 = Semaphore()
|
||
|
|
sem2 = Semaphore()
|
||
|
|
|
||
|
|
gevent.spawn(sem1.release)
|
||
|
|
ready = next(gevent.iwait((sem1, sem2)))
|
||
|
|
self.assertEqual(sem1, ready)
|
||
|
|
|
||
|
|
def test_iwait_partial(self):
|
||
|
|
# Test that the iwait context manager allows the iterator to be
|
||
|
|
# consumed partially without a memory leak.
|
||
|
|
|
||
|
|
sem = Semaphore()
|
||
|
|
let = gevent.spawn(sem.release)
|
||
|
|
with gevent.iwait((sem,), timeout=0.01) as iterator:
|
||
|
|
self.assertEqual(sem, next(iterator))
|
||
|
|
let.get()
|
||
|
|
|
||
|
|
def test_iwait_nogarbage(self):
|
||
|
|
sem1 = Semaphore()
|
||
|
|
sem2 = Semaphore()
|
||
|
|
let = gevent.spawn(sem1.release)
|
||
|
|
with gevent.iwait((sem1, sem2)) as iterator:
|
||
|
|
self.assertEqual(sem1, next(iterator))
|
||
|
|
self.assertEqual(sem2.linkcount(), 1)
|
||
|
|
|
||
|
|
self.assertEqual(sem2.linkcount(), 0)
|
||
|
|
let.get()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
greentest.main()
|