python-eventlet/tests/isolated/env_tpool_size.py

   1 __test__ = False
   2
   3 if __name__ == '__main__':
   4     import sys
   5     import time
   6     from eventlet import tpool
   7     import eventlet
   8
   9     current = [0]
  10     highwater = [0]
  11
  12     def count():
  13         current[0] += 1
  14         time.sleep(0.01)
  15         if current[0] > highwater[0]:
  16             highwater[0] = current[0]
  17         current[0] -= 1
  18
  19     expected = int(sys.argv[1])
  20     normal = int(sys.argv[2])
  21     p = eventlet.GreenPool()
  22     for i in range(expected * 2):
  23         p.spawn(tpool.execute, count)
  24     p.waitall()
  25     assert highwater[0] > normal, "Highwater %s <= %s" % (highwater[0], normal)
  26     print('pass')