1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
import gevent
import greentest
class appender(object):
def __init__(self, lst, item):
self.lst = lst
self.item = item
def __call__(self, *args):
self.lst.append(self.item)
class Test(greentest.TestCase):
count = 2
def test_greenlet_link(self):
lst = []
# test that links are executed in the same order as they were added
g = gevent.spawn(lst.append, 0)
for i in xrange(1, self.count):
g.link(appender(lst, i))
g.join()
self.assertEqual(lst, range(self.count))
class Test3(Test):
count = 3
class Test4(Test):
count = 4
class TestM(Test):
count = 1000
class TestSleep0(greentest.TestCase):
def test(self):
lst = []
gevent.spawn(sleep0, lst, '1')
gevent.spawn(sleep0, lst, '2')
gevent.wait()
self.assertEqual(' '.join(lst), '1A 2A 1B 2B')
def sleep0(lst, param):
lst.append(param + 'A')
gevent.sleep(0)
lst.append(param + 'B')
if __name__ == '__main__':
greentest.main()
|
